From bd952f83cce8f9a15ff63cc00adc237dc5dc398b Mon Sep 17 00:00:00 2001 From: Patrick McDermott Date: Sat, 29 Apr 2023 19:48:14 -0400 Subject: opk: Split reading out into opk/read --- (limited to 'src/opk') diff --git a/src/opk/local.mk b/src/opk/local.mk new file mode 100644 index 0000000..a535a4f --- /dev/null +++ b/src/opk/local.mk @@ -0,0 +1,3 @@ +opkg_opk_SOURCES += \ + %reldir%/opk.h \ + %reldir%/read.c diff --git a/src/opk/opk.h b/src/opk/opk.h new file mode 100644 index 0000000..e5fc4d2 --- /dev/null +++ b/src/opk/opk.h @@ -0,0 +1,51 @@ +/* + * Copyright (C) 2023 Patrick McDermott + * + * This file is part of opkg-opkg. + * + * opkg-opkg 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 3 of the License, or + * (at your option) any later version. + * + * opkg-opkg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with opkg-opkg. If not, see . + */ + +#ifndef OPKG_OPK_OPK_OPK_H_ +#define OPKG_OPK_OPK_OPK_H_ + +#include +#include "../gzip.h" +#include "../opk.h" +#include "../ustar.h" + +struct _opkg_opk_opk_seek_name { + const char *name; + struct _opkg_opk_opk_seek_name *prev; + struct _opkg_opk_opk_seek_name *next; +}; + +struct opkg_opk_opk { + struct _opkg_opk_opk_seek_name *print_control_head; + struct _opkg_opk_opk_seek_name *print_control_tail; + struct _opkg_opk_opk_seek_name *print_data_head; + struct _opkg_opk_opk_seek_name *print_data_tail; + int list_control; + int list_data; + const char *control_dir; + FILE *file; + char file_buffer[8192]; + struct opkg_opk_gzip *outer_gzip; + struct opkg_opk_ustar *outer_ustar; + struct opkg_opk_gzip *inner_gzip; + struct opkg_opk_ustar *inner_ustar; + int previously_printed; +}; + +#endif /* OPKG_OPK_OPK_OPK_H_ */ diff --git a/src/opk/read.c b/src/opk/read.c new file mode 100644 index 0000000..3bbb128 --- /dev/null +++ b/src/opk/read.c @@ -0,0 +1,556 @@ +/* + * Copyright (C) 2023 Patrick McDermott + * + * This file is part of opkg-opkg. + * + * opkg-opkg 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 3 of the License, or + * (at your option) any later version. + * + * opkg-opkg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with opkg-opkg. If not, see . + */ + +#include +#include +#include +#include +#include +#include +#include "../defs.h" +#include "../gzip.h" +#include "../i18n.h" +#include "../opk.h" +#include "../ustar.h" +#include "opk.h" + +static int +_opkg_opk_opk_file_read(void *user_data, char **buffer, size_t *size) +{ + struct opkg_opk_opk *opk = user_data; + + *buffer = opk->file_buffer; + *size = fread(opk->file_buffer, 1, sizeof(opk->file_buffer), + opk->file); + if (feof(opk->file)) { + return OPKG_OPK_END; + } else if (ferror(opk->file) || *size == 0) { + return OPKG_OPK_ERROR; + } else { + return OPKG_OPK_OK; + } +} + +static int +_opkg_opk_opk_init_inner(struct opkg_opk_opk *opk) +{ + /* Initialize inner gzip decompressor. */ + opk->inner_gzip = opkg_opk_gzip_init_read( + (opkg_opk_gzip_read_func *) &opkg_opk_ustar_read, + opk->outer_ustar); + if (opk->inner_gzip == NULL) { + fputs(_("Error: Failed to initialize\n"), stderr); + return OPKG_OPK_ERROR; + } + + /* Initialize inner ustar unarchiver. */ + opk->inner_ustar = opkg_opk_ustar_init(opk->inner_gzip); + if (opk->inner_ustar == NULL) { + fputs(_("Error: Failed to initialize\n"), stderr); + opkg_opk_gzip_free(opk->inner_gzip); + return OPKG_OPK_ERROR; + } + + return OPKG_OPK_OK; +} + +static void +_opkg_opk_opk_free_inner(struct opkg_opk_opk *opk) +{ + opkg_opk_ustar_free(opk->inner_ustar); + opkg_opk_gzip_free(opk->inner_gzip); +} + +static int +_opkg_opk_opk_check_name(const char *member_name, + struct _opkg_opk_opk_seek_name **head, + struct _opkg_opk_opk_seek_name **tail) +{ + struct _opkg_opk_opk_seek_name *seek_name; + + if (member_name[0] == '.' && member_name[1] == '/') { + member_name += 2; + } else if (member_name[0] == '/') { + member_name += 1; + } + + /* Check each requested name. */ + for (seek_name = *head; seek_name != NULL; seek_name = seek_name->next) + { + if (strcmp(member_name, seek_name->name) == 0) { + if (seek_name->prev != NULL) { + seek_name->prev->next = seek_name->next; + } else { + /* This was the head. Update. */ + *head = seek_name->next; + } + if (seek_name->next != NULL) { + seek_name->next->prev = seek_name->prev; + } else { + /* This was the tail. Update. */ + *tail = seek_name->prev; + } + free(seek_name); + return OPKG_OPK_OK; + } + } + + return OPKG_OPK_ERROR; /* Member not found */ +} + +static int +_opkg_opk_opk_read_control(struct opkg_opk_opk *opk) +{ + struct stat stat_buf; + char *path; + struct opkg_opk_ustar_member *member; + int ret_list; + FILE *fp; + int print; + int ret_read; + char *buffer; + size_t size; + struct _opkg_opk_opk_seek_name *seek_name; + + if (opk->control_dir != NULL) { + if (stat(opk->control_dir, &stat_buf) == 0) { + if (!S_ISDIR(stat_buf.st_mode)) { + fputs(_("Error: Cannot create control directory" + "\n"), stderr); + return OPKG_OPK_ERROR; + } + } else if (mkdir(opk->control_dir, 0755) != 0) { + fputs(_("Error: Cannot create control directory" + "\n"), stderr); + return OPKG_OPK_ERROR; + } + path = malloc(strlen(opk->control_dir) + + OPKG_OPK_USTAR_NAME_SIZE + 1); + if (path == NULL) { + return OPKG_OPK_ERROR; + } + } else if (opk->list_control == 0 && opk->print_control_head == NULL) { + /* Not listing or printing any control files. */ + return OPKG_OPK_OK; + } + + if (_opkg_opk_opk_init_inner(opk) != OPKG_OPK_OK) { + if (opk->control_dir != NULL) { + free(path); + } + return OPKG_OPK_ERROR; + } + + while ((ret_list = opkg_opk_ustar_list(opk->inner_ustar, &member)) == + OPKG_OPK_OK) { + /* Remove leading "./" if present and skip if nothing's left. */ + buffer = member->name; + if (buffer[0] == '.' && buffer[1] == '/') { + buffer += 2; + } + if (buffer[0] == '\0') { + free(member); + continue; + } + + /* Open file for extraction. */ + if (opk->control_dir != NULL) { + if (sprintf(path, "%s/%s", opk->control_dir, + member->name) <= 0) { + free(member); + _opkg_opk_opk_free_inner(opk); + free(path); + return OPKG_OPK_ERROR; + } + fp = fopen(path, "wb"); + if (fp == NULL) { + fputs(_("Error: Failed to extract control file" + "\n"), stderr); + free(member); + _opkg_opk_opk_free_inner(opk); + free(path); + return OPKG_OPK_ERROR; + } + } + + if (opk->list_control > 0) { + puts(buffer); + opk->previously_printed = 1; + } + + if (opk->print_control_head != NULL && + _opkg_opk_opk_check_name(member->name, + &opk->print_control_head, + &opk->print_control_tail) == + OPKG_OPK_OK) { + /* Name requested for printing. */ + print = 1; + } else { + print = 0; + } + free(member); + if (print == 1 && opk->previously_printed == 1) { + puts(""); + } + while ((ret_read = opkg_opk_ustar_read(opk->inner_ustar, + &buffer, &size)) == OPKG_OPK_OK) + { + if (print == 1 && fwrite(buffer, 1, size, stdout) != + size) { + fputs(_("Error: Failed to print control file\n") + , stderr); + _opkg_opk_opk_free_inner(opk); + if (opk->control_dir != NULL) { + free(path); + } + return OPKG_OPK_ERROR; + } + if (opk->control_dir != NULL && fwrite(buffer, 1, size, + fp) != size) { + fputs(_("Error: Failed to write control file\n") + , stderr); + _opkg_opk_opk_free_inner(opk); + free(path); + return OPKG_OPK_ERROR; + } + } + if (ret_read == OPKG_OPK_ERROR) { + fputs(_("Error: Failed to read control file\n"), + stderr); + _opkg_opk_opk_free_inner(opk); + if (opk->control_dir != NULL) { + free(path); + } + return OPKG_OPK_ERROR; + } + opk->previously_printed = 1; + + /* Close file for extraction. */ + if (opk->control_dir != NULL) { + if (fclose(fp) != 0) { + _opkg_opk_opk_free_inner(opk); + free(path); + return OPKG_OPK_ERROR; + } + } + } + if (opk->control_dir != NULL) { + /* Done with path buffer. */ + free(path); + } + if (ret_list == OPKG_OPK_ERROR) { + fputs(_("Error: Failed to list control files\n"), stderr); + _opkg_opk_opk_free_inner(opk); + return OPKG_OPK_ERROR; + } + + /* Check for files not found. */ + if (opk->print_control_head != NULL) { + for (seek_name = opk->print_control_head; seek_name != NULL; + seek_name = seek_name->next) { + fprintf(stderr, _("Error: Failed to find control file " + "\"%s\"\n"), seek_name->name); + } + _opkg_opk_opk_free_inner(opk); + return OPKG_OPK_ERROR; + } + + _opkg_opk_opk_free_inner(opk); + + return OPKG_OPK_OK; +} + +static int +_opkg_opk_opk_read_data(struct opkg_opk_opk *opk) +{ + struct opkg_opk_ustar_member *head; + struct opkg_opk_ustar_member *tail; + struct opkg_opk_ustar_member *member; + int ret; + size_t uname_len; + size_t uname_len_max; + size_t gname_len; + size_t gname_len_max; + uint64_t size_max; + int ret_read; + char *buffer; + size_t size; + long int size_len_max; + char fmt[26]; + /* "%c%s %-32s/%-32s %11d %s " */ + int len; + char mode[10]; + char mtime[20]; + + if (opk->list_data == 0 && opk->print_data_head == NULL) { + /* Not listing or printing any data files. */ + return OPKG_OPK_OK; + } + + if (_opkg_opk_opk_init_inner(opk) != OPKG_OPK_OK) { + return OPKG_OPK_ERROR; + } + + /* Build singly-linked list and find maximum column widths. */ + head = NULL; + tail = NULL; + uname_len_max = 0; + gname_len_max = 0; + size_max = 0; + while ((ret = opkg_opk_ustar_list(opk->inner_ustar, &member)) == + OPKG_OPK_OK) { + /* If listing, link members and update max column widths. */ + if (opk->list_data > 0) { + if (head == NULL) { + head = member; + } else { + tail->next = member; + } + tail = member; + uname_len = strlen(member->uname); + if (uname_len > uname_len_max) { + uname_len_max = uname_len; + } + gname_len = strlen(member->gname); + if (gname_len > gname_len_max) { + gname_len_max = gname_len; + } + if (member->size > size_max) { + size_max = member->size; + } + } + + if (opk->print_data_head == NULL) { + if (opk->list_data == 0) { + free(member); + } + continue; + } + if (_opkg_opk_opk_check_name(member->name, + &opk->print_data_head, + &opk->print_data_tail) != + OPKG_OPK_OK) { + /* Name not requested for printing. */ + if (opk->list_data == 0) { + free(member); + } + continue; + } + if (opk->list_data == 0) { + free(member); + } + if (opk->previously_printed == 1) { + puts(""); + } + while ((ret_read = opkg_opk_ustar_read(opk->inner_ustar, + &buffer, &size)) == OPKG_OPK_OK) + { + if (fwrite(buffer, 1, size, stdout) != size) { + fputs(_("Error: Failed to print data file\n") + , stderr); + _opkg_opk_opk_free_inner(opk); + return OPKG_OPK_ERROR; + } + } + if (ret_read == OPKG_OPK_ERROR) { + fputs(_("Error: Failed to read data file\n"), + stderr); + _opkg_opk_opk_free_inner(opk); + return OPKG_OPK_ERROR; + } + opk->previously_printed = 1; + } + if (ret == OPKG_OPK_ERROR) { + fputs(_("Error: Failed to list data files\n"), stderr); + _opkg_opk_opk_free_inner(opk); + return OPKG_OPK_ERROR; + } + if (tail != NULL) { + tail->next = NULL; + } + + if (opk->list_data == 0) { + _opkg_opk_opk_free_inner(opk); + return OPKG_OPK_OK; + } + + /* Print and free members. */ + if (opk->previously_printed == 1) { + puts(""); + } + ret = OPKG_OPK_OK; + size_len_max = lrint(ceil(log10(size_max))); + snprintf(fmt, sizeof(fmt), "%%c%%s %%-%zus/%%-%zus %%%lid %%s ", + uname_len_max, gname_len_max, size_len_max); + len = 34 + uname_len_max + gname_len_max + size_len_max; + for (member = head; member != NULL;) { + if (member->mode & 00400) mode[0] = 'r'; else mode[0] = '-'; + if (member->mode & 00200) mode[1] = 'w'; else mode[1] = '-'; + if (member->mode & 00100) mode[2] = 'x'; else mode[2] = '-'; + if (member->mode & 04000) mode[2] = 's'; + if (member->mode & 00040) mode[3] = 'r'; else mode[3] = '-'; + if (member->mode & 00020) mode[4] = 'w'; else mode[4] = '-'; + if (member->mode & 00010) mode[5] = 'x'; else mode[5] = '-'; + if (member->mode & 02000) mode[5] = 's'; + if (member->mode & 00004) mode[6] = 'r'; else mode[6] = '-'; + if (member->mode & 00002) mode[7] = 'w'; else mode[7] = '-'; + if (member->mode & 00001) mode[8] = 'x'; else mode[8] = '-'; + if (member->mode & 01000) mode[8] = 't'; + mode[9] = '\0'; + strftime(mtime, sizeof(mtime), "%Y-%m-%d %H:%M:%S", + localtime(&member->mtime)); +#pragma GCC diagnostic push +#pragma GCC diagnostic ignored "-Wformat-nonliteral" + if (printf(fmt, member->type, mode, + member->uname, member->gname, + member->size, mtime) != len) { + ret = OPKG_OPK_ERROR; + } +#pragma GCC diagnostic pop + /* Print name, ensuring it begins with "/". */ + if (member->name[0] == '.' && member->name[1] == '/') { + if (fputs(member->name + 1, stdout) == EOF) { + ret = OPKG_OPK_ERROR; + } + } else { + if (member->name[0] != '/' && putchar('/') == EOF) { + ret = OPKG_OPK_ERROR; + } + if (fputs(member->name, stdout) == EOF) { + ret = OPKG_OPK_ERROR; + } + } + /* Print link name, if any. */ + if (member->type == 'l' && printf(" -> %s", member->linkname) != + (int) (strlen(member->linkname) + 4)) { + ret = OPKG_OPK_ERROR; + } + if (puts("") == EOF) { + ret = OPKG_OPK_ERROR; + } + head = member; + member = member->next; + free(head); + } + + opk->previously_printed = 1; + + _opkg_opk_opk_free_inner(opk); + + return ret; +} + +int +opkg_opk_opk_read(struct opkg_opk_opk *opk, const char *file_name) +{ + int ret; + struct opkg_opk_ustar_member *member; + char *version_buffer; + size_t version_size; + + ret = OPKG_OPK_OK; + + /* Open outer archive. */ + opk->file = fopen(file_name, "rb"); + if (opk->file == NULL) { + fprintf(stderr, _("Error: Failed to open file \"%s\"\n"), + file_name); + ret = OPKG_OPK_ERROR; + goto out0; + } + + /* Initialize outer gzip decompressor. */ + opk->outer_gzip = opkg_opk_gzip_init_read(&_opkg_opk_opk_file_read, + opk); + if (opk->outer_gzip == NULL) { + fputs(_("Error: Failed to initialize\n"), stderr); + ret = OPKG_OPK_ERROR; + goto out1; + } + + /* Initialize outer ustar unarchiver. */ + opk->outer_ustar = opkg_opk_ustar_init(opk->outer_gzip); + if (opk->outer_ustar == NULL) { + fputs(_("Error: Failed to initialize\n"), stderr); + ret = OPKG_OPK_ERROR; + goto out2; + } + + /* Check package version. */ + if (opkg_opk_ustar_list(opk->outer_ustar, &member) != OPKG_OPK_OK || + strcmp(member->name, "debian-binary") != 0) { + free(member); + fputs(_("Error: Failed to find \"debian-binary\" in archive\n"), + stderr); + ret = OPKG_OPK_ERROR; + goto out3; + } + free(member); + if (opkg_opk_ustar_read(opk->outer_ustar, + &version_buffer, &version_size) != + OPKG_OPK_OK) { + fputs(_("Error: Failed to read \"debian-binary\" in archive\n"), + stderr); + ret = OPKG_OPK_ERROR; + goto out3; + } + if (version_size < 4 || strncmp(version_buffer, "2.", 2) != 0) + { + fputs(_("Error: Unsupported package version\n"), stderr); + ret = OPKG_OPK_ERROR; + goto out3; + } + + /* Read control archive. */ + if (opkg_opk_ustar_list(opk->outer_ustar, &member) != OPKG_OPK_OK || + strcmp(member->name, "control.tar.gz") != 0) { + free(member); + fputs(_("Error: Failed to find \"control.tar.gz\" in archive\n") + , stderr); + return OPKG_OPK_ERROR; + } + free(member); + if (_opkg_opk_opk_read_control(opk) != OPKG_OPK_OK) { + ret = OPKG_OPK_ERROR; + goto out3; + } + + /* Read data archive. */ + if (opkg_opk_ustar_list(opk->outer_ustar, &member) != OPKG_OPK_OK || + strcmp(member->name, "data.tar.gz") != 0) { + free(member); + fputs(_("Error: Failed to find \"data.tar.gz\" in archive\n"), + stderr); + return OPKG_OPK_ERROR; + } + free(member); + if (_opkg_opk_opk_read_data(opk) != OPKG_OPK_OK) { + ret = OPKG_OPK_ERROR; + goto out3; + } + + out3: + opkg_opk_ustar_free(opk->outer_ustar); + out2: + opkg_opk_gzip_free(opk->outer_gzip); + out1: + fclose(opk->file); + out0: + return ret; +} -- cgit v0.9.1