svcctl: use offered/needed for buffer sizes as in spoolss.
[ira/wip.git] / source3 / rpc_server / srv_svcctl_nt.c
index a7215ac68601d4e600bbc0a9662df945c9725965..ddfe0df10b9f1ff2ee67776d2e054d6527299a5e 100644 (file)
@@ -6,7 +6,7 @@
  *
  *  Largely Rewritten (Again) by:
  *  Copyright (C) Gerald (Jerry) Carter             2005.
- *  Copyright (C) Guenther Deschner                 2008.
+ *  Copyright (C) Guenther Deschner                 2008,2009.
  *
  *  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
@@ -32,8 +32,6 @@ struct service_control_op {
        SERVICE_CONTROL_OPS *ops;
 };
 
-#define SVCCTL_NUM_INTERNAL_SERVICES   4
-
 /* handle external services */
 extern SERVICE_CONTROL_OPS rcinit_svc_ops;
 
@@ -123,16 +121,12 @@ static struct service_control_op* find_service_by_name( const char *name )
 static NTSTATUS svcctl_access_check( SEC_DESC *sec_desc, NT_USER_TOKEN *token,
                                      uint32 access_desired, uint32 *access_granted )
 {
-       NTSTATUS result;
-
        if ( geteuid() == sec_initial_uid() ) {
                DEBUG(5,("svcctl_access_check: using root's token\n"));
                token = get_root_nt_token();
        }
 
-       se_access_check( sec_desc, token, access_desired, access_granted, &result );
-
-       return result;
+       return se_access_check( sec_desc, token, access_desired, access_granted);
 }
 
 /********************************************************************
@@ -143,7 +137,7 @@ static SEC_DESC* construct_scm_sd( TALLOC_CTX *ctx )
        SEC_ACE ace[2];
        size_t i = 0;
        SEC_DESC *sd;
-       SEC_ACL *acl;
+       SEC_ACL *theacl;
        size_t sd_size;
 
        /* basic access for Everyone */
@@ -159,31 +153,22 @@ static SEC_DESC* construct_scm_sd( TALLOC_CTX *ctx )
 
        /* create the security descriptor */
 
-       if ( !(acl = make_sec_acl(ctx, NT4_ACL_REVISION, i, ace)) )
+       if ( !(theacl = make_sec_acl(ctx, NT4_ACL_REVISION, i, ace)) )
                return NULL;
 
        if ( !(sd = make_sec_desc(ctx, SECURITY_DESCRIPTOR_REVISION_1,
                                  SEC_DESC_SELF_RELATIVE, NULL, NULL, NULL,
-                                 acl, &sd_size)) )
+                                 theacl, &sd_size)) )
                return NULL;
 
        return sd;
 }
 
-/******************************************************************
- free() function for REGISTRY_KEY
- *****************************************************************/
-
-static void free_service_handle_info(void *ptr)
-{
-       TALLOC_FREE( ptr );
-}
-
 /******************************************************************
  Find a registry key handle and return a SERVICE_INFO
  *****************************************************************/
 
-static SERVICE_INFO *find_service_info_by_hnd(pipes_struct *p, POLICY_HND *hnd)
+static SERVICE_INFO *find_service_info_by_hnd(pipes_struct *p, struct policy_handle *hnd)
 {
        SERVICE_INFO *service_info = NULL;
 
@@ -198,7 +183,7 @@ static SERVICE_INFO *find_service_info_by_hnd(pipes_struct *p, POLICY_HND *hnd)
 /******************************************************************
  *****************************************************************/
 
-static WERROR create_open_service_handle( pipes_struct *p, POLICY_HND *handle, uint32 type,
+static WERROR create_open_service_handle( pipes_struct *p, struct policy_handle *handle, uint32 type,
                                           const char *service, uint32 access_granted )
 {
        SERVICE_INFO *info = NULL;
@@ -248,19 +233,20 @@ static WERROR create_open_service_handle( pipes_struct *p, POLICY_HND *handle, u
 
        /* store the SERVICE_INFO and create an open handle */
 
-       if ( !create_policy_hnd( p, handle, free_service_handle_info, info ) ) {
+       if ( !create_policy_hnd( p, handle, info ) ) {
                result = WERR_ACCESS_DENIED;
                goto done;
        }
 
 done:
        if ( !W_ERROR_IS_OK(result) )
-               free_service_handle_info( info );
+               TALLOC_FREE(info);
 
        return result;
 }
 
 /********************************************************************
+ _svcctl_OpenSCManagerW
 ********************************************************************/
 
 WERROR _svcctl_OpenSCManagerW(pipes_struct *p,
@@ -276,7 +262,8 @@ WERROR _svcctl_OpenSCManagerW(pipes_struct *p,
                return WERR_NOMEM;
 
        se_map_generic( &r->in.access_mask, &scm_generic_map );
-       status = svcctl_access_check( sec_desc, p->pipe_user.nt_user_token, r->in.access_mask, &access_granted );
+       status = svcctl_access_check( sec_desc, p->server_info->ptok,
+                                     r->in.access_mask, &access_granted );
        if ( !NT_STATUS_IS_OK(status) )
                return ntstatus_to_werror( status );
 
@@ -313,7 +300,8 @@ WERROR _svcctl_OpenServiceW(pipes_struct *p,
                return WERR_NOMEM;
 
        se_map_generic( &r->in.access_mask, &svc_generic_map );
-       status = svcctl_access_check( sec_desc, p->pipe_user.nt_user_token, r->in.access_mask, &access_granted );
+       status = svcctl_access_check( sec_desc, p->server_info->ptok,
+                                     r->in.access_mask, &access_granted );
        if ( !NT_STATUS_IS_OK(status) )
                return ntstatus_to_werror( status );
 
@@ -321,9 +309,11 @@ WERROR _svcctl_OpenServiceW(pipes_struct *p,
 }
 
 /********************************************************************
+ _svcctl_CloseServiceHandle
 ********************************************************************/
 
-WERROR _svcctl_CloseServiceHandle(pipes_struct *p, struct svcctl_CloseServiceHandle *r)
+WERROR _svcctl_CloseServiceHandle(pipes_struct *p,
+                                 struct svcctl_CloseServiceHandle *r)
 {
        if ( !close_policy_hnd( p, r->in.handle ) )
                return  WERR_BADFID;
@@ -351,7 +341,8 @@ WERROR _svcctl_GetServiceDisplayNameW(pipes_struct *p,
 
        service = r->in.service_name;
 
-       display_name = svcctl_lookup_dispname(p->mem_ctx, service, p->pipe_user.nt_user_token );
+       display_name = svcctl_lookup_dispname(p->mem_ctx, service,
+                                             p->server_info->ptok);
        if (!display_name) {
                display_name = "";
        }
@@ -387,27 +378,27 @@ WERROR _svcctl_QueryServiceStatus(pipes_struct *p,
 /********************************************************************
 ********************************************************************/
 
-static int enumerate_status( TALLOC_CTX *ctx, ENUM_SERVICES_STATUS **status, NT_USER_TOKEN *token )
+static int enumerate_status( TALLOC_CTX *ctx, struct ENUM_SERVICE_STATUSW **status, NT_USER_TOKEN *token )
 {
        int num_services = 0;
        int i;
-       ENUM_SERVICES_STATUS *st;
+       struct ENUM_SERVICE_STATUSW *st;
        const char *display_name;
 
        /* just count */
        while ( svcctl_ops[num_services].name )
                num_services++;
 
-       if ( !(st = TALLOC_ARRAY( ctx, ENUM_SERVICES_STATUS, num_services )) ) {
+       if ( !(st = TALLOC_ARRAY( ctx, struct ENUM_SERVICE_STATUSW, num_services )) ) {
                DEBUG(0,("enumerate_status: talloc() failed!\n"));
                return -1;
        }
 
        for ( i=0; i<num_services; i++ ) {
-               init_unistr( &st[i].servicename, svcctl_ops[i].name );
+               st[i].service_name = talloc_strdup(st, svcctl_ops[i].name );
 
                display_name = svcctl_lookup_dispname(ctx, svcctl_ops[i].name, token );
-               init_unistr( &st[i].displayname, display_name ? display_name : "");
+               st[i].display_name = talloc_strdup(st, display_name ? display_name : "");
 
                svcctl_ops[i].ops->service_status( svcctl_ops[i].name, &st[i].status );
        }
@@ -418,17 +409,20 @@ static int enumerate_status( TALLOC_CTX *ctx, ENUM_SERVICES_STATUS **status, NT_
 }
 
 /********************************************************************
+ _svcctl_EnumServicesStatusW
 ********************************************************************/
 
-WERROR _svcctl_enum_services_status(pipes_struct *p, SVCCTL_Q_ENUM_SERVICES_STATUS *q_u, SVCCTL_R_ENUM_SERVICES_STATUS *r_u)
+WERROR _svcctl_EnumServicesStatusW(pipes_struct *p,
+                                  struct svcctl_EnumServicesStatusW *r)
 {
-       ENUM_SERVICES_STATUS *services = NULL;
+       struct ENUM_SERVICE_STATUSW *services = NULL;
        int num_services;
        int i = 0;
        size_t buffer_size = 0;
        WERROR result = WERR_OK;
-       SERVICE_INFO *info = find_service_info_by_hnd( p, &q_u->handle );
-       NT_USER_TOKEN *token = p->pipe_user.nt_user_token;
+       SERVICE_INFO *info = find_service_info_by_hnd( p, r->in.handle );
+       NT_USER_TOKEN *token = p->server_info->ptok;
+       DATA_BLOB blob = data_blob_null;
 
        /* perform access checks */
 
@@ -445,30 +439,41 @@ WERROR _svcctl_enum_services_status(pipes_struct *p, SVCCTL_Q_ENUM_SERVICES_STAT
        }
 
         for ( i=0; i<num_services; i++ ) {
-               buffer_size += svcctl_sizeof_enum_services_status(&services[i]);
+               buffer_size += ndr_size_ENUM_SERVICE_STATUSW(&services[i], NULL, 0);
        }
 
        buffer_size += buffer_size % 4;
 
-       if (buffer_size > q_u->buffer_size ) {
+       if (buffer_size > r->in.offered) {
                num_services = 0;
                result = WERR_MORE_DATA;
        }
 
-       rpcbuf_init(&r_u->buffer, q_u->buffer_size, p->mem_ctx);
-
        if ( W_ERROR_IS_OK(result) ) {
-               for ( i=0; i<num_services; i++ )
-                       svcctl_io_enum_services_status( "", &services[i], &r_u->buffer, 0 );
-       }
 
-       r_u->needed      = (buffer_size > q_u->buffer_size) ? buffer_size : q_u->buffer_size;
-       r_u->returned    = (uint32)num_services;
+               enum ndr_err_code ndr_err;
+               struct ndr_push *ndr;
 
-       if ( !(r_u->resume = TALLOC_P( p->mem_ctx, uint32 )) )
-               return WERR_NOMEM;
+               ndr = ndr_push_init_ctx(p->mem_ctx, NULL);
+               if (ndr == NULL) {
+                       return WERR_INVALID_PARAM;
+               }
 
-       *r_u->resume = 0x0;
+               ndr_err = ndr_push_ENUM_SERVICE_STATUSW_array(
+                       ndr, num_services, services);
+               if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+                       return ntstatus_to_werror(ndr_map_error2ntstatus(ndr_err));
+               }
+
+               blob = ndr_push_blob(ndr);
+               if (blob.length >= r->in.offered) {
+                       memcpy(r->out.service, blob.data, r->in.offered);
+               }
+       }
+
+       *r->out.needed                  = (buffer_size > r->in.offered) ? buffer_size : r->in.offered;
+       *r->out.services_returned       = (uint32)num_services;
+       *r->out.resume_handle           = 0x0;
 
        return result;
 }
@@ -521,11 +526,9 @@ WERROR _svcctl_ControlService(pipes_struct *p,
 
                return info->ops->service_status( info->name,
                                                  r->out.service_status );
+       default:
+               return WERR_ACCESS_DENIED;
        }
-
-       /* default control action */
-
-       return WERR_ACCESS_DENIED;
 }
 
 /********************************************************************
@@ -549,7 +552,7 @@ WERROR _svcctl_EnumDependentServicesW(pipes_struct *p,
           incoming buffer size (even in the case of failure */
        /* this is done in the autogenerated server already - gd */
 
-       *r->out.bytes_needed = r->in.buf_size;
+       *r->out.needed = r->in.offered;
 
        /* no dependent services...basically a stub function */
        *r->out.services_returned = 0;
@@ -558,11 +561,13 @@ WERROR _svcctl_EnumDependentServicesW(pipes_struct *p,
 }
 
 /********************************************************************
+ _svcctl_QueryServiceStatusEx
 ********************************************************************/
 
-WERROR _svcctl_query_service_status_ex( pipes_struct *p, SVCCTL_Q_QUERY_SERVICE_STATUSEX *q_u, SVCCTL_R_QUERY_SERVICE_STATUSEX *r_u )
+WERROR _svcctl_QueryServiceStatusEx(pipes_struct *p,
+                                   struct svcctl_QueryServiceStatusEx *r)
 {
-       SERVICE_INFO *info = find_service_info_by_hnd( p, &q_u->handle );
+       SERVICE_INFO *info = find_service_info_by_hnd( p, r->in.handle );
        uint32 buffer_size;
 
        /* perform access checks */
@@ -575,22 +580,29 @@ WERROR _svcctl_query_service_status_ex( pipes_struct *p, SVCCTL_Q_QUERY_SERVICE_
 
        /* we have to set the outgoing buffer size to the same as the
           incoming buffer size (even in the case of failure) */
+       *r->out.needed = r->in.offered;
 
-       rpcbuf_init( &r_u->buffer, q_u->buffer_size, p->mem_ctx );
-       r_u->needed = q_u->buffer_size;
-
-       switch ( q_u->level ) {
+       switch ( r->in.info_level ) {
                case SVC_STATUS_PROCESS_INFO:
                {
-                       SERVICE_STATUS_PROCESS svc_stat_proc;
+                       struct SERVICE_STATUS_PROCESS svc_stat_proc;
+                       enum ndr_err_code ndr_err;
+                       DATA_BLOB blob;
 
                        /* Get the status of the service.. */
                        info->ops->service_status( info->name, &svc_stat_proc.status );
                        svc_stat_proc.process_id     = sys_getpid();
                        svc_stat_proc.service_flags  = 0x0;
 
-                       svcctl_io_service_status_process( "", &svc_stat_proc, &r_u->buffer, 0 );
-                       buffer_size = sizeof(SERVICE_STATUS_PROCESS);
+                       ndr_err = ndr_push_struct_blob(&blob, p->mem_ctx, NULL,
+                                                      &svc_stat_proc,
+                                                      (ndr_push_flags_fn_t)ndr_push_SERVICE_STATUS_PROCESS);
+                       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+                               return WERR_INVALID_PARAM;
+                       }
+
+                       r->out.buffer = blob.data;
+                       buffer_size = sizeof(struct SERVICE_STATUS_PROCESS);
                        break;
                }
 
@@ -600,10 +612,11 @@ WERROR _svcctl_query_service_status_ex( pipes_struct *p, SVCCTL_Q_QUERY_SERVICE_
 
 
         buffer_size += buffer_size % 4;
-       r_u->needed = (buffer_size > q_u->buffer_size) ? buffer_size : q_u->buffer_size;
+       *r->out.needed = (buffer_size > r->in.offered) ? buffer_size : r->in.offered;
 
-        if (buffer_size > q_u->buffer_size )
-                return WERR_MORE_DATA;
+        if (buffer_size > r->in.offered ) {
+                return WERR_INSUFFICIENT_BUFFER;
+       }
 
        return WERR_OK;
 }
@@ -642,7 +655,7 @@ static WERROR fill_svc_config( TALLOC_CTX *ctx, const char *name,
        /* loadordergroup and dependencies are empty */
 
        config->tag_id           = 0x00000000;                  /* unassigned loadorder group */
-       config->service_type     = SVCCTL_WIN32_OWN_PROC;
+       config->service_type     = SERVICE_TYPE_WIN32_OWN_PROCESS;
        config->error_control    = SVCCTL_SVC_ERROR_NORMAL;
 
        /* set the start type.  NetLogon and WINS are disabled to prevent
@@ -684,17 +697,18 @@ WERROR _svcctl_QueryServiceConfigW(pipes_struct *p,
        /* we have to set the outgoing buffer size to the same as the
           incoming buffer size (even in the case of failure */
 
-       *r->out.bytes_needed = r->in.buf_size;
+       *r->out.needed = r->in.offered;
 
-       wresult = fill_svc_config( p->mem_ctx, info->name, r->out.query, p->pipe_user.nt_user_token );
+       wresult = fill_svc_config( p->mem_ctx, info->name, r->out.query,
+                                  p->server_info->ptok);
        if ( !W_ERROR_IS_OK(wresult) )
                return wresult;
 
-       buffer_size = ndr_size_QUERY_SERVICE_CONFIG(r->out.query, 0);
-       *r->out.bytes_needed = (buffer_size > r->in.buf_size) ? buffer_size : r->in.buf_size;
+       buffer_size = ndr_size_QUERY_SERVICE_CONFIG(r->out.query, NULL, 0);
+       *r->out.needed = (buffer_size > r->in.offered) ? buffer_size : r->in.offered;
 
-        if (buffer_size > r->in.buf_size ) {
-               TALLOC_FREE(r->out.query);
+        if (buffer_size > r->in.offered ) {
+               ZERO_STRUCTP(r->out.query);
                 return WERR_INSUFFICIENT_BUFFER;
        }
 
@@ -702,11 +716,13 @@ WERROR _svcctl_QueryServiceConfigW(pipes_struct *p,
 }
 
 /********************************************************************
+ _svcctl_QueryServiceConfig2W
 ********************************************************************/
 
-WERROR _svcctl_query_service_config2( pipes_struct *p, SVCCTL_Q_QUERY_SERVICE_CONFIG2 *q_u, SVCCTL_R_QUERY_SERVICE_CONFIG2 *r_u )
+WERROR _svcctl_QueryServiceConfig2W(pipes_struct *p,
+                                   struct svcctl_QueryServiceConfig2W *r)
 {
-       SERVICE_INFO *info = find_service_info_by_hnd( p, &q_u->handle );
+       SERVICE_INFO *info = find_service_info_by_hnd( p, r->in.handle );
        uint32 buffer_size;
 
        /* perform access checks */
@@ -719,36 +735,53 @@ WERROR _svcctl_query_service_config2( pipes_struct *p, SVCCTL_Q_QUERY_SERVICE_CO
 
        /* we have to set the outgoing buffer size to the same as the
           incoming buffer size (even in the case of failure */
+       *r->out.needed = r->in.offered;
 
-       rpcbuf_init( &r_u->buffer, q_u->buffer_size, p->mem_ctx );
-       r_u->needed = q_u->buffer_size;
-
-       switch ( q_u->level ) {
+       switch ( r->in.info_level ) {
        case SERVICE_CONFIG_DESCRIPTION:
                {
-                       SERVICE_DESCRIPTION desc_buf;
+                       struct SERVICE_DESCRIPTION desc_buf;
                        const char *description;
+                       enum ndr_err_code ndr_err;
+                       DATA_BLOB blob;
 
-                       description = svcctl_lookup_description(p->mem_ctx, info->name, p->pipe_user.nt_user_token );
+                       description = svcctl_lookup_description(
+                               p->mem_ctx, info->name, p->server_info->ptok);
 
-                       ZERO_STRUCTP( &desc_buf );
+                       desc_buf.description = description;
 
-                       init_service_description_buffer( &desc_buf, description ? description : "");
-                       svcctl_io_service_description( "", &desc_buf, &r_u->buffer, 0 );
-                       buffer_size = svcctl_sizeof_service_description( &desc_buf );
+                       ndr_err = ndr_push_struct_blob(&blob, p->mem_ctx, NULL,
+                                                      &desc_buf,
+                                                      (ndr_push_flags_fn_t)ndr_push_SERVICE_DESCRIPTION);
+                       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+                               return WERR_INVALID_PARAM;
+                       }
+
+                       buffer_size = ndr_size_SERVICE_DESCRIPTION(&desc_buf, NULL, 0);
+                       r->out.buffer = blob.data;
 
                        break;
                }
                break;
        case SERVICE_CONFIG_FAILURE_ACTIONS:
                {
-                       SERVICE_FAILURE_ACTIONS actions;
+                       struct SERVICE_FAILURE_ACTIONS actions;
+                       enum ndr_err_code ndr_err;
+                       DATA_BLOB blob;
 
                        /* nothing to say...just service the request */
 
-                       ZERO_STRUCTP( &actions );
-                       svcctl_io_service_fa( "", &actions, &r_u->buffer, 0 );
-                       buffer_size = svcctl_sizeof_service_fa( &actions );
+                       ZERO_STRUCT( actions );
+
+                       ndr_err = ndr_push_struct_blob(&blob, p->mem_ctx, NULL,
+                                                      &actions,
+                                                      (ndr_push_flags_fn_t)ndr_push_SERVICE_FAILURE_ACTIONS);
+                       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+                               return WERR_INVALID_PARAM;
+                       }
+
+                       buffer_size = ndr_size_SERVICE_FAILURE_ACTIONS(&actions, NULL, 0);
+                       r->out.buffer = blob.data;
 
                        break;
                }
@@ -759,9 +792,9 @@ WERROR _svcctl_query_service_config2( pipes_struct *p, SVCCTL_Q_QUERY_SERVICE_CO
        }
 
        buffer_size += buffer_size % 4;
-       r_u->needed = (buffer_size > q_u->buffer_size) ? buffer_size : q_u->buffer_size;
+       *r->out.needed = (buffer_size > r->in.offered) ? buffer_size : r->in.offered;
 
-        if (buffer_size > q_u->buffer_size )
+        if (buffer_size > r->in.offered)
                 return WERR_INSUFFICIENT_BUFFER;
 
        return WERR_OK;
@@ -839,7 +872,7 @@ WERROR _svcctl_QueryServiceObjectSecurity(pipes_struct *p,
        if ( !(sec_desc = svcctl_get_secdesc( p->mem_ctx, info->name, get_root_nt_token() )) )
                 return WERR_NOMEM;
 
-       *r->out.needed = ndr_size_security_descriptor( sec_desc, 0 );
+       *r->out.needed = ndr_size_security_descriptor( sec_desc, NULL, 0 );
 
        if ( *r->out.needed > r->in.buffer_size ) {
                ZERO_STRUCTP( &r->out.buffer );
@@ -909,7 +942,8 @@ WERROR _svcctl_SetServiceObjectSecurity(pipes_struct *p,
 
        /* store the new SD */
 
-       if ( !svcctl_set_secdesc( p->mem_ctx, info->name, sec_desc, p->pipe_user.nt_user_token ) )
+       if ( !svcctl_set_secdesc( p->mem_ctx, info->name, sec_desc,
+                                 p->server_info->ptok) )
                return WERR_ACCESS_DENIED;
 
        return WERR_OK;
@@ -952,12 +986,6 @@ WERROR _svcctl_CreateServiceW(pipes_struct *p, struct svcctl_CreateServiceW *r)
        return WERR_NOT_SUPPORTED;
 }
 
-WERROR _svcctl_EnumServicesStatusW(pipes_struct *p, struct svcctl_EnumServicesStatusW *r)
-{
-       p->rng_fault_state = True;
-       return WERR_NOT_SUPPORTED;
-}
-
 WERROR _svcctl_QueryServiceLockStatusW(pipes_struct *p, struct svcctl_QueryServiceLockStatusW *r)
 {
        p->rng_fault_state = True;
@@ -1072,18 +1100,6 @@ WERROR _svcctl_QueryServiceConfig2A(pipes_struct *p, struct svcctl_QueryServiceC
        return WERR_NOT_SUPPORTED;
 }
 
-WERROR _svcctl_QueryServiceConfig2W(pipes_struct *p, struct svcctl_QueryServiceConfig2W *r)
-{
-       p->rng_fault_state = True;
-       return WERR_NOT_SUPPORTED;
-}
-
-WERROR _svcctl_QueryServiceStatusEx(pipes_struct *p, struct svcctl_QueryServiceStatusEx *r)
-{
-       p->rng_fault_state = True;
-       return WERR_NOT_SUPPORTED;
-}
-
 WERROR _EnumServicesStatusExA(pipes_struct *p, struct EnumServicesStatusExA *r)
 {
        p->rng_fault_state = True;