samba-mirror/source/scripting/ejs/ejsrpc.c

/* 
   Unix SMB/CIFS implementation.

   provide interfaces to rpc calls from ejs scripts

   Copyright (C) Andrew Tridgell 2005
   
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.
   
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
   
   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#include "includes.h"
#include "lib/ejs/ejs.h"
#include "scripting/ejs/ejsrpc.h"
#include "librpc/gen_ndr/ndr_security.h"

NTSTATUS ejs_pull_rpc(int eid, const char *callname, 
		      struct MprVar *v, void *ptr, ejs_pull_function_t ejs_pull)
{
	struct ejs_rpc *ejs = talloc(ptr, struct ejs_rpc);	
	NT_STATUS_HAVE_NO_MEMORY(ejs);
	ejs->eid = eid;
	ejs->callname = callname;
	return ejs_pull(ejs, v, ptr);
}


NTSTATUS ejs_push_rpc(int eid, const char *callname, 
		      struct MprVar *v, const void *ptr, ejs_push_function_t ejs_push)
{
	struct ejs_rpc *ejs = talloc(ptr, struct ejs_rpc);
	NT_STATUS_HAVE_NO_MEMORY(ejs);
	ejs->eid = eid;
	ejs->callname = callname;
	return ejs_push(ejs, v, ptr);
}

/*
  set the switch var to be used by the next union switch
*/
void ejs_set_switch(struct ejs_rpc *ejs, uint32_t switch_var)
{
	ejs->switch_var = switch_var;
}

/*
  panic in the ejs wrapper code
 */
NTSTATUS ejs_panic(struct ejs_rpc *ejs, const char *why)
{
	ejsSetErrorMsg(ejs->eid, "rpc_call '%s' failed - %s", ejs->callname, why);
	return NT_STATUS_INTERNAL_ERROR;
}

/*
  find a mpr component, allowing for sub objects, using the '.' convention
*/
static struct MprVar *mprGetVar(struct MprVar *v, const char *name)
{
	const char *p = strchr(name, '.');
	char *objname;
	struct MprVar *v2;
	if (p == NULL) {
		return mprGetProperty(v, name, NULL);
	}
	objname = talloc_strndup(mprMemCtx(), name, p-name);
	if (objname == NULL) {
		return NULL;
	}
	v2 = mprGetProperty(v, objname, NULL);
	if (v2 == NULL) {
		talloc_free(objname);
		return NULL;
	}
	v2 = mprGetVar(v2, p+1);
	talloc_free(objname);
	return v2;
}


/*
  set a mpr component, allowing for sub objects, using the '.' convention
*/
static NTSTATUS mprSetVar(struct MprVar *v, const char *name, struct MprVar val)
{
	const char *p = strchr(name, '.');
	char *objname;
	struct MprVar *v2;
	NTSTATUS status;
	if (p == NULL) {
		v2 = mprSetProperty(v, name, &val);
		if (v2 == NULL) {
			DEBUG(1,("mprSetVar unable to set '%s'\n", name));
			return NT_STATUS_INVALID_PARAMETER_MIX;
		}
		return NT_STATUS_OK;
	}
	objname = talloc_strndup(mprMemCtx(), name, p-name);
	if (objname == NULL) {
		return NT_STATUS_NO_MEMORY;
	}
	v2 = mprGetProperty(v, objname, NULL);
	if (v2 == NULL) {
		struct MprVar val2 = mprCreateObjVar(objname, MPR_DEFAULT_HASH_SIZE);
		v2 = mprCreateProperty(v, objname, &val2);
	}
	status = mprSetVar(v2, p+1, val);
	talloc_free(objname);
	return status;
}


/*
  start the ejs pull process for a structure
*/
NTSTATUS ejs_pull_struct_start(struct ejs_rpc *ejs, struct MprVar **v, const char *name)
{
	*v = mprGetVar(*v, name);
	if (*v == NULL) {
		DEBUG(1,("ejs_pull_struct_start: missing structure '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER;
	}
	return NT_STATUS_OK;
}


/*
  start the ejs push process for a structure
*/
NTSTATUS ejs_push_struct_start(struct ejs_rpc *ejs, struct MprVar **v, const char *name)
{
	NDR_CHECK(mprSetVar(*v, name, mprCreateObjVar(name, MPR_DEFAULT_HASH_SIZE)));
	*v = mprGetVar(*v, name);
	if (*v == NULL) {
		DEBUG(1,("ejs_push_struct_start: missing structure '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER;
	}
	return NT_STATUS_OK;
}

/*
  pull a uint8 from a mpr variable to a C element
*/
NTSTATUS ejs_pull_uint8(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, uint8_t *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_uint8: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*r = mprVarToInteger(var);
	return NT_STATUS_OK;
	
}

NTSTATUS ejs_push_uint8(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, const uint8_t *r)
{
	return mprSetVar(v, name, mprCreateIntegerVar(*r));
}

/*
  pull a uint16 from a mpr variable to a C element
*/
NTSTATUS ejs_pull_uint16(struct ejs_rpc *ejs, 
			 struct MprVar *v, const char *name, uint16_t *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_uint16: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*r = mprVarToInteger(var);
	return NT_STATUS_OK;
	
}

NTSTATUS ejs_push_uint16(struct ejs_rpc *ejs, 
			 struct MprVar *v, const char *name, const uint16_t *r)
{
	return mprSetVar(v, name, mprCreateIntegerVar(*r));
}

/*
  pull a uint32 from a mpr variable to a C element
*/
NTSTATUS ejs_pull_uint32(struct ejs_rpc *ejs, 
			 struct MprVar *v, const char *name, uint32_t *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_push_uint32: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*r = mprVarToInteger(var);
	return NT_STATUS_OK;
}

NTSTATUS ejs_push_uint32(struct ejs_rpc *ejs, 
			 struct MprVar *v, const char *name, const uint32_t *r)
{
	return mprSetVar(v, name, mprCreateIntegerVar(*r));
}

NTSTATUS ejs_pull_hyper(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, uint64_t *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_hyper: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*r = mprVarToInteger(var);
	return NT_STATUS_OK;
}

NTSTATUS ejs_push_hyper(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, const uint64_t *r)
{
	return mprSetVar(v, name, mprCreateIntegerVar(*r));
}

NTSTATUS ejs_pull_dlong(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, uint64_t *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_dlong: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*r = mprVarToInteger(var);
	return NT_STATUS_OK;
}

NTSTATUS ejs_push_dlong(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, const uint64_t *r)
{
	return mprSetVar(v, name, mprCreateIntegerVar(*r));
}

NTSTATUS ejs_pull_udlong(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, uint64_t *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_udlong: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*r = mprVarToInteger(var);
	return NT_STATUS_OK;
}

NTSTATUS ejs_push_udlong(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, const uint64_t *r)
{
	return mprSetVar(v, name, mprCreateIntegerVar(*r));
}

NTSTATUS ejs_pull_NTTIME(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, uint64_t *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_NTTIME: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*r = mprVarToInteger(var);
	return NT_STATUS_OK;
}

NTSTATUS ejs_push_NTTIME(struct ejs_rpc *ejs, 
			struct MprVar *v, const char *name, const uint64_t *r)
{
	return mprSetVar(v, name, mprCreateIntegerVar(*r));
}


/*
  pull a enum from a mpr variable to a C element
  a enum is just treating as an unsigned integer at this level
*/
NTSTATUS ejs_pull_enum(struct ejs_rpc *ejs, 
		       struct MprVar *v, const char *name, unsigned *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_enum: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*r = mprVarToInteger(var);
	return NT_STATUS_OK;
	
}

NTSTATUS ejs_push_enum(struct ejs_rpc *ejs, 
		       struct MprVar *v, const char *name, const unsigned *r)
{
	return mprSetVar(v, name, mprCreateIntegerVar(*r));
}


/*
  pull a string
*/
NTSTATUS ejs_pull_string(struct ejs_rpc *ejs, 
			 struct MprVar *v, const char *name, const char **s)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_string: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	*s = mprToString(var);
	return NT_STATUS_OK;
}

/*
  push a string
*/
NTSTATUS ejs_push_string(struct ejs_rpc *ejs, 
			 struct MprVar *v, const char *name, const char *s)
{
	return mprSetVar(v, name, mprCreateStringVar(s, True));
}

/*
  setup a constant int
*/
void ejs_set_constant_int(int eid, const char *name, int value)
{
	struct MprVar *v = ejsGetGlobalObject(eid);
	mprSetVar(v, name, mprCreateIntegerVar(value));
}

/*
  setup a constant string
*/
void ejs_set_constant_string(int eid, const char *name, const char *value)
{
	struct MprVar *v = ejsGetGlobalObject(eid);
	mprSetVar(v, name, mprCreateStringVar(value, False));
}


NTSTATUS ejs_pull_dom_sid(struct ejs_rpc *ejs, 
			  struct MprVar *v, const char *name, struct dom_sid *r)
{
	struct MprVar *var;
	struct dom_sid *sid;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_dom_sid: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	sid = dom_sid_parse_talloc(ejs, mprToString(var));
	NT_STATUS_HAVE_NO_MEMORY(sid);
	*r = *sid;
	return NT_STATUS_OK;
}

NTSTATUS ejs_push_dom_sid(struct ejs_rpc *ejs, 
			  struct MprVar *v, const char *name, const struct dom_sid *r)
{
	char *sidstr = dom_sid_string(ejs, r);
	NT_STATUS_HAVE_NO_MEMORY(sidstr);
	return mprSetVar(v, name, mprCreateStringVar(sidstr, True));
}

NTSTATUS ejs_pull_GUID(struct ejs_rpc *ejs, 
		       struct MprVar *v, const char *name, struct GUID *r)
{
	struct MprVar *var;
	var = mprGetVar(v, name);
	if (var == NULL) {
		DEBUG(1,("ejs_pull_GUID: unable to find '%s'\n", name));
		return NT_STATUS_INVALID_PARAMETER_MIX;
	}
	return GUID_from_string(mprToString(var), r);
}

NTSTATUS ejs_push_GUID(struct ejs_rpc *ejs, 
		       struct MprVar *v, const char *name, const struct GUID *r)
{
	char *guid = GUID_string(ejs, r);
	NT_STATUS_HAVE_NO_MEMORY(guid);
	return mprSetVar(v, name, mprCreateStringVar(guid, True));
}

NTSTATUS ejs_push_null(struct ejs_rpc *ejs, struct MprVar *v, const char *name)
{
	return mprSetVar(v, name, mprCreatePtrVar(NULL, name));
}

BOOL ejs_pull_null(struct ejs_rpc *ejs, struct MprVar *v, const char *name)
{
	v = mprGetVar(v, name);
	if (v == NULL) {
		return True;
	}
	if (v->type == MPR_TYPE_PTR && v->ptr == NULL) {
		return True;
	}
	return False;
}