Remove alloca() from strntoul() helper in cpio
[platform/upstream/rpm.git] / lib / cpio.c
1 /** \ingroup payload
2  * \file lib/cpio.c
3  *  Handle cpio payloads within rpm packages.
4  *
5  * \warning FIXME: We don't translate between cpio and system mode bits! These
6  * should both be the same, but really odd things are going to happen if
7  * that's not true!
8  */
9
10 #include "system.h"
11
12 #include <rpm/rpmio.h>
13 #include <rpm/rpmlog.h>
14
15 #include "lib/cpio.h"
16 #include "lib/fsm.h"
17
18 #include "debug.h"
19
20
21 /**
22  * Convert string to unsigned integer (with buffer size check).
23  * @param str           input string
24  * @retval endptr       address of 1st character not processed
25  * @param base          numerical conversion base
26  * @param num           max no. of bytes to read
27  * @return              converted integer
28  */
29 static int strntoul(const char *str,char **endptr, int base, size_t num)
30 {
31     char buf[num+1], * end;
32     unsigned long ret;
33
34     rstrlcpy(buf, str, num+1);
35
36     ret = strtoul(buf, &end, base);
37     if (*end != '\0')
38         *endptr = ((char *)str) + (end - buf);  /* XXX discards const */
39     else
40         *endptr = ((char *)str) + strlen(buf);
41
42     return ret;
43 }
44
45 #define GET_NUM_FIELD(phys, log) \
46         \
47         log = strntoul(phys, &end, 16, sizeof(phys)); \
48         \
49         if ( (end - phys) != sizeof(phys) ) return CPIOERR_BAD_HEADER;
50 #define SET_NUM_FIELD(phys, val, space) \
51         sprintf(space, "%8.8lx", (unsigned long) (val)); \
52         \
53         memcpy(phys, space, 8) \
54
55 int cpioTrailerWrite(FSM_t fsm)
56 {
57     struct cpioCrcPhysicalHeader * hdr =
58         (struct cpioCrcPhysicalHeader *)fsm->rdbuf;
59     int rc;
60
61     memset(hdr, '0', PHYS_HDR_SIZE);
62     memcpy(hdr->magic, CPIO_NEWC_MAGIC, sizeof(hdr->magic));
63     memcpy(hdr->nlink, "00000001", 8);
64     memcpy(hdr->namesize, "0000000b", 8);
65     memcpy(fsm->rdbuf + PHYS_HDR_SIZE, CPIO_TRAILER, sizeof(CPIO_TRAILER));
66
67     /* XXX DWRITE uses rdnb for I/O length. */
68     fsm->rdnb = PHYS_HDR_SIZE + sizeof(CPIO_TRAILER);
69     rc = fsmNext(fsm, FSM_DWRITE);
70
71     /*
72      * GNU cpio pads to 512 bytes here, but we don't. This may matter for
73      * tape device(s) and/or concatenated cpio archives. <shrug>
74      */
75     if (!rc)
76         rc = fsmNext(fsm, FSM_PAD);
77
78     return rc;
79 }
80
81 int cpioHeaderWrite(FSM_t fsm, struct stat * st)
82 {
83     struct cpioCrcPhysicalHeader * hdr = (struct cpioCrcPhysicalHeader *)fsm->rdbuf;
84     char field[64];
85     size_t len;
86     dev_t dev;
87     int rc = 0;
88
89     memcpy(hdr->magic, CPIO_NEWC_MAGIC, sizeof(hdr->magic));
90     SET_NUM_FIELD(hdr->inode, st->st_ino, field);
91     SET_NUM_FIELD(hdr->mode, st->st_mode, field);
92     SET_NUM_FIELD(hdr->uid, st->st_uid, field);
93     SET_NUM_FIELD(hdr->gid, st->st_gid, field);
94     SET_NUM_FIELD(hdr->nlink, st->st_nlink, field);
95     SET_NUM_FIELD(hdr->mtime, st->st_mtime, field);
96     SET_NUM_FIELD(hdr->filesize, st->st_size, field);
97
98     dev = major(st->st_dev); SET_NUM_FIELD(hdr->devMajor, dev, field);
99     dev = minor(st->st_dev); SET_NUM_FIELD(hdr->devMinor, dev, field);
100     dev = major(st->st_rdev); SET_NUM_FIELD(hdr->rdevMajor, dev, field);
101     dev = minor(st->st_rdev); SET_NUM_FIELD(hdr->rdevMinor, dev, field);
102
103     len = strlen(fsm->path) + 1; SET_NUM_FIELD(hdr->namesize, len, field);
104     memcpy(hdr->checksum, "00000000", 8);
105     memcpy(fsm->rdbuf + PHYS_HDR_SIZE, fsm->path, len);
106
107     /* XXX DWRITE uses rdnb for I/O length. */
108     fsm->rdnb = PHYS_HDR_SIZE + len;
109     rc = fsmNext(fsm, FSM_DWRITE);
110     if (!rc && fsm->rdnb != fsm->wrnb)
111         rc = CPIOERR_WRITE_FAILED;
112     if (!rc)
113         rc = fsmNext(fsm, FSM_PAD);
114     return rc;
115 }
116
117 int cpioHeaderRead(FSM_t fsm, struct stat * st)
118 {
119     struct cpioCrcPhysicalHeader hdr;
120     int nameSize;
121     char * end;
122     unsigned int major, minor;
123     int rc = 0;
124
125     fsm->wrlen = PHYS_HDR_SIZE;
126     rc = fsmNext(fsm, FSM_DREAD);
127     if (!rc && fsm->rdnb != fsm->wrlen)
128         rc = CPIOERR_READ_FAILED;
129     if (rc) return rc;
130     memcpy(&hdr, fsm->wrbuf, fsm->rdnb);
131
132     if (strncmp(CPIO_CRC_MAGIC, hdr.magic, sizeof(CPIO_CRC_MAGIC)-1) &&
133         strncmp(CPIO_NEWC_MAGIC, hdr.magic, sizeof(CPIO_NEWC_MAGIC)-1))
134         return CPIOERR_BAD_MAGIC;
135
136     GET_NUM_FIELD(hdr.inode, st->st_ino);
137     GET_NUM_FIELD(hdr.mode, st->st_mode);
138     GET_NUM_FIELD(hdr.uid, st->st_uid);
139     GET_NUM_FIELD(hdr.gid, st->st_gid);
140     GET_NUM_FIELD(hdr.nlink, st->st_nlink);
141     GET_NUM_FIELD(hdr.mtime, st->st_mtime);
142     GET_NUM_FIELD(hdr.filesize, st->st_size);
143
144     GET_NUM_FIELD(hdr.devMajor, major);
145     GET_NUM_FIELD(hdr.devMinor, minor);
146     st->st_dev = makedev(major, minor);
147
148     GET_NUM_FIELD(hdr.rdevMajor, major);
149     GET_NUM_FIELD(hdr.rdevMinor, minor);
150     st->st_rdev = makedev(major, minor);
151
152     GET_NUM_FIELD(hdr.namesize, nameSize);
153     if (nameSize >= fsm->wrsize)
154         return CPIOERR_BAD_HEADER;
155
156     {   char * t = xmalloc(nameSize + 1);
157         fsm->wrlen = nameSize;
158         rc = fsmNext(fsm, FSM_DREAD);
159         if (!rc && fsm->rdnb != fsm->wrlen)
160             rc = CPIOERR_BAD_HEADER;
161         if (rc) {
162             t = _free(t);
163             fsm->path = NULL;
164             return rc;
165         }
166         memcpy(t, fsm->wrbuf, fsm->rdnb);
167         t[nameSize] = '\0';
168         fsm->path = t;
169     }
170
171     return 0;
172 }
173
174 const char * cpioStrerror(int rc)
175 {
176     static char msg[256];
177     const char *s;
178     int myerrno = errno;
179     size_t l;
180
181     strcpy(msg, "cpio: ");
182     switch (rc) {
183     default: {
184         char *t = msg + strlen(msg);
185         sprintf(t, _("(error 0x%x)"), (unsigned)rc);
186         s = NULL;
187         break;
188     }
189     case CPIOERR_BAD_MAGIC:     s = _("Bad magic");             break;
190     case CPIOERR_BAD_HEADER:    s = _("Bad/unreadable  header");break;
191
192     case CPIOERR_OPEN_FAILED:   s = "open";     break;
193     case CPIOERR_CHMOD_FAILED:  s = "chmod";    break;
194     case CPIOERR_CHOWN_FAILED:  s = "chown";    break;
195     case CPIOERR_WRITE_FAILED:  s = "write";    break;
196     case CPIOERR_UTIME_FAILED:  s = "utime";    break;
197     case CPIOERR_UNLINK_FAILED: s = "unlink";   break;
198     case CPIOERR_RENAME_FAILED: s = "rename";   break;
199     case CPIOERR_SYMLINK_FAILED: s = "symlink"; break;
200     case CPIOERR_STAT_FAILED:   s = "stat";     break;
201     case CPIOERR_LSTAT_FAILED:  s = "lstat";    break;
202     case CPIOERR_MKDIR_FAILED:  s = "mkdir";    break;
203     case CPIOERR_RMDIR_FAILED:  s = "rmdir";    break;
204     case CPIOERR_MKNOD_FAILED:  s = "mknod";    break;
205     case CPIOERR_MKFIFO_FAILED: s = "mkfifo";   break;
206     case CPIOERR_LINK_FAILED:   s = "link";     break;
207     case CPIOERR_READLINK_FAILED: s = "readlink";       break;
208     case CPIOERR_READ_FAILED:   s = "read";     break;
209     case CPIOERR_COPY_FAILED:   s = "copy";     break;
210     case CPIOERR_LSETFCON_FAILED: s = "lsetfilecon";    break;
211
212     case CPIOERR_HDR_SIZE:      s = _("Header size too big");   break;
213     case CPIOERR_UNKNOWN_FILETYPE: s = _("Unknown file type");  break;
214     case CPIOERR_MISSING_HARDLINK: s = _("Missing hard link(s)"); break;
215     case CPIOERR_MD5SUM_MISMATCH: s = _("MD5 sum mismatch");    break;
216     case CPIOERR_INTERNAL:      s = _("Internal error");        break;
217     case CPIOERR_UNMAPPED_FILE: s = _("Archive file not in header"); break;
218     case CPIOERR_ENOENT:        s = strerror(ENOENT); break;
219     case CPIOERR_ENOTEMPTY:     s = strerror(ENOTEMPTY); break;
220     }
221
222     l = sizeof(msg) - strlen(msg) - 1;
223     if (s != NULL) {
224         if (l > 0) strncat(msg, s, l);
225         l -= strlen(s);
226     }
227     if ((rc & CPIOERR_CHECK_ERRNO) && myerrno) {
228         s = _(" failed - ");
229         if (l > 0) strncat(msg, s, l);
230         l -= strlen(s);
231         if (l > 0) strncat(msg, strerror(myerrno), l);
232     }
233     return msg;
234 }