lib: Remove global xfile.h includes
[samba.git] / source3 / modules / vfs_expand_msdfs.c
index 7176c11e4abd11b81cc64d31a1689ea764930cd6..40c57c2d1079796acb7a5a6fb6318100310a489f 100644 (file)
@@ -5,7 +5,7 @@
  *
  * 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
+ * the Free Software Foundation; either version 3 of the License, or
  * (at your option) any later version.
  *  
  * This program is distributed in the hope that it will be useful,
  * 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.
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
  */
 
 #include "includes.h"
+#include "system/filesys.h"
+#include "smbd/smbd.h"
+#include "../librpc/gen_ndr/ndr_netlogon.h"
+#include "smbd/globals.h"
+#include "auth.h"
+#include "../lib/tsocket/tsocket.h"
+#include "lib/util/xfile.h"
 
 #undef DBGC_CLASS
 #define DBGC_CLASS DBGC_VFS
 
-extern userdom_struct current_user_info;
-
 /**********************************************************
   Under mapfile we expect a table of the following format:
 
@@ -38,19 +42,20 @@ extern userdom_struct current_user_info;
   This is to redirect a DFS client to a host close to it.
 ***********************************************************/
 
-static BOOL read_target_host(const char *mapfile, pstring targethost)
+static char *read_target_host(TALLOC_CTX *ctx, const char *mapfile,
+                             const char *clientaddr)
 {
        XFILE *f;
-       pstring buf;
+       char buf[1024];
        char *space = buf;
-       BOOL found = False;
-       
+       bool found = false;
+
        f = x_fopen(mapfile, O_RDONLY, 0);
 
        if (f == NULL) {
                DEBUG(0,("can't open IP map %s. Error %s\n",
                         mapfile, strerror(errno) ));
-               return False;
+               return NULL;
        }
 
        DEBUG(10, ("Scanning mapfile [%s]\n", mapfile));
@@ -71,24 +76,24 @@ static BOOL read_target_host(const char *mapfile, pstring targethost)
 
                *space = '\0';
 
-               if (strncmp(client_addr(), buf, strlen(buf)) == 0) {
-                       found = True;
+               if (strncmp(clientaddr, buf, strlen(buf)) == 0) {
+                       found = true;
                        break;
                }
        }
 
        x_fclose(f);
 
-       if (!found)
-               return False;
+       if (!found) {
+               return NULL;
+       }
 
        space += 1;
 
        while (isspace(*space))
                space += 1;
 
-       pstrcpy(targethost, space);
-       return True;
+       return talloc_strdup(ctx, space);
 }
 
 /**********************************************************
@@ -104,94 +109,128 @@ static BOOL read_target_host(const char *mapfile, pstring targethost)
 
 ***********************************************************/
 
-static BOOL expand_msdfs_target(connection_struct* conn, pstring target)
+static char *expand_msdfs_target(TALLOC_CTX *ctx,
+                               connection_struct *conn,
+                               char *target)
 {
-       pstring mapfilename;
+       char *mapfilename = NULL;
        char *filename_start = strchr_m(target, '@');
-       char *filename_end;
-       int filename_len;
-       pstring targethost;
-       pstring new_target;
+       char *filename_end = NULL;
+       int filename_len = 0;
+       char *targethost = NULL;
+       char *new_target = NULL;
+       char *raddr;
 
        if (filename_start == NULL) {
                DEBUG(10, ("No filename start in %s\n", target));
-               return False;
+               return NULL;
        }
 
        filename_end = strchr_m(filename_start+1, '@');
 
        if (filename_end == NULL) {
                DEBUG(10, ("No filename end in %s\n", target));
-               return False;
+               return NULL;
        }
 
        filename_len = PTR_DIFF(filename_end, filename_start+1);
-       pstrcpy(mapfilename, filename_start+1);
+       mapfilename = talloc_strdup(ctx, filename_start+1);
+       if (!mapfilename) {
+               return NULL;
+       }
        mapfilename[filename_len] = '\0';
 
        DEBUG(10, ("Expanding from table [%s]\n", mapfilename));
 
-       if (!read_target_host(mapfilename, targethost)) {
+       raddr = tsocket_address_inet_addr_string(conn->sconn->remote_address,
+                                                ctx);
+       if (raddr == NULL) {
+               return NULL;
+       }
+
+       targethost = read_target_host(
+               ctx, raddr, mapfilename);
+       if (targethost == NULL) {
                DEBUG(1, ("Could not expand target host from file %s\n",
                          mapfilename));
-               return False;
+               return NULL;
        }
 
-       standard_sub_advanced(lp_servicename(SNUM(conn)), conn->user,
-                             conn->connectpath, conn->gid,
-                             get_current_username(),
-                             current_user_info.domain,
-                             mapfilename, sizeof(mapfilename));
+       targethost = talloc_sub_advanced(ctx,
+                               lp_servicename(talloc_tos(), SNUM(conn)),
+                               conn->session_info->unix_info->unix_name,
+                               conn->connectpath,
+                               conn->session_info->unix_token->gid,
+                               conn->session_info->unix_info->sanitized_username,
+                               conn->session_info->info->domain_name,
+                               targethost);
 
        DEBUG(10, ("Expanded targethost to %s\n", targethost));
 
+       /* Replace the part between '@...@' */
        *filename_start = '\0';
-       pstrcpy(new_target, target);
-       pstrcat(new_target, targethost);
-       pstrcat(new_target, filename_end+1);
+       new_target = talloc_asprintf(ctx,
+                               "%s%s%s",
+                               target,
+                               targethost,
+                               filename_end+1);
+       if (!new_target) {
+               return NULL;
+       }
 
        DEBUG(10, ("New DFS target: %s\n", new_target));
-       pstrcpy(target, new_target);
-       return True;
+       return new_target;
 }
 
 static int expand_msdfs_readlink(struct vfs_handle_struct *handle,
                                 const char *path, char *buf, size_t bufsiz)
 {
-       pstring target;
+       TALLOC_CTX *ctx = talloc_tos();
        int result;
+       char *target = talloc_array(ctx, char, PATH_MAX+1);
+       size_t len;
+
+       if (!target) {
+               errno = ENOMEM;
+               return -1;
+       }
+       if (bufsiz == 0) {
+               errno = EINVAL;
+               return -1;
+       }
 
        result = SMB_VFS_NEXT_READLINK(handle, path, target,
-                                      sizeof(target));
+                                      PATH_MAX);
 
-       if (result < 0)
+       if (result <= 0)
                return result;
 
        target[result] = '\0';
 
-       if ((strncmp(target, "msdfs:", strlen("msdfs:")) == 0) &&
+       if ((strncmp(target, "msdfs:", 6) == 0) &&
            (strchr_m(target, '@') != NULL)) {
-               if (!expand_msdfs_target(handle->conn, target)) {
+               target = expand_msdfs_target(ctx, handle->conn, target);
+               if (!target) {
                        errno = ENOENT;
                        return -1;
                }
        }
 
-       safe_strcpy(buf, target, bufsiz-1);
-       return strlen(buf);
-}
+       len = MIN(bufsiz, strlen(target));
+
+       memcpy(buf, target, len);
 
-/* VFS operations structure */
+       TALLOC_FREE(target);
+       return len;
+}
 
-static vfs_op_tuple expand_msdfs_ops[] = {     
-       {SMB_VFS_OP(expand_msdfs_readlink), SMB_VFS_OP_READLINK,
-        SMB_VFS_LAYER_TRANSPARENT},
-       {SMB_VFS_OP(NULL), SMB_VFS_OP_NOOP, SMB_VFS_LAYER_NOOP}
+static struct vfs_fn_pointers vfs_expand_msdfs_fns = {
+       .readlink_fn = expand_msdfs_readlink
 };
 
 NTSTATUS vfs_expand_msdfs_init(void);
 NTSTATUS vfs_expand_msdfs_init(void)
 {
        return smb_register_vfs(SMB_VFS_INTERFACE_VERSION, "expand_msdfs",
-                               expand_msdfs_ops);
+                               &vfs_expand_msdfs_fns);
 }