| #include "cpio.h" |
| |
| #include <stdint.h> |
| |
| #include "std.h" |
| |
| #pragma pack(push, 1) |
| struct cpio_header { |
| uint16_t magic; |
| uint16_t dev; |
| uint16_t ino; |
| uint16_t mode; |
| uint16_t uid; |
| uint16_t gid; |
| uint16_t nlink; |
| uint16_t rdev; |
| uint16_t mtime[2]; |
| uint16_t namesize; |
| uint16_t filesize[2]; |
| }; |
| #pragma pack(pop) |
| |
| void cpio_init(struct cpio *c, const void *buf, size_t size) |
| { |
| c->first = buf; |
| c->total_size = size; |
| } |
| |
| void cpio_init_iter(struct cpio *c, struct cpio_iter *iter) |
| { |
| iter->cur = c->first; |
| iter->size_left = c->total_size; |
| } |
| |
| bool cpio_next(struct cpio_iter *iter, const char **name, const void **contents, |
| size_t *size) |
| { |
| const struct cpio_header *h = iter->cur; |
| size_t size_left; |
| size_t filelen; |
| size_t namelen; |
| |
| size_left = iter->size_left; |
| if (size_left < sizeof(struct cpio_header)) { |
| return false; |
| } |
| |
| /* TODO: Check magic. */ |
| |
| size_left -= sizeof(struct cpio_header); |
| namelen = (h->namesize + 1) & ~1; |
| if (size_left < namelen) { |
| return false; |
| } |
| |
| size_left -= namelen; |
| filelen = (size_t)h->filesize[0] << 16 | h->filesize[1]; |
| if (size_left < filelen) { |
| return false; |
| } |
| |
| /* TODO: Check that string is null-terminated. */ |
| /* TODO: Check that trailler is not returned. */ |
| |
| /* Stop enumerating files when we hit the end marker. */ |
| if (!strcmp((const char *)(iter->cur + 1), "TRAILER!!!")) { |
| return false; |
| } |
| |
| size_left -= filelen; |
| |
| *name = (const char *)(iter->cur + 1); |
| *contents = *name + namelen; |
| *size = filelen; |
| |
| iter->cur = (struct cpio_header *)((char *)*contents + filelen); |
| iter->cur = |
| (struct cpio_header *)(char *)(((size_t)iter->cur + 1) & ~1); |
| iter->size_left = size_left; |
| |
| return true; |
| } |