1 /*
2    Unix SMB/CIFS implementation.
3 
4    provide access to randomisation functions
5 
6    Copyright (C) Andrew Tridgell 2005
7 
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12 
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17 
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22 
23 #include "includes.h"
24 #include "scripting/ejs/smbcalls.h"
25 #include "lib/appweb/ejs/ejs.h"
26 #include "system/passwd.h"
27 #include "librpc/gen_ndr/ndr_misc.h"
28 
29 /*
30   usage:
31       var i = random();
32 */
ejs_random(MprVarHandle eid,int argc,struct MprVar ** argv)33 static int ejs_random(MprVarHandle eid, int argc, struct MprVar **argv)
34 {
35 	mpr_Return(eid, mprCreateIntegerVar(generate_random()));
36 	return 0;
37 }
38 
39 /*
40   usage:
41       var s = randpass(len);
42 */
ejs_randpass(MprVarHandle eid,int argc,struct MprVar ** argv)43 static int ejs_randpass(MprVarHandle eid, int argc, struct MprVar **argv)
44 {
45 	char *s;
46 	if (argc != 1 || !mprVarIsNumber(argv[0]->type)) {
47 		ejsSetErrorMsg(eid, "randpass invalid arguments");
48 		return -1;
49 	}
50 	s = generate_random_str(mprMemCtx(), mprToInt(argv[0]));
51 	mpr_Return(eid, mprString(s));
52 	talloc_free(s);
53 	return 0;
54 }
55 
56 /*
57   usage:
58       var guid = randguid();
59 */
ejs_randguid(MprVarHandle eid,int argc,struct MprVar ** argv)60 static int ejs_randguid(MprVarHandle eid, int argc, struct MprVar **argv)
61 {
62 	struct GUID guid = GUID_random();
63 	char *s = GUID_string(mprMemCtx(), &guid);
64 	mpr_Return(eid, mprString(s));
65 	talloc_free(s);
66 	return 0;
67 }
68 
69 /*
70   usage:
71       var sid = randsid();
72 */
ejs_randsid(MprVarHandle eid,int argc,struct MprVar ** argv)73 static int ejs_randsid(MprVarHandle eid, int argc, struct MprVar **argv)
74 {
75 	char *s = talloc_asprintf(mprMemCtx(), "S-1-5-21-%8u-%8u-%8u",
76 				  (unsigned)generate_random(),
77 				  (unsigned)generate_random(),
78 				  (unsigned)generate_random());
79 	mpr_Return(eid, mprString(s));
80 	talloc_free(s);
81 	return 0;
82 }
83 
84 /*
85   initialise random ejs subsystem
86 */
ejs_random_init(MprVarHandle eid,int argc,struct MprVar ** argv)87 static int ejs_random_init(MprVarHandle eid, int argc, struct MprVar **argv)
88 {
89 	struct MprVar *obj = mprInitObject(eid, "random", argc, argv);
90 
91 	mprSetCFunction(obj, "random", ejs_random);
92 	mprSetCFunction(obj, "randpass", ejs_randpass);
93 	mprSetCFunction(obj, "randguid", ejs_randguid);
94 	mprSetCFunction(obj, "randsid", ejs_randsid);
95 	return 0;
96 }
97 
98 /*
99   setup C functions that be called from ejs
100 */
smb_setup_ejs_random(void)101 NTSTATUS smb_setup_ejs_random(void)
102 {
103 	ejsDefineCFunction(-1, "random_init", ejs_random_init, NULL, MPR_VAR_SCRIPT_HANDLE);
104 	return NT_STATUS_OK;
105 }
106