2 Unix SMB/CIFS implementation.
4 provide hooks into smbd C calls from ejs scripts
6 Copyright (C) Andrew Tridgell 2005
7 Copyright (C) Jelmer Vernooij 2005
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
24 #include "scripting/ejs/smbcalls.h"
25 #include "lib/appweb/ejs/ejs.h"
26 #include "lib/ldb/include/ldb.h"
27 #include "lib/ldb/include/ldb_errors.h"
29 #include "dsdb/samdb/samdb.h"
30 #include "librpc/ndr/libndr.h"
31 #include "libcli/security/security.h"
36 static struct ldb_context *ejs_get_ldb_context(int eid)
38 struct ldb_context *ldb = (struct ldb_context *)mprGetThisPtr(eid, "db");
40 ejsSetErrorMsg(eid, "invalid ldb connection");
46 perform an ldb search, returning an array of results
49 res = ldb.search("expression");
50 var attrs = new Array("attr1", "attr2", "attr3");
51 ldb.search("expression", attrs);
52 var basedn = "cn=this,dc=is,dc=a,dc=test";
53 ldb.search("expression", basedn, ldb.SCOPE_SUBTREE, attrs);
54 ldb.search("expression", basedn, ldb.SCOPE_SUBTREE, attrs, controls);
56 static int ejs_ldbSearch(MprVarHandle eid, int argc, struct MprVar **argv)
58 const char **attrs = NULL;
59 const char *expression;
60 const char *base = NULL;
61 struct ldb_dn *basedn = NULL;
62 int scope = LDB_SCOPE_DEFAULT;
63 TALLOC_CTX *tmp_ctx = talloc_new(mprMemCtx());
64 struct ldb_context *ldb;
66 struct ldb_control **parsed_controls = NULL;
67 struct ldb_result *res=NULL;
68 struct ldb_request *req;
70 /* validate arguments */
71 if (argc < 1 || argc > 5) {
72 ejsSetErrorMsg(eid, "ldb.search invalid number of arguments");
75 if (argc > 3 && argv[3]->type != MPR_TYPE_OBJECT) {
76 ejsSetErrorMsg(eid, "ldb.search attributes must be an object");
80 ldb = ejs_get_ldb_context(eid);
85 expression = mprToString(argv[0]);
87 base = mprToString(argv[1]);
88 /* a null basedn is valid */
91 basedn = ldb_dn_new(tmp_ctx, ldb, base);
92 if ( ! ldb_dn_validate(basedn)) {
93 ejsSetErrorMsg(eid, "ldb.search malformed base dn");
97 basedn = ldb_get_default_basedn(ldb);
100 scope = mprToInt(argv[2]);
102 case LDB_SCOPE_DEFAULT:
104 case LDB_SCOPE_ONELEVEL:
105 case LDB_SCOPE_SUBTREE:
108 ejsSetErrorMsg(eid, "ldb.search invalid scope");
113 attrs = mprToList(tmp_ctx, argv[3]);
116 const char **controls;
117 controls = mprToList(tmp_ctx, argv[4]);
119 parsed_controls = ldb_parse_control_strings(ldb, tmp_ctx, controls);
120 if (!parsed_controls) {
121 ejsSetErrorMsg(eid, "ldb.search cannot parse controls: %s",
128 res = talloc_zero(tmp_ctx, struct ldb_result);
130 return LDB_ERR_OPERATIONS_ERROR;
133 ret = ldb_build_search_req(&req, ldb, tmp_ctx,
140 ldb_search_default_callback);
142 if (ret == LDB_SUCCESS) {
144 ldb_set_timeout(ldb, req, 0); /* use default timeout */
146 ret = ldb_request(ldb, req);
148 if (ret == LDB_SUCCESS) {
149 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
153 if (ret != LDB_SUCCESS) {
154 ejsSetErrorMsg(eid, "ldb.search failed - %s", ldb_errstring(ldb));
155 mpr_Return(eid, mprLdbResult(ldb, ret, NULL));
157 mpr_Return(eid, mprLdbResult(ldb, ret, res));
160 talloc_free(tmp_ctx);
164 talloc_free(tmp_ctx);
170 perform an ldb add or modify
172 static int ejs_ldbAddModify(MprVarHandle eid, int argc, struct MprVar **argv,
173 int fn(struct ldb_context *, const struct ldb_message *))
175 const char *ldifstring;
176 struct ldb_context *ldb;
177 struct ldb_ldif *ldif;
178 int ret = 0, count=0;
181 ejsSetErrorMsg(eid, "ldb.add/modify invalid arguments");
185 ldifstring = mprToString(argv[0]);
186 if (ldifstring == NULL) {
187 ejsSetErrorMsg(eid, "ldb.add/modify invalid arguments");
191 ldb = ejs_get_ldb_context(eid);
196 while ((ldif = ldb_ldif_read_string(ldb, &ldifstring))) {
198 ret = fn(ldb, ldif->msg);
204 ejsSetErrorMsg(eid, "ldb.add/modify invalid ldif");
208 mpr_Return(eid, mprLdbResult(ldb, ret, NULL));
214 perform an ldb delete
218 static int ejs_ldbDelete(MprVarHandle eid, int argc, struct MprVar **argv)
221 struct ldb_context *ldb;
225 ejsSetErrorMsg(eid, "ldb.delete invalid arguments");
229 ldb = ejs_get_ldb_context(eid);
234 dn = ldb_dn_new(ldb, ldb, mprToString(argv[0]));
235 if ( ! ldb_dn_validate(dn)) {
236 ejsSetErrorMsg(eid, "ldb.delete malformed dn");
240 ret = ldb_delete(ldb, dn);
244 mpr_Return(eid, mprLdbResult(ldb, ret, NULL));
249 perform an ldb rename
251 ok = ldb.rename(dn1, dn2);
253 static int ejs_ldbRename(MprVarHandle eid, int argc, struct MprVar **argv)
255 struct ldb_dn *dn1, *dn2;
256 struct ldb_context *ldb;
260 ejsSetErrorMsg(eid, "ldb.rename invalid arguments");
264 ldb = ejs_get_ldb_context(eid);
269 dn1 = ldb_dn_new(ldb, ldb, mprToString(argv[0]));
270 dn2 = ldb_dn_new(ldb, ldb, mprToString(argv[1]));
271 if ( ! ldb_dn_validate(dn1) || ! ldb_dn_validate(dn2)) {
272 ejsSetErrorMsg(eid, "ldb.rename invalid or malformed arguments");
276 ret = ldb_rename(ldb, dn1, dn2);
281 mpr_Return(eid, mprLdbResult(ldb, ret, NULL));
286 get last error message
288 ok = ldb.errstring();
290 static int ejs_ldbErrstring(MprVarHandle eid, int argc, struct MprVar **argv)
292 struct ldb_context *ldb;
294 ldb = ejs_get_ldb_context(eid);
299 mpr_Return(eid, mprString(ldb_errstring(ldb)));
306 dataout = ldb.encode(datain)
308 static int ejs_base64encode(MprVarHandle eid, int argc, struct MprVar **argv)
313 ejsSetErrorMsg(eid, "ldb.base64encode invalid argument count");
317 if (argv[0]->type == MPR_TYPE_STRING) {
318 const char *orig = mprToString(argv[0]);
319 ret = ldb_base64_encode(mprMemCtx(), orig, strlen(orig));
323 blob = mprToDataBlob(argv[0]);
325 ret = ldb_base64_encode(mprMemCtx(), (char *)blob->data, blob->length);
329 mpr_Return(eid, mprCreateUndefinedVar());
331 mpr_Return(eid, mprString(ret));
342 dataout = ldb.decode(datain)
344 static int ejs_base64decode(MprVarHandle eid, int argc, struct MprVar **argv)
350 ejsSetErrorMsg(eid, "ldb.base64encode invalid argument count");
354 tmp = talloc_strdup(mprMemCtx(), mprToString(argv[0]));
355 ret = ldb_base64_decode(tmp);
357 mpr_Return(eid, mprCreateUndefinedVar());
360 blob.data = (uint8_t *)tmp;
362 mpr_Return(eid, mprDataBlob(blob));
373 dataout = ldb.dn_escape(datain)
375 static int ejs_dn_escape(MprVarHandle eid, int argc, struct MprVar **argv)
381 ejsSetErrorMsg(eid, "ldb.dn_escape invalid argument count");
385 val = data_blob_string_const(mprToString(argv[0]));
387 ret = ldb_dn_escape_value(mprMemCtx(), val);
389 mpr_Return(eid, mprCreateUndefinedVar());
391 mpr_Return(eid, mprString(ret));
402 ok = ldb.add(ldifstring);
404 static int ejs_ldbAdd(MprVarHandle eid, int argc, struct MprVar **argv)
406 return ejs_ldbAddModify(eid, argc, argv, ldb_add);
410 perform an ldb modify
413 ok = ldb.modify(ldifstring);
415 static int ejs_ldbModify(MprVarHandle eid, int argc, struct MprVar **argv)
417 return ejs_ldbAddModify(eid, argc, argv, ldb_modify);
421 connect to a database
423 ok = ldb.connect(dbfile);
424 ok = ldb.connect(dbfile, "modules:modlist");
426 ldb.credentials or ldb.session_info may be setup first
429 static int ejs_ldbConnect(MprVarHandle eid, int argc, char **argv)
431 struct ldb_context *ldb;
432 struct auth_session_info *session_info = NULL;
433 struct cli_credentials *creds = NULL;
434 struct MprVar *credentials, *session;
435 struct MprVar *this = mprGetProperty(ejsGetLocalObject(eid), "this", 0);
440 ejsSetErrorMsg(eid, "ldb.connect invalid arguments");
444 credentials = mprGetProperty(this, "credentials", NULL);
446 creds = talloc_get_type(mprGetPtr(credentials, "creds"), struct cli_credentials);
449 session = mprGetProperty(this, "session_info", NULL);
451 session_info = talloc_get_type(mprGetPtr(session, "session_info"), struct auth_session_info);
456 ldb = ldb_wrap_connect(mprMemCtx(), mprEventCtx(), mprLpCtx(), dbfile,
458 0, (const char **)(argv+1));
460 ejsSetErrorMsg(eid, "ldb.connect failed to open %s", dbfile);
463 mprSetThisPtr(eid, "db", ldb);
464 mpr_Return(eid, mprCreateBoolVar(ldb != NULL));
470 close a db connection
472 static int ejs_ldbClose(MprVarHandle eid, int argc, struct MprVar **argv)
474 struct ldb_context *ldb;
477 ejsSetErrorMsg(eid, "ldb.close invalid arguments");
481 ldb = ejs_get_ldb_context(eid);
486 mprSetThisPtr(eid, "db", NULL);
487 mpr_Return(eid, mprCreateBoolVar(true));
493 start a ldb transaction
495 ok = ldb.transaction_start();
497 static int ejs_ldbTransactionStart(MprVarHandle eid, int argc, struct MprVar **argv)
499 struct ldb_context *ldb;
503 ejsSetErrorMsg(eid, "ldb.transaction_start invalid arguments");
507 ldb = ejs_get_ldb_context(eid);
512 ret = ldb_transaction_start(ldb);
514 mpr_Return(eid, mprCreateBoolVar(ret == 0));
519 cancel a ldb transaction
521 ok = ldb.transaction_cancel();
523 static int ejs_ldbTransactionCancel(MprVarHandle eid, int argc, struct MprVar **argv)
525 struct ldb_context *ldb;
529 ejsSetErrorMsg(eid, "ldb.transaction_cancel invalid arguments");
533 ldb = ejs_get_ldb_context(eid);
538 ret = ldb_transaction_cancel(ldb);
540 mpr_Return(eid, mprCreateBoolVar(ret == 0));
545 commit a ldb transaction
547 ok = ldb.transaction_commit();
549 static int ejs_ldbTransactionCommit(MprVarHandle eid, int argc, struct MprVar **argv)
551 struct ldb_context *ldb;
555 ejsSetErrorMsg(eid, "ldb.transaction_commit invalid arguments");
559 ldb = ejs_get_ldb_context(eid);
564 ret = ldb_transaction_commit(ldb);
566 mpr_Return(eid, mprCreateBoolVar(ret == 0));
571 commit a ldb attach a dsdb_schema from ldif files
573 ok = ldb.attach_dsdb_schema_from_ldif("prefixMap ldif content", "definition ldif content")
575 static int ejs_ldb_attach_dsdb_schema_from_ldif(MprVarHandle eid, int argc, char **argv)
577 struct ldb_context *ldb;
583 ejsSetErrorMsg(eid, "ldb.attach_dsdb_schema_from_ldif invalid arguments");
587 ldb = ejs_get_ldb_context(eid);
595 status = dsdb_attach_schema_from_ldif_file(ldb, pf, df);
597 mpr_Return(eid, mprWERROR(status));
602 set a particular invocationId against the running LDB
604 ok = ldb.set_ntds_invocationId("7729aa4b-f990-41ad-b81a-8b6a14090f41");
606 static int ejs_ldb_set_ntds_invocationId(MprVarHandle eid, int argc, char **argv)
608 struct ldb_context *ldb;
615 ejsSetErrorMsg(eid, "ldb.set_ntds_invocationId invalid arguments");
619 ldb = ejs_get_ldb_context(eid);
626 status = GUID_from_string(guid_str, &guid);
627 if (!NT_STATUS_IS_OK(status)) {
628 ejsSetErrorMsg(eid, "ldb.set_ntds_invocationId - failed to parse GUID '%s' %s\n",
629 guid_str, nt_errstr(status));
633 ok = samdb_set_ntds_invocation_id(ldb, &guid);
635 ejsSetErrorMsg(eid, "ldb.set_ntds_invocationId - failed to set cached ntds invocationId\n");
639 mpr_Return(eid, mprCreateBoolVar(ok));
644 attach a particular ntds objectGUID against the current ldb
646 ok = ldb.set_ntds_objectGUID("7729aa4b-f990-41ad-b81a-8b6a14090f41");
648 static int ejs_ldb_set_ntds_objectGUID(MprVarHandle eid, int argc, char **argv)
650 struct ldb_context *ldb;
657 ejsSetErrorMsg(eid, "ldb.set_ntds_objectGUID invalid arguments");
661 ldb = ejs_get_ldb_context(eid);
668 status = GUID_from_string(guid_str, &guid);
669 if (!NT_STATUS_IS_OK(status)) {
670 ejsSetErrorMsg(eid, "ldb.set_ntds_objectGUID - failed to parse GUID '%s' %s\n",
671 guid_str, nt_errstr(status));
675 ok = samdb_set_ntds_invocation_id(ldb, &guid);
677 ejsSetErrorMsg(eid, "ldb.set_ntds_objectGUID - failed to set cached ntds invocationId\n");
681 mpr_Return(eid, mprCreateBoolVar(ok));
686 attach a particular domain SID against the current ldb
688 ok = ldb.set_domain_sid("S-S-1-5-21-3065342217-3567412576-2214182334");
690 static int ejs_ldb_set_domain_sid(MprVarHandle eid, int argc, char **argv)
692 struct ldb_context *ldb;
693 struct dom_sid *dom_sid;
698 ejsSetErrorMsg(eid, "ldb.set_domain_sid invalid arguments");
702 ldb = ejs_get_ldb_context(eid);
707 dom_sid_str = argv[0];
709 dom_sid = dom_sid_parse_talloc(NULL, dom_sid_str);
711 ejsSetErrorMsg(eid, "ldb.set_domain_sid - failed to parse domain sid '%s'\n",
716 ok = samdb_set_domain_sid(ldb, dom_sid);
717 talloc_free(dom_sid);
719 ejsSetErrorMsg(eid, "ldb.set_domain_sid - failed to set cached ntds invocationId\n");
723 mpr_Return(eid, mprCreateBoolVar(ok));
728 initialise ldb ejs subsystem
730 static int ejs_ldb_init(MprVarHandle eid, int argc, struct MprVar **argv)
732 struct MprVar *ldb = mprInitObject(eid, "ldb", argc, argv);
734 mprSetStringCFunction(ldb, "connect", ejs_ldbConnect);
735 mprSetCFunction(ldb, "search", ejs_ldbSearch);
736 mprSetCFunction(ldb, "add", ejs_ldbAdd);
737 mprSetCFunction(ldb, "modify", ejs_ldbModify);
738 mprSetCFunction(ldb, "del", ejs_ldbDelete);
739 mprSetCFunction(ldb, "rename", ejs_ldbRename);
740 mprSetCFunction(ldb, "errstring", ejs_ldbErrstring);
741 mprSetCFunction(ldb, "encode", ejs_base64encode);
742 mprSetCFunction(ldb, "decode", ejs_base64decode);
743 mprSetCFunction(ldb, "dn_escape", ejs_dn_escape);
744 mprSetCFunction(ldb, "close", ejs_ldbClose);
745 mprSetCFunction(ldb, "transaction_start", ejs_ldbTransactionStart);
746 mprSetCFunction(ldb, "transaction_cancel", ejs_ldbTransactionCancel);
747 mprSetCFunction(ldb, "transaction_commit", ejs_ldbTransactionCommit);
748 mprSetStringCFunction(ldb, "attach_dsdb_schema_from_ldif",
749 ejs_ldb_attach_dsdb_schema_from_ldif);
750 mprSetStringCFunction(ldb, "set_ntds_invocationId",
751 ejs_ldb_set_ntds_invocationId);
752 mprSetStringCFunction(ldb, "set_ntds_objectGUID",
753 ejs_ldb_set_ntds_objectGUID);
754 mprSetStringCFunction(ldb, "set_domain_sid",
755 ejs_ldb_set_domain_sid);
756 mprSetVar(ldb, "SCOPE_BASE", mprCreateNumberVar(LDB_SCOPE_BASE));
757 mprSetVar(ldb, "SCOPE_ONE", mprCreateNumberVar(LDB_SCOPE_ONELEVEL));
758 mprSetVar(ldb, "SCOPE_SUBTREE", mprCreateNumberVar(LDB_SCOPE_SUBTREE));
759 mprSetVar(ldb, "SCOPE_DEFAULT", mprCreateNumberVar(LDB_SCOPE_DEFAULT));
766 setup C functions that be called from ejs
768 NTSTATUS smb_setup_ejs_ldb(void)
770 ejsDefineCFunction(-1, "ldb_init", ejs_ldb_init, NULL, MPR_VAR_SCRIPT_HANDLE);