Utilities/cmtar/encode.c

   1 /*
   2 **  Copyright 1998-2003 University of Illinois Board of Trustees
   3 **  Copyright 1998-2003 Mark D. Roth
   4 **  All rights reserved.
   5 **
   6 **  encode.c - libtar code to encode tar header blocks
   7 **
   8 **  Mark D. Roth <roth@uiuc.edu>
   9 **  Campus Information Technologies and Educational Services
  10 **  University of Illinois at Urbana-Champaign
  11 */
  12
  13 #include <libtarint/internal.h>
  14
  15 #include <stdio.h>
  16 #ifndef WIN32
  17 #include <pwd.h>
  18 #include <grp.h>
  19 #endif
  20 #include <sys/types.h>
  21
  22 #ifdef STDC_HEADERS
  23 # include <string.h>
  24 # include <stdlib.h>
  25 #endif
  26
  27
  28 /* magic, version, and checksum */
  29 void
  30 th_finish(TAR *t)
  31 {
  32   int i, sum = 0;
  33
  34   if (t->options & TAR_GNU)
  35     strncpy(t->th_buf.magic, "ustar  ", 8);
  36   else
  37   {
  38     strncpy(t->th_buf.version, TVERSION, TVERSLEN);
  39     strncpy(t->th_buf.magic, TMAGIC, TMAGLEN);
  40   }
  41
  42   for (i = 0; i < T_BLOCKSIZE; i++)
  43     sum += ((char *)(&(t->th_buf)))[i];
  44   for (i = 0; i < 8; i++)
  45     sum += (' ' - t->th_buf.chksum[i]);
  46   int_to_oct(sum, t->th_buf.chksum, 8);
  47 }
  48
  49
  50 /* map a file mode to a typeflag */
  51 void
  52 th_set_type(TAR *t, mode_t mode)
  53 {
  54 #ifdef S_ISLNK
  55   if (S_ISLNK(mode))
  56     t->th_buf.typeflag = SYMTYPE;
  57 #endif
  58   if (S_ISREG(mode))
  59     t->th_buf.typeflag = REGTYPE;
  60   if (S_ISDIR(mode))
  61     t->th_buf.typeflag = DIRTYPE;
  62 #ifdef S_ISCHR
  63   if (S_ISCHR(mode))
  64     t->th_buf.typeflag = CHRTYPE;
  65 #endif
  66   if (S_ISBLK(mode))
  67     t->th_buf.typeflag = BLKTYPE;
  68   if (S_ISFIFO(mode)
  69 #ifdef S_ISSOCK
  70      || S_ISSOCK(mode))
  71 #else
  72     )
  73 #endif
  74     t->th_buf.typeflag = FIFOTYPE;
  75 }
  76
  77
  78 /* encode file path */
  79 void
  80 th_set_path(TAR *t, char *pathname)
  81 {
  82   char suffix[2] = "";
  83   char *tmp;
  84
  85 #ifdef DEBUG
  86   printf("in th_set_path(th, pathname=\"%s\")\n", pathname);
  87 #endif
  88
  89   if (t->th_buf.gnu_longname != NULL)
  90     free(t->th_buf.gnu_longname);
  91   t->th_buf.gnu_longname = NULL;
  92
  93   if (pathname[strlen(pathname) - 1] != '/' && TH_ISDIR(t))
  94     strcpy(suffix, "/");
  95
  96   if (strlen(pathname)+strlen(suffix) >= T_NAMELEN && (t->options & TAR_GNU))
  97   {
  98     /* GNU-style long name */
  99     t->th_buf.gnu_longname = strdup(pathname);
 100     strncpy(t->th_buf.name, t->th_buf.gnu_longname, T_NAMELEN);
 101   }
 102   else if (strlen(pathname)+ strlen(suffix) >= T_NAMELEN)
 103   {
 104     /* POSIX-style prefix field */
 105     tmp = strrchr(pathname, '/');
 106     if (tmp == NULL)
 107     {
 108       printf("!!! '/' not found in \"%s\"\n", pathname);
 109       return;
 110     }
 111     snprintf(t->th_buf.name, 100, "%s%s", &(tmp[1]), suffix);
 112     snprintf(t->th_buf.prefix,
 113        ((tmp - pathname + 1) <
 114         155 ? (tmp - pathname + 1) : 155), "%s", pathname);
 115   }
 116   else
 117     /* classic tar format */
 118     snprintf(t->th_buf.name, 100, "%s%s", pathname, suffix);
 119
 120 #ifdef DEBUG
 121   puts("returning from th_set_path()...");
 122 #endif
 123 }
 124
 125
 126 /* encode link path */
 127 void
 128 th_set_link(TAR *t, char *linkname)
 129 {
 130 #ifdef DEBUG
 131   printf("==> th_set_link(th, linkname=\"%s\")\n", linkname);
 132 #endif
 133
 134   if (strlen(linkname) > T_NAMELEN && (t->options & TAR_GNU))
 135   {
 136     /* GNU longlink format */
 137     t->th_buf.gnu_longlink = strdup(linkname);
 138     strcpy(t->th_buf.linkname, "././@LongLink");
 139   }
 140   else
 141   {
 142     /* classic tar format */
 143     strlcpy(t->th_buf.linkname, linkname,
 144       sizeof(t->th_buf.linkname));
 145     if (t->th_buf.gnu_longlink != NULL)
 146       free(t->th_buf.gnu_longlink);
 147     t->th_buf.gnu_longlink = NULL;
 148   }
 149 }
 150
 151
 152 /* encode device info */
 153 void
 154 th_set_device(TAR *t, dev_t device)
 155 {
 156 #ifdef DEBUG
 157   printf("th_set_device(): major = %d, minor = %d\n",
 158          major(device), minor(device));
 159 #endif
 160   int_to_oct(major(device), t->th_buf.devmajor, 8);
 161   int_to_oct(minor(device), t->th_buf.devminor, 8);
 162 }
 163
 164
 165 /* encode user info */
 166 void
 167 th_set_user(TAR *t, uid_t uid)
 168 {
 169 #ifndef WIN32
 170   struct passwd *pw;
 171
 172   pw = getpwuid(uid);
 173   if (pw != NULL)
 174     strlcpy(t->th_buf.uname, pw->pw_name, sizeof(t->th_buf.uname));
 175 #endif
 176   int_to_oct(uid, t->th_buf.uid, 8);
 177 }
 178
 179
 180 /* encode group info */
 181 void
 182 th_set_group(TAR *t, gid_t gid)
 183 {
 184 #ifndef WIN32
 185   struct group *gr;
 186
 187   gr = getgrgid(gid);
 188   if (gr != NULL)
 189     strlcpy(t->th_buf.gname, gr->gr_name, sizeof(t->th_buf.gname));
 190 #endif
 191   int_to_oct(gid, t->th_buf.gid, 8);
 192 }
 193
 194
 195 /* encode file mode */
 196 void
 197 th_set_mode(TAR *t, mode_t fmode)
 198 {
 199 #ifndef WIN32
 200 #ifndef __BEOS__
 201   if (S_ISSOCK(fmode))
 202   {
 203     fmode &= ~S_IFSOCK;
 204     fmode |= S_IFIFO;
 205   }
 206 #endif
 207 #endif
 208   /* Looks like on windows the st_mode is longer than 8 characters. */
 209   int_to_oct(fmode & 07777777, (t)->th_buf.mode, 8);
 210 }
 211
 212
 213 void
 214 th_set_from_stat(TAR *t, struct stat *s)
 215 {
 216   th_set_type(t, s->st_mode);
 217 #ifndef WIN32
 218   if (S_ISCHR(s->st_mode) || S_ISBLK(s->st_mode))
 219     th_set_device(t, s->st_rdev);
 220 #endif
 221   th_set_user(t, s->st_uid);
 222   th_set_group(t, s->st_gid);
 223   th_set_mode(t, s->st_mode);
 224   th_set_mtime(t, s->st_mtime);
 225   if (S_ISREG(s->st_mode))
 226     th_set_size(t, s->st_size);
 227   else
 228     th_set_size(t, 0);
 229 }
 230
 231