]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/findlib/attribs.c
This commit was manufactured by cvs2svn to create tag
[bacula/bacula] / bacula / src / findlib / attribs.c
index 3d8576761683eb3e50ad8b21ca3313711cc2c948..f0d65be64cebafa9eb498af37b99a9763de1e6e9 100755 (executable)
 
 #include "bacula.h"
 #include "find.h"
-#include "jcr.h"
 
 #ifdef HAVE_CYGWIN
-#include <windows.h>
 
 /* Forward referenced subroutines */
-static
-int set_win32_attributes(void *jcr, char *fname, char *ofile, char *lname, 
-                        int type, int stream, struct stat *statp,
-                        char *attribsEx, int *ofd);
+static int set_win32_attributes(JCR *jcr, ATTR *attr, BFILE *ofd);
 void unix_name_to_win32(POOLMEM **win32_name, char *name);
-extern "C" HANDLE get_osfhandle(int fd);
-void win_error(void *jcr, char *prefix, POOLMEM *ofile);
+void win_error(JCR *jcr, char *prefix, POOLMEM *ofile);
+HANDLE bget_handle(BFILE *bfd);
 #endif
 
 /* For old systems that don't have lchown() use chown() */
@@ -56,17 +51,55 @@ void win_error(void *jcr, char *prefix, POOLMEM *ofile);
 /*                                                            */
 /*=============================================================*/
 
+/*
+ * Return the data stream that will be used 
+ */
+int select_data_stream(FF_PKT *ff_pkt)
+{
+   int stream;
+
+   /* Note, no sparse option for win32_data */
+   if (!is_portable_backup(&ff_pkt->bfd)) {
+      stream = STREAM_WIN32_DATA;
+      ff_pkt->flags &= ~FO_SPARSE;
+   } else if (ff_pkt->flags & FO_SPARSE) {
+      stream = STREAM_SPARSE_DATA;
+   } else {
+      stream = STREAM_FILE_DATA;
+   }
+#ifdef HAVE_LIBZ
+   if (ff_pkt->flags & FO_GZIP) {
+      if (stream == STREAM_WIN32_DATA) {
+        stream = STREAM_WIN32_GZIP_DATA;
+      } else if (stream == STREAM_FILE_DATA) {
+        stream = STREAM_GZIP_DATA;
+      } else {
+        stream = STREAM_SPARSE_GZIP_DATA;
+      }
+   }
+#endif
+   return stream;
+}
+
 
-/* Encode a stat structure into a base64 character string */
-void encode_stat(char *buf, struct stat *statp, uint32_t LinkFI)
+/* 
+ * Encode a stat structure into a base64 character string   
+ *   All systems must create such a structure.
+ *   In addition, we tack on the LinkFI, which is non-zero in
+ *   the case of a hard linked file that has no data.  This
+ *   is a File Index pointing to the link that does have the
+ *   data (always the first one encountered in a save).
+ * You may piggyback attributes on this packet by encoding
+ *   them in the encode_attribsEx() subroutine, but this is
+ *   not recommended.
+ */
+void encode_stat(char *buf, FF_PKT *ff_pkt, int data_stream)
 {
    char *p = buf;
+   struct stat *statp = &ff_pkt->statp;
    /*
-    * NOTE: we should use rdev as major and minor device if
-    * it is a block or char device (S_ISCHR(statp->st_mode)
-    * or S_ISBLK(statp->st_mode)).  In all other cases,
-    * it is not used.  
-    *
+    *  Encode a stat packet.  I should have done this more intelligently
+    *  with a length so that it could be easily expanded.
     */
    p += to_base64((int64_t)statp->st_dev, p);
    *p++ = ' ';                        /* separate fields with a space */
@@ -94,7 +127,17 @@ void encode_stat(char *buf, struct stat *statp, uint32_t LinkFI)
    *p++ = ' ';
    p += to_base64((int64_t)statp->st_ctime, p);
    *p++ = ' ';
-   p += to_base64((int64_t)LinkFI, p);
+   p += to_base64((int64_t)ff_pkt->LinkFI, p);
+   *p++ = ' ';
+
+#ifdef HAVE_CHFLAGS
+   /* FreeBSD function */
+   p += to_base64((int64_t)statp->st_flags, p);  /* output st_flags */
+#else
+   p += to_base64((int64_t)0, p);     /* output place holder */
+#endif
+   *p++ = ' ';
+   p += to_base64((int64_t)data_stream, p);
    *p = 0;
    return;
 }
@@ -102,8 +145,7 @@ void encode_stat(char *buf, struct stat *statp, uint32_t LinkFI)
 
 
 /* Decode a stat packet from base64 characters */
-void
-decode_stat(char *buf, struct stat *statp, uint32_t *LinkFI) 
+int decode_stat(char *buf, struct stat *statp, int32_t *LinkFI) 
 {
    char *p = buf;
    int64_t val;
@@ -146,14 +188,36 @@ decode_stat(char *buf, struct stat *statp, uint32_t *LinkFI)
    p++;
    p += from_base64(&val, p);
    statp->st_ctime = val;
+
    /* Optional FileIndex of hard linked file data */
    if (*p == ' ' || (*p != 0 && *(p+1) == ' ')) {
       p++;
       p += from_base64(&val, p);
       *LinkFI = (uint32_t)val;
-  } else {
+   } else {
       *LinkFI = 0;
-  }
+      return 0;
+   }
+
+   /* FreeBSD user flags */
+   if (*p == ' ' || (*p != 0 && *(p+1) == ' ')) {
+      p++;
+      p += from_base64(&val, p);
+#ifdef HAVE_CHFLAGS
+      statp->st_flags = (uint32_t)val;
+   } else {
+      statp->st_flags  = 0;
+#endif
+   }
+    
+   /* Look for data stream id */
+   if (*p == ' ' || (*p != 0 && *(p+1) == ' ')) {
+      p++;
+      p += from_base64(&val, p);
+   } else {
+      val = 0;
+   }
+   return (int)val;
 }
 
 /*
@@ -165,15 +229,28 @@ decode_stat(char *buf, struct stat *statp, uint32_t *LinkFI)
  * Returns:  1 on success
  *          0 on failure
  */
-int set_attributes(void *jcr, char *fname, char *ofile, char *lname, 
-                  int type, int stream, struct stat *statp,
-                  char *attribsEx, int *ofd)
+int set_attributes(JCR *jcr, ATTR *attr, BFILE *ofd)
 {
    struct utimbuf ut;   
+   mode_t old_mask;
+   int stat = 1;
 
 #ifdef HAVE_CYGWIN
-   if (set_win32_attributes(jcr, fname, ofile, lname, type, stream,
-                           statp, attribsEx, ofd)) {
+   if (attr->data_stream == STREAM_WIN32_DATA ||
+       attr->data_stream == STREAM_WIN32_GZIP_DATA) {
+      if (is_bopen(ofd)) {
+        bclose(ofd); 
+      }
+      pm_strcpy(&attr->ofname, "*none*");
+      return 1;
+   }
+
+   if (attr->stream == STREAM_UNIX_ATTRIBUTES_EX &&
+       set_win32_attributes(jcr, attr, ofd)) {
+      if (is_bopen(ofd)) {
+        bclose(ofd); 
+      }
+      pm_strcpy(&attr->ofname, "*none*");
       return 1;
    }
    /*
@@ -183,43 +260,58 @@ int set_attributes(void *jcr, char *fname, char *ofile, char *lname,
     */
 #endif
 
-   if (*ofd != -1) {
-      close(*ofd);                   /* first close file */
-      *ofd = -1;
+   old_mask = umask(0);
+   if (is_bopen(ofd)) {
+      bclose(ofd);                   /* first close file */
    }
 
-   ut.actime = statp->st_atime;
-   ut.modtime = statp->st_mtime;
+   ut.actime = attr->statp.st_atime;
+   ut.modtime = attr->statp.st_mtime;
 
    /* ***FIXME**** optimize -- don't do if already correct */
-   if (type == FT_LNK) {
-      if (lchown(ofile, statp->st_uid, statp->st_gid) < 0) {
-         Jmsg2(jcr, M_ERROR, 0, "Unable to set file owner %s: ERR=%s\n",
-           ofile, strerror(errno));
-        return 0;
+   /* 
+    * For link, change owner of link using lchown, but don't
+    *  try to do a chmod as that will update the file behind it.
+    */
+   if (attr->type == FT_LNK) {
+      /* Change owner of link, not of real file */
+      if (lchown(attr->ofname, attr->statp.st_uid, attr->statp.st_gid) < 0) {
+         Jmsg2(jcr, M_ERROR, 0, _("Unable to set file owner %s: ERR=%s\n"),
+           attr->ofname, strerror(errno));
+        stat = 0;
       }
    } else {
-      if (chown(ofile, statp->st_uid, statp->st_gid) < 0) {
-         Jmsg2(jcr, M_ERROR, 0, "Unable to set file owner %s: ERR=%s\n",
-           ofile, strerror(errno));
-        return 0;
+      if (chown(attr->ofname, attr->statp.st_uid, attr->statp.st_gid) < 0) {
+         Jmsg2(jcr, M_ERROR, 0, _("Unable to set file owner %s: ERR=%s\n"),
+           attr->ofname, strerror(errno));
+        stat = 0;
+      }
+      if (chmod(attr->ofname, attr->statp.st_mode) < 0) {
+         Jmsg2(jcr, M_ERROR, 0, _("Unable to set file modes %s: ERR=%s\n"),
+           attr->ofname, strerror(errno));
+        stat = 0;
       }
-   }
-   if (chmod(ofile, statp->st_mode) < 0) {
-      Jmsg2(jcr, M_ERROR, 0, "Unable to set file modes %s: ERR=%s\n",
-        ofile, strerror(errno));
-      return 0;
-   }
 
-   /*
-    * Reset file times.
-    */
-   if (utime(ofile, &ut) < 0) {
-      Jmsg2(jcr, M_ERROR, 0, "Unable to set file times %s: ERR=%s\n",
-        ofile, strerror(errno));
-      return 0;
+      /* FreeBSD user flags */
+#ifdef HAVE_CHFLAGS
+      if (chflags(attr->ofname, attr->statp.st_flags) < 0) {
+         Jmsg2(jcr, M_ERROR, 0, _("Unable to set file flags %s: ERR=%s\n"),
+           attr->ofname, strerror(errno));
+        stat = 0;
+      }
+#endif
+      /*
+       * Reset file times.
+       */
+      if (utime(attr->ofname, &ut) < 0) {
+         Jmsg2(jcr, M_ERROR, 0, _("Unable to set file times %s: ERR=%s\n"),
+           attr->ofname, strerror(errno));
+        stat = 0;
+      }
    }
-   return 1;
+   pm_strcpy(&attr->ofname, "*none*");
+   umask(old_mask);
+   return stat;
 }
 
 
@@ -232,10 +324,15 @@ int set_attributes(void *jcr, char *fname, char *ofile, char *lname,
 #ifndef HAVE_CYGWIN
     
 /*
- * If you have a Unix system with extended attributes (e.g.
- *  ACLs for Solaris, do it here.
+ * It is possible to piggyback additional data e.g. ACLs on
+ *   the encode_stat() data by returning the extended attributes
+ *   here.  They must be "self-contained" (i.e. you keep track
+ *   of your own length), and they must be in ASCII string
+ *   format. Using this feature is not recommended.
+ * The code below shows how to return nothing. See the Win32
+ *   code below for returning something in the attributes.
  */
-int encode_attribsEx(void *jcr, char *attribsEx, FF_PKT *ff_pkt)
+int encode_attribsEx(JCR *jcr, char *attribsEx, FF_PKT *ff_pkt)
 {
    *attribsEx = 0;                   /* no extended attributes */
    return STREAM_UNIX_ATTRIBUTES;
@@ -253,9 +350,7 @@ int encode_attribsEx(void *jcr, char *attribsEx, FF_PKT *ff_pkt)
 
 #ifdef HAVE_CYGWIN
 
-int NoGetFileAttributesEx = 0;
-
-int encode_attribsEx(void *jcr, char *attribsEx, FF_PKT *ff_pkt)
+int encode_attribsEx(JCR *jcr, char *attribsEx, FF_PKT *ff_pkt)
 {
    char *p = attribsEx;
    WIN32_FILE_ATTRIBUTE_DATA atts;
@@ -263,15 +358,15 @@ int encode_attribsEx(void *jcr, char *attribsEx, FF_PKT *ff_pkt)
 
    attribsEx[0] = 0;                 /* no extended attributes */
 
-   if (NoGetFileAttributesEx) {
+   if (!p_GetFileAttributesEx) {                                
       return STREAM_UNIX_ATTRIBUTES;
    }
 
    unix_name_to_win32(&ff_pkt->sys_fname, ff_pkt->fname);
-   if (!GetFileAttributesEx(ff_pkt->sys_fname, GetFileExInfoStandard,
+   if (!p_GetFileAttributesEx(ff_pkt->sys_fname, GetFileExInfoStandard,
                            (LPVOID)&atts)) {
       win_error(jcr, "GetFileAttributesEx:", ff_pkt->sys_fname);
-      return STREAM_WIN32_ATTRIBUTES;
+      return STREAM_UNIX_ATTRIBUTES;
    }
 
    p += to_base64((uint64_t)atts.dwFileAttributes, p);
@@ -292,7 +387,7 @@ int encode_attribsEx(void *jcr, char *attribsEx, FF_PKT *ff_pkt)
    *p++ = ' ';
    p += to_base64((uint64_t)atts.nFileSizeLow, p);
    *p = 0;
-   return STREAM_WIN32_ATTRIBUTES;
+   return STREAM_UNIX_ATTRIBUTES_EX;
 }
 
 /* Define attributes that are legal to set with SetFileAttributes() */
@@ -316,27 +411,26 @@ int encode_attribsEx(void *jcr, char *attribsEx, FF_PKT *ff_pkt)
  * Returns:  1 on success
  *          0 on failure
  */
-static
-int set_win32_attributes(void *jcr, char *fname, char *ofile, char *lname, 
-                        int type, int stream, struct stat *statp,
-                        char *attribsEx, int *ofd)
+static int set_win32_attributes(JCR *jcr, ATTR *attr, BFILE *ofd)
 {
-   char *p = attribsEx;
+   char *p = attr->attrEx;
    int64_t val;
    WIN32_FILE_ATTRIBUTE_DATA atts;
    ULARGE_INTEGER li;
-   int fid, stat;
    POOLMEM *win32_ofile;
 
+   if (!p_GetFileAttributesEx) {                                
+      return 0;
+   }
+
    if (!p || !*p) {                  /* we should have attributes */
-      Dmsg2(100, "Attributes missing. of=%s ofd=%d\n", ofile, *ofd);
-      if (*ofd != -1) {
-        close(*ofd);
-        *ofd = -1;
+      Dmsg2(100, "Attributes missing. of=%s ofd=%d\n", attr->ofname, ofd->fid);
+      if (is_bopen(ofd)) {
+        bclose(ofd);
       }
       return 0;
    } else {
-      Dmsg2(100, "Attribs %s = %s\n", ofile, attribsEx);
+      Dmsg2(100, "Attribs %s = %s\n", attr->ofname, attr->attrEx);
    }
 
    p += from_base64(&val, p);
@@ -363,45 +457,42 @@ int set_win32_attributes(void *jcr, char *fname, char *ofile, char *lname,
    p += from_base64(&val, p);
    atts.nFileSizeLow = val;
 
-   /* At this point, we have reconstructed the WIN32_FILE_ATTRIBUTE_DATA pkt */
-
    /* Convert to Windows path format */
    win32_ofile = get_pool_memory(PM_FNAME);
-   unix_name_to_win32(&win32_ofile, ofile);
+   unix_name_to_win32(&win32_ofile, attr->ofname);
 
-   if (*ofd == -1) {
-      Dmsg1(100, "File not open: %s\n", ofile);
-      fid = open(ofile, O_RDWR|O_BINARY);   /* attempt to open the file */
-      if (fid >= 0) {
-        *ofd = fid;
-      }
+
+
+   /* At this point, we have reconstructed the WIN32_FILE_ATTRIBUTE_DATA pkt */
+
+   if (!is_bopen(ofd)) {
+      Dmsg1(100, "File not open: %s\n", attr->ofname);
+      bopen(ofd, attr->ofname, O_WRONLY|O_BINARY, 0);  /* attempt to open the file */
    }
 
-   if (*ofd != -1) {
-      Dmsg1(100, "SetFileTime %s\n", ofile);
-      stat = SetFileTime(get_osfhandle(*ofd),
+   if (is_bopen(ofd)) {
+      Dmsg1(100, "SetFileTime %s\n", attr->ofname);
+      if (!SetFileTime(bget_handle(ofd),
                         &atts.ftCreationTime,
                         &atts.ftLastAccessTime,
-                        &atts.ftLastWriteTime);
-      if (stat != 1) {
+                        &atts.ftLastWriteTime)) {
          win_error(jcr, "SetFileTime:", win32_ofile);
       }
-      close(*ofd);
-      *ofd = -1;
+      bclose(ofd);
    }
 
-   Dmsg1(100, "SetFileAtts %s\n", ofile);
-   stat = SetFileAttributes(win32_ofile, atts.dwFileAttributes & SET_ATTRS);
-   if (stat != 1) {
-      win_error(jcr, "SetFileAttributes:", win32_ofile);
+   Dmsg1(100, "SetFileAtts %s\n", attr->ofname);
+   if (!(atts.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) {
+      if (!SetFileAttributes(win32_ofile, atts.dwFileAttributes & SET_ATTRS)) {
+         win_error(jcr, "SetFileAttributes:", win32_ofile);
+      }
    }
    free_pool_memory(win32_ofile);
    return 1;
 }
 
-void win_error(void *vjcr, char *prefix, POOLMEM *win32_ofile)
+void win_error(JCR *jcr, char *prefix, POOLMEM *win32_ofile)
 {
-   JCR *jcr = (JCR *)vjcr; 
    DWORD lerror = GetLastError();
    LPTSTR msg;
    FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|
@@ -413,10 +504,32 @@ void win_error(void *vjcr, char *prefix, POOLMEM *win32_ofile)
                 0,
                 NULL);
    Dmsg3(100, "Error in %s on file %s: ERR=%s\n", prefix, win32_ofile, msg);
-   Jmsg3(jcr, M_INFO, 0, _("Error in %s file %s: ERR=%s\n"), prefix, win32_ofile, msg);
+   strip_trailing_junk(msg);
+   Jmsg(jcr, M_ERROR, 0, _("Error in %s file %s: ERR=%s\n"), prefix, win32_ofile, msg);
    LocalFree(msg);
 }
 
+void win_error(JCR *jcr, char *prefix, DWORD lerror)
+{
+   LPTSTR msg;
+   FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|
+                FORMAT_MESSAGE_FROM_SYSTEM,
+                NULL,
+                lerror,
+                0,
+                (LPTSTR)&msg,
+                0,
+                NULL);
+   strip_trailing_junk(msg);
+   if (jcr) {
+      Jmsg2(jcr, M_ERROR, 0, _("Error in %s: ERR=%s\n"), prefix, msg);
+   } else {
+      MessageBox(NULL, msg, prefix, MB_OK);
+   }
+   LocalFree(msg);
+}
+
+
 /* Cygwin API definition */
 extern "C" void cygwin_conv_to_win32_path(const char *path, char *win32_path);