Refactoring: Move stuff around for creating a pipe
authorVolker Lendecke <vl@samba.org>
Sun, 20 Apr 2008 10:19:27 +0000 (12:19 +0200)
committerVolker Lendecke <vl@samba.org>
Sun, 20 Apr 2008 12:08:10 +0000 (14:08 +0200)
source/include/client.h
source/rpc_client/cli_pipe.c

index d9f4727203cafd3e07329438b3e22b4fa519b142..5cfc9a6f9203ebd3e38950053426aadf447f70ad 100644 (file)
@@ -68,11 +68,11 @@ struct rpc_pipe_client {
        const struct ndr_syntax_id *abstract_syntax;
        const struct ndr_syntax_id *transfer_syntax;
 
-       const char *desthost;
-       const char *srv_name_slash;
+       char *desthost;
+       char *srv_name_slash;
 
-       const char *domain;
-       const char *user_name;
+       char *domain;
+       char *user_name;
        struct pwd_info pwd;
 
        uint16 max_xmit_frag;
index 2979d168c5d2b970861362908d65eb273dbc3160..3e03887bfdb00c842aa2f461dbab7bac2271e392 100644 (file)
@@ -2164,51 +2164,34 @@ static struct rpc_pipe_client *cli_rpc_pipe_open(struct cli_state *cli, int pipe
                return NULL;
        }
 
-       if ( pipe_idx >= PI_MAX_PIPES ) {
-               DEBUG(0, ("cli_rpc_pipe_open: Programmer error!  Invalid pipe "
-                         "index [%d]\n", pipe_idx));
-               *perr = NT_STATUS_INVALID_PARAMETER;
-               return NULL;
-       }
-
        /* The pipe name index must fall within our array */
        SMB_ASSERT((pipe_idx >= 0) && (pipe_idx < PI_MAX_PIPES));
 
        result = TALLOC_ZERO_P(NULL, struct rpc_pipe_client);
        if (result == NULL) {
+               *perr = NT_STATUS_NO_MEMORY;
                return NULL;
        }
 
        result->pipe_name = cli_get_pipe_name(pipe_idx);
 
-       fnum = cli_nt_create(cli, result->pipe_name, DESIRED_ACCESS_PIPE);
-
-       if (fnum == -1) {
-               DEBUG(1,("cli_rpc_pipe_open: cli_nt_create failed on pipe %s "
-                        "to machine %s.  Error was %s\n",
-                        result->pipe_name, cli->desthost,
-                        cli_errstr(cli)));
-               *perr = cli_get_nt_error(cli);
-               talloc_destroy(result);
-               return NULL;
-       }
-
-       result->fnum = fnum;
        result->cli = cli;
        result->abstract_syntax = pipe_names[pipe_idx].abstr_syntax;
        result->transfer_syntax = pipe_names[pipe_idx].trans_syntax;
        result->auth.auth_type = PIPE_AUTH_TYPE_NONE;
        result->auth.auth_level = PIPE_AUTH_LEVEL_NONE;
 
+       result->domain = talloc_strdup(result, cli->domain);
+       result->user_name = talloc_strdup(result, cli->user_name);
        result->desthost = talloc_strdup(result, cli->desthost);
-       if (result->desthost == NULL) {
-               TALLOC_FREE(result);
-               return NULL;
-       }
-
        result->srv_name_slash = talloc_asprintf_strupper_m(
                result, "\\\\%s", result->desthost);
-       if (result->srv_name_slash == NULL) {
+
+       if ((result->domain == NULL)
+           || (result->user_name == NULL)
+           || (result->desthost == NULL)
+           || (result->srv_name_slash == NULL)) {
+               *perr = NT_STATUS_NO_MEMORY;
                TALLOC_FREE(result);
                return NULL;
        }
@@ -2217,11 +2200,25 @@ static struct rpc_pipe_client *cli_rpc_pipe_open(struct cli_state *cli, int pipe
                /* Set up a netlogon credential chain for a netlogon pipe. */
                result->dc = TALLOC_ZERO_P(result, struct dcinfo);
                if (result->dc == NULL) {
-                       talloc_destroy(result);
+                       *perr = NT_STATUS_NO_MEMORY;
+                       TALLOC_FREE(result);
                        return NULL;
                }
        }
 
+       fnum = cli_nt_create(cli, result->pipe_name, DESIRED_ACCESS_PIPE);
+       if (fnum == -1) {
+               DEBUG(1,("cli_rpc_pipe_open: cli_nt_create failed on pipe %s "
+                        "to machine %s.  Error was %s\n",
+                        result->pipe_name, cli->desthost,
+                        cli_errstr(cli)));
+               *perr = cli_get_nt_error(cli);
+               talloc_destroy(result);
+               return NULL;
+       }
+
+       result->fnum = fnum;
+
        DLIST_ADD(cli->pipe_list, result);
        *perr = NT_STATUS_OK;
 
@@ -2241,19 +2238,10 @@ struct rpc_pipe_client *cli_rpc_pipe_open_noauth(struct cli_state *cli, int pipe
                return NULL;
        }
 
-       result->domain = talloc_strdup(result, cli->domain);
-       result->user_name = talloc_strdup(result, cli->user_name);
-
-       if ((result->domain == NULL) || (result->user_name == NULL)) {
-               *perr = NT_STATUS_NO_MEMORY;
-               cli_rpc_pipe_close(result);
-               return NULL;
-       }
-
        *perr = rpc_pipe_bind(result, PIPE_AUTH_TYPE_NONE, PIPE_AUTH_LEVEL_NONE);
        if (!NT_STATUS_IS_OK(*perr)) {
                int lvl = 0;
-               if (pipe_idx == PI_DSSETUP) {
+               if (rpccli_is_pipe_idx(result, PI_DSSETUP)) {
                        /* non AD domains just don't have this pipe, avoid
                         * level 0 statement in that case - gd */
                        lvl = 3;
@@ -2305,6 +2293,9 @@ static struct rpc_pipe_client *cli_rpc_pipe_open_ntlmssp_internal(struct cli_sta
        
        result->auth.cli_auth_data_free_func = cli_ntlmssp_auth_free;
 
+       TALLOC_FREE(result->domain);
+       TALLOC_FREE(result->user_name);
+
        result->domain = talloc_strdup(result, domain);
        result->user_name = talloc_strdup(result, username);
 
@@ -2525,7 +2516,14 @@ struct rpc_pipe_client *cli_rpc_pipe_open_schannel_with_key(struct cli_state *cl
                return NULL;
        }
 
-       result->domain = domain;
+       TALLOC_FREE(result->domain);
+       result->domain = talloc_strdup(result, domain);
+       if (result->domain == NULL) {
+               cli_rpc_pipe_close(result);
+               *perr = NT_STATUS_NO_MEMORY;
+               return NULL;
+       }
+
        memcpy(result->auth.a_u.schannel_auth->sess_key, pdc->sess_key, 16);
 
        *perr = rpc_pipe_bind(result, PIPE_AUTH_TYPE_SCHANNEL, auth_level);