Subversion Repositories SvarDOS

/*

 * This file is part of pkginst (SvarDOS)

 * Copyright (C) 2012-2021 Mateusz Viste

 * Changes by TK Chia

 */

#include <ctype.h>     /* toupper() */

#include <stdio.h>

#include <stdlib.h>    /* system() */

#include <string.h>    /* strcpy() */

#include <unistd.h>    /* read() */

#include <sys/types.h> /* struct utimbuf */

#include "crc32.h"     /* all CRC32 related stuff */

#include "helpers.h"   /* slash2backslash(), strtolower() */

#include "fileexst.h"

#include "kprintf.h"

#include "libunzip.h"  /* zip_listfiles()... */

#include "showinst.h"  /* pkg_loadflist() */

#include "pkginst.h"   /* include self for control */

#include "version.h"

/* return 1 if fname looks like a link filename, 0 otherwise */

static int islinkfile(const char *fname) {

  char *link1 = "LINKS\\";

  char *link2 = "links\\";

  int x;

  for (x = 0; ; x++) {

    if (link1[x] == 0) return(1);

    if ((fname[x] != link1[x]) && (fname[x] != link2[x])) return(0);

  }

}

/* validate a filename (8+3, no weird characters, etc). returns 0 on success,

 * nonzero otherwise. */

static int validfilename(const char *fname) {

  int i, i2;

  char *validchars = "!#$%&'()-@^_`{}~";

  int elemlen = 0;

  int elemmaxlen = 8; /* switches from 8 to 3 depending wheter I am analyzing

                         a filename or an extension */

  /* look for invalid chars in the entire string, and check the length of the

   * element at the same time */

  for (i = 0; fname[i] != 0; i++) {

    /* director separators are threated specially */

    if (fname[i] == '\\') {

      elemlen = 0;

      elemmaxlen = 8;

      continue;

    }

    /* '.' switches the check into extension mode */

    if (fname[i] == '.') {

      if (elemlen == 0) return(-1); /* a file must have a base name */

      if (elemmaxlen == 3) return(-2); /* a file cannot have two extensions */

      elemlen = 0;

      elemmaxlen = 3;

      continue;

    }

    /* check that the element is not too long */

    if (++elemlen > elemmaxlen) return(-3);

    /* look for valid characters */

    if ((fname[i] >= 'a') && (fname[i] <= 'z')) continue;

    if ((fname[i] >= 'A') && (fname[i] <= 'Z')) continue;

    if ((fname[i] >= '0') && (fname[i] <= '9')) continue;

    if ((fname[i] & 128) != 0) continue; /* high bytes are okay (NLS) */

    /* look for other valid characters */

    for (i2 = 0; validchars[i2] != 0; i2++) {

      if (fname[i] == validchars[i2]) break;

    }

    if (validchars[i2] != 0) continue;

    /* if we are here, then the character is invalid */

    return(-4);

  }

  /* all checks passed */

  return(0);

}

/* processes a link file - that is, reads the target inside, and overwrite

 * the file with new content */

static void processlinkfile(const char *linkfile, const char *dosdir, const struct customdirs *dirlist, char *buff) {

  char origtarget[512];

  int x;

  char *shortfile;

  unsigned char comstub[] = { /* machine code of a COM stub launcher */

    0xBC,0x00,0x00,0xBB,0x00,0x10,0xB4,0x4A,0xCD,0x21,0xBB,0x2A,0x01,0x8C,0x5F,0x04,

    0x8C,0x5F,0x08,0x8C,0x5F,0x0C,0xB8,0x00,0x4B,0xBA,0x38,0x01,0xCD,0x21,0xB0,0x7F,

    0x72,0x04,0xB4,0x4D,0xCD,0x21,0xB4,0x4C,0xCD,0x21,0x00,0x00,0x80,0x00,0x00,0x00,

    0x5C,0x00,0x00,0x00,0x6C,0x00,0x00,0x00}; /* read comlink.asm for details */

  const unsigned stack_paras = 12;

  unsigned alloc_paras, alloc_bytes;

  FILE *fd;

  /* open the link file and read the original target */

  fd = fopen(linkfile, "r");

  if (fd == NULL) {

    kitten_printf(3, 21, "Error: Failed to open link file '%s' for read access.", linkfile);

    puts("");

    return;

  }

  x = fread(origtarget, 1, sizeof(origtarget) - 1, fd);

  origtarget[x] = 0;

  fclose(fd);

  /* validate the original target (ltrim to first \r or \n) */

  for (x = 0; origtarget[x] != 0; x++) {

    if ((origtarget[x] == '\r') || (origtarget[x] == '\n')) {

      origtarget[x] = 0;

      break;

    }

  }

  /* translate the original target to a local path */

  shortfile = computelocalpath(origtarget, buff, dosdir, dirlist);

  /* compute the amount of memory the stub should leave for itself:

	- 16 paragraphs needed for the PSP

	- sizeof(comstub) bytes needed for code and data

	- strlen(shortfile) + 1 for the command line

	- stack_paras paragraphs for the stack */

  alloc_paras = 16 + (sizeof(comstub) + strlen(shortfile) + 16) / 16 + stack_paras;

  alloc_bytes = 16 * alloc_paras;

  comstub[1] = alloc_bytes & 0xff;

  comstub[2] = alloc_bytes >> 8;

  comstub[4] = alloc_paras & 0xff;

  comstub[5] = alloc_paras >> 8;

  /* write new content into the link file */

  fd = fopen(linkfile, "wb");

  if (fd == NULL) {

    kitten_printf(3, 22, "Error: Failed to open link file '%s' for write access.", linkfile);

    puts("");

    return;

  }

  fwrite(comstub, 1, sizeof(comstub), fd);

  fprintf(fd, "%s%s%c", buff, shortfile, 0);

  fclose(fd);

}

/* returns 0 if pkgname is not installed, non-zero otherwise */

int is_package_installed(const char *pkgname, const char *dosdir) {

  char fname[512];

  sprintf(fname, "%s\\packages\\%s.lst", dosdir, pkgname);

  return(fileexists(fname)); /* file exists -> package is installed */

}

/* checks that pkgname is NOT installed. return 0 on success, non-zero otherwise. */

static int validate_package_not_installed(const char *pkgname, const char *dosdir) {

  if (is_package_installed(pkgname, dosdir) != 0) {

    kitten_printf(3, 18, "Package %s is already installed! You might want to use the 'update' action.", pkgname);

    puts("");

    return(-1);

  }

  return(0);

}

/* find a filename in a flist linked list, and returns a pointer to it */

static struct flist_t *findfileinlist(struct flist_t *flist, const char *fname) {

  while (flist != NULL) {

    if (strcmp(flist->fname, fname) == 0) return(flist);

    flist = flist->next;

  }

  return(NULL);

}

/* prepare a package for installation. this is mandatory before actually installing it!

 * returns a pointer to the zip file's index on success, NULL on failure. the **zipfd pointer is updated with a file descriptor to the open zip file to install. */

struct ziplist *pkginstall_preparepackage(const char *pkgname, const char *zipfile, int flags, FILE **zipfd, const char *dosdir, const struct customdirs *dirlist) {

  char fname[256];

  char appinfofile[256];

  int appinfopresence;

  char *shortfile;

  struct ziplist *ziplinkedlist = NULL, *curzipnode, *prevzipnode;

  struct flist_t *flist = NULL;

  sprintf(appinfofile, "appinfo\\%s.lsm", pkgname); /* Prepare the appinfo/xxxx.lsm filename string for later use */

  /* check if not already installed, if already here, print a message "you might want to use update instead"

   * of course this must not be done if we are in the process of upgrading said package */

  if (((flags & PKGINST_UPDATE) == 0) && (validate_package_not_installed(pkgname, dosdir) != 0)) {

    return(NULL);

  }

  /* Now let's check the content of the zip file */

  *zipfd = fopen(zipfile, "rb");

  if (*zipfd == NULL) {

    kitten_puts(3, 8, "Error: Invalid zip archive! Package not installed.");

    goto RAII;

  }

  ziplinkedlist = zip_listfiles(*zipfd);

  if (ziplinkedlist == NULL) {

    kitten_puts(3, 8, "Error: Invalid zip archive! Package not installed.");

    goto RAII;

  }

  /* if updating, load the list of files belonging to the current package */

  if ((flags & PKGINST_UPDATE) != 0) {

    flist = pkg_loadflist(pkgname, dosdir);

  }

  /* Verify that there's no collision with existing local files, look for the appinfo presence, get rid of sources if required, and rename BAT links into COM files */

  appinfopresence = 0;

  prevzipnode = NULL;

  for (curzipnode = ziplinkedlist; curzipnode != NULL;) {

    /* change all slashes to backslashes, and switch into all-lowercase */

    slash2backslash(curzipnode->filename);

    strtolower(curzipnode->filename);

    /* remove 'directory' ZIP entries to avoid false alerts about directory already existing */

    if ((curzipnode->flags & ZIP_FLAG_ISADIR) != 0) {

      curzipnode->filename[0] = 0; /* mark it "empty", will be removed in a short moment */

    }

    /* is it a "link file"? */

    if (fdnpkg_strcasestr(curzipnode->filename, "links\\") == curzipnode->filename) {

      /* skip links, if that's what the user wants */

      if ((flags & PKGINST_SKIPLINKS) != 0) {

        curzipnode->filename[0] = 0; /* in fact, we just mark the file as 'empty' on the filename.. see later below */

      } else {

        /* if it's a *.BAT link, then rename it to *.COM */

        char *ext = getfext(curzipnode->filename);

        if (strcasecmp(ext, "bat") == 0) sprintf(ext, "com");

      }

    }

    if (curzipnode->filename[0] == 0) { /* ignore empty filenames (maybe it was empty originally, or has been emptied because it's a dropped source or link) */

      if (prevzipnode == NULL) {  /* take the item out of the list */

        ziplinkedlist = curzipnode->nextfile;

        free(curzipnode); /* free the item */

        curzipnode = ziplinkedlist;

      } else {

        prevzipnode->nextfile = curzipnode->nextfile;

        free(curzipnode); /* free the item */

        curzipnode = prevzipnode->nextfile;

      }

      continue; /* go to the next item */

    }

    /* validate that the file has a valid filename (8+3, no shady chars...) */

    if (validfilename(curzipnode->filename) != 0) {

      kitten_puts(3, 23, "Error: Package contains an invalid filename:");

      printf(" %s\n", curzipnode->filename);

      goto RAII_ERR;

    }

    /* look out for collisions with already existing files (unless we are

     * updating the package and the local file belongs to it */

    shortfile = computelocalpath(curzipnode->filename, fname, dosdir, dirlist);

    strcat(fname, shortfile);

    if ((findfileinlist(flist, fname) == NULL) && (fileexists(fname) != 0)) {

      kitten_puts(3, 9, "Error: Package contains a file that already exists locally:");

      printf(" %s\n", fname);

      goto RAII_ERR;

    }

    /* abort if any entry is encrypted */

    if ((curzipnode->flags & ZIP_FLAG_ENCRYPTED) != 0) {

      kitten_printf(3, 20, "Error: Package contains an encrypted file:");

      puts("");

      printf(" %s\n", curzipnode->filename);

      goto RAII_ERR;

    }

    /* abort if any file is compressed with an unsupported method */

    if ((curzipnode->compmethod != 0/*store*/) && (curzipnode->compmethod != 8/*deflate*/)) { /* unsupported compression method */

      kitten_printf(8, 2, "Error: Package contains a file compressed with an unsupported method (%d):", curzipnode->compmethod);

      puts("");

      printf(" %s\n", curzipnode->filename);

      goto RAII_ERR;

    }

    if (strcmp(curzipnode->filename, appinfofile) == 0) appinfopresence = 1;

    prevzipnode = curzipnode;

    curzipnode = curzipnode->nextfile;

  }

  /* if appinfo file not found, this is not a real FreeDOS package */

  if (appinfopresence != 1) {

    kitten_printf(3, 12, "Error: Package do not contain the %s file! Not a valid FreeDOS package.", appinfofile);

    puts("");

    goto RAII_ERR;

  }

  goto RAII;

  RAII_ERR:

  zip_freelist(&ziplinkedlist);

  ziplinkedlist = NULL;

  if ((zipfd != NULL) && (*zipfd != NULL)) {

    fclose(*zipfd);

    *zipfd = NULL;

  }

  RAII:

  pkg_freeflist(flist);

  return(ziplinkedlist);

}

/* install a package that has been prepared already. returns 0 on success,

 * or a negative value on error, or a positive value on warning */

int pkginstall_installpackage(const char *pkgname, const char *dosdir, const struct customdirs *dirlist, struct ziplist *ziplinkedlist, FILE *zipfd) {

  char *buff;

  char *fulldestfilename;

  char packageslst[64];

  char *shortfile;

  long filesextractedsuccess = 0, filesextractedfailure = 0;

  struct ziplist *curzipnode;

  FILE *lstfd;

  sprintf(packageslst, "packages\\%s.lst", pkgname); /* Prepare the packages/xxxx.lst filename string for later use */

  buff = malloc(512);

  fulldestfilename = malloc(1024);

  if ((buff == NULL) || (fulldestfilename == NULL)) {

    kitten_puts(8, 0, "Out of memory!");

    zip_freelist(&ziplinkedlist);

    free(buff);

    free(fulldestfilename);

    return(-1);

  }

  /* create the %DOSDIR%/packages directory, just in case it doesn't exist yet */

  sprintf(buff, "%s\\packages\\", dosdir);

  mkpath(buff);

  /* open the lst file */

  sprintf(buff, "%s\\%s", dosdir, packageslst);

  lstfd = fopen(buff, "wb"); /* opening it in binary mode, because I like to have control over line terminators (CR/LF) */

  if (lstfd == NULL) {

    kitten_printf(3, 10, "Error: Could not create %s!", buff);

    puts("");

    zip_freelist(&ziplinkedlist);

    free(buff);

    free(fulldestfilename);

    return(-2);

  }

  /* write list of files in zip into the lst, and create the directories structure */

  for (curzipnode = ziplinkedlist; curzipnode != NULL; curzipnode = curzipnode->nextfile) {

    int unzip_result;

    if ((curzipnode->flags & ZIP_FLAG_ISADIR) != 0) continue; /* skip directories */

    if (strcasecmp(curzipnode->filename, packageslst) == 0) continue; /* skip silently the package.lst file, if found */

    shortfile = computelocalpath(curzipnode->filename, buff, dosdir, dirlist); /* substitute paths to custom dirs */

    /* log the filename to packages\pkg.lst (with original, unmapped drive) */

    fprintf(lstfd, "%s%s\r\n", buff, shortfile);

    /* create the path, just in case it doesn't exist yet */

    mkpath(buff);

    sprintf(fulldestfilename, "%s%s", buff, shortfile);

    /* Now unzip the file */

    unzip_result = zip_unzip(zipfd, curzipnode, fulldestfilename);

    if (unzip_result != 0) {

      kitten_printf(8, 3, "Error while extracting '%s' to '%s'!", curzipnode->filename, fulldestfilename);

      printf(" [%d]\n", unzip_result);

      filesextractedfailure += 1;

    } else {

      printf(" %s -> %s\n", curzipnode->filename, buff);

      filesextractedsuccess += 1;

    }

    /* if it's a LINK file, recompute a new content */

    if (islinkfile(curzipnode->filename) != 0) {

      processlinkfile(fulldestfilename, dosdir, dirlist, buff);

    }

  }

  fclose(lstfd);

  /* free the ziplist and close file descriptor */

  zip_freelist(&ziplinkedlist);

  free(buff);

  free(fulldestfilename);

  kitten_printf(3, 19, "Package %s installed: %ld files extracted, %ld errors.", pkgname, filesextractedsuccess, filesextractedfailure);

  puts("");

  return(filesextractedfailure);

}