r4777: added a smb_composite_sesssetup() async composite function. This
[gd/samba-autobuild/.git] / source4 / libcli / raw / clitree.c
index e0072a31b1ece0f63b4dbc7ec241ef2b2b24572d..c6b3fa5ad9e951946f5f9b94dac46e4b9ee1230b 100644 (file)
@@ -1,7 +1,9 @@
 /* 
    Unix SMB/CIFS implementation.
+
    SMB client tree context management functions
-   Copyright (C) Andrew Tridgell 1994-1998
+
+   Copyright (C) Andrew Tridgell 1994-2005
    Copyright (C) James Myers 2003 <myersjj@samba.org>
    
    This program is free software; you can redistribute it and/or modify
 */
 
 #include "includes.h"
+#include "libcli/raw/libcliraw.h"
+#include "libcli/composite/composite.h"
 
 #define SETUP_REQUEST_TREE(cmd, wct, buflen) do { \
        req = smbcli_request_setup(tree, cmd, wct, buflen); \
        if (!req) return NULL; \
 } while (0)
 
-/*
-  destroy a smbcli_tree
-*/
-static int tree_destructor(void *ptr)
-{
-       struct smbcli_tree *tree = ptr;
-       talloc_free(tree->session);
-       return 0;
-}
-
 /****************************************************************************
  Initialize the tree context
 ****************************************************************************/
@@ -43,14 +37,12 @@ struct smbcli_tree *smbcli_tree_init(struct smbcli_session *session)
 {
        struct smbcli_tree *tree;
 
-       tree = talloc_p(session, struct smbcli_tree);
+       tree = talloc_zero(session, struct smbcli_tree);
        if (!tree) {
                return NULL;
        }
 
-       ZERO_STRUCTP(tree);
-       tree->session = session;
-       talloc_set_destructor(tree, tree_destructor);
+       tree->session = talloc_reference(tree, session);
 
        return tree;
 }
@@ -58,7 +50,8 @@ struct smbcli_tree *smbcli_tree_init(struct smbcli_session *session)
 /****************************************************************************
  Send a tconX (async send)
 ****************************************************************************/
-struct smbcli_request *smb_tree_connect_send(struct smbcli_tree *tree, union smb_tcon *parms)
+struct smbcli_request *smb_tree_connect_send(struct smbcli_tree *tree, 
+                                            union smb_tcon *parms)
 {
        struct smbcli_request *req = NULL;
 
@@ -93,9 +86,10 @@ struct smbcli_request *smb_tree_connect_send(struct smbcli_tree *tree, union smb
 /****************************************************************************
  Send a tconX (async recv)
 ****************************************************************************/
-NTSTATUS smb_tree_connect_recv(struct smbcli_request *req, TALLOC_CTX *mem_ctx, union smb_tcon *parms)
+NTSTATUS smb_tree_connect_recv(struct smbcli_request *req, TALLOC_CTX *mem_ctx, 
+                              union smb_tcon *parms)
 {
-       char *p;
+       uint8_t *p;
 
        if (!smbcli_request_receive(req) ||
            smbcli_request_is_error(req)) {
@@ -106,12 +100,12 @@ NTSTATUS smb_tree_connect_recv(struct smbcli_request *req, TALLOC_CTX *mem_ctx,
        case RAW_TCON_TCON:
                SMBCLI_CHECK_WCT(req, 2);
                parms->tcon.out.max_xmit = SVAL(req->in.vwv, VWV(0));
-               parms->tcon.out.cnum = SVAL(req->in.vwv, VWV(1));
+               parms->tcon.out.tid = SVAL(req->in.vwv, VWV(1));
                break;
 
        case RAW_TCON_TCONX:
                ZERO_STRUCT(parms->tconx.out);
-               parms->tconx.out.cnum = SVAL(req->in.hdr, HDR_TID);
+               parms->tconx.out.tid = SVAL(req->in.hdr, HDR_TID);
                if (req->in.wct >= 4) {
                        parms->tconx.out.options = SVAL(req->in.vwv, VWV(3));
                }
@@ -134,7 +128,8 @@ failed:
 /****************************************************************************
  Send a tconX (sync interface)
 ****************************************************************************/
-NTSTATUS smb_tree_connect(struct smbcli_tree *tree, TALLOC_CTX *mem_ctx, union smb_tcon *parms)
+NTSTATUS smb_tree_connect(struct smbcli_tree *tree, TALLOC_CTX *mem_ctx, 
+                         union smb_tcon *parms)
 {
        struct smbcli_request *req = smb_tree_connect_send(tree, parms);
        return smb_tree_connect_recv(req, mem_ctx, parms);
@@ -159,142 +154,33 @@ NTSTATUS smb_tree_disconnect(struct smbcli_tree *tree)
 
 
 /*
-  a convenient function to establish a smbcli_tree from scratch, using reasonable default
-  parameters
+  a convenient function to establish a smbcli_tree from scratch
 */
-NTSTATUS smbcli_tree_full_connection(struct smbcli_tree **ret_tree, 
+NTSTATUS smbcli_tree_full_connection(TALLOC_CTX *parent_ctx,
+                                    struct smbcli_tree **ret_tree, 
                                     const char *my_name, 
                                     const char *dest_host, int port,
                                     const char *service, const char *service_type,
                                     const char *user, const char *domain, 
                                     const char *password)
 {
-       struct smbcli_socket *sock;
-       struct smbcli_transport *transport;
-       struct smbcli_session *session;
-       struct smbcli_tree *tree;
+       struct smb_composite_connect io;
        NTSTATUS status;
-       struct nmb_name calling;
-       struct nmb_name called;
-       union smb_sesssetup setup;
-       union smb_tcon tcon;
-       TALLOC_CTX *mem_ctx;
-       char *in_path = NULL;
-
-       *ret_tree = NULL;
-
-       sock = smbcli_sock_init();
-       if (!sock) {
-               return NT_STATUS_NO_MEMORY;
-       }
 
-       talloc_set_name_const(sock, "smbcli_tree_full_connection");
-
-       /* open a TCP socket to the server */
-       if (!smbcli_sock_connect_byname(sock, dest_host, port)) {
-               DEBUG(2,("Failed to establish socket connection - %s\n", strerror(errno)));
-               return NT_STATUS_UNSUCCESSFUL;
-       }
-
-       transport = smbcli_transport_init(sock);
-       if (!transport) {
-               talloc_free(sock);
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       /* send a NBT session request, if applicable */
-       make_nmb_name(&calling, my_name, 0x0);
-       choose_called_name(&called,  dest_host, 0x20);
-
-       if (!smbcli_transport_connect(transport, &calling, &called)) {
-               talloc_free(transport);
-               return NT_STATUS_UNSUCCESSFUL;
-       }
-
-
-       /* negotiate protocol options with the server */
-       status = smb_raw_negotiate(transport);
-       if (!NT_STATUS_IS_OK(status)) {
-               talloc_free(transport);
-               return status;
-       }
-
-       session = smbcli_session_init(transport);
-       if (!session) {
-               talloc_free(transport);
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       /* prepare a session setup to establish a security context */
-       setup.generic.level = RAW_SESSSETUP_GENERIC;
-       setup.generic.in.sesskey = transport->negotiate.sesskey;
-       setup.generic.in.capabilities = transport->negotiate.capabilities;
-       if (!user || !user[0]) {
-               setup.generic.in.password = NULL;
-               setup.generic.in.user = "";
-               setup.generic.in.domain = "";
-               setup.generic.in.capabilities &= ~CAP_EXTENDED_SECURITY;
-       } else {
-               setup.generic.in.password = password;
-               setup.generic.in.user = user;
-               setup.generic.in.domain = domain;
-       }
-
-       mem_ctx = talloc_init("tcon");
-       if (!mem_ctx) {
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       status = smb_raw_session_setup(session, mem_ctx, &setup);
-       if (!NT_STATUS_IS_OK(status)) {
-               talloc_free(session);
-               talloc_free(mem_ctx);
-               return status;
-       }
-
-       session->vuid = setup.generic.out.vuid;
-
-       tree = smbcli_tree_init(session);
-       if (!tree) {
-               talloc_free(session);
-               talloc_free(mem_ctx);
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       /* connect to a share using a tree connect */
-       tcon.generic.level = RAW_TCON_TCONX;
-       tcon.tconx.in.flags = 0;
-       tcon.tconx.in.password = data_blob(NULL, 0);    
-       asprintf(&in_path, "\\\\%s\\%s", dest_host, service);
-       tcon.tconx.in.path = in_path;
-       if (!service_type) {
-               if (strequal(service, "IPC$"))
-                       service_type = "IPC";
-               else
-                       service_type = "?????";
-       }
-       tcon.tconx.in.device = service_type;
+       io.in.dest_host = dest_host;
+       io.in.port = port;
+       io.in.called_name = dest_host;
+       io.in.calling_name = my_name;
+       io.in.service = service;
+       io.in.service_type = service_type;
+       io.in.user = user;
+       io.in.domain = domain;
+       io.in.password = password;
        
-       status = smb_tree_connect(tree, mem_ctx, &tcon);
-
-       SAFE_FREE(in_path);
-
-       if (!NT_STATUS_IS_OK(status)) {
-               talloc_free(tree);
-               talloc_free(mem_ctx);
-               return status;
-       }
-
-       tree->tid = tcon.tconx.out.cnum;
-       if (tcon.tconx.out.dev_type) {
-               tree->device = talloc_strdup(tree, tcon.tconx.out.dev_type);
+       status = smb_composite_connect(&io, parent_ctx);
+       if (NT_STATUS_IS_OK(status)) {
+               *ret_tree = io.out.tree;
        }
-       if (tcon.tconx.out.fs_type) {
-               tree->fs_type = talloc_strdup(tree, tcon.tconx.out.fs_type);
-       }
-
-       talloc_free(mem_ctx);
 
-       *ret_tree = tree;
-       return NT_STATUS_OK;
+       return status;
 }