+#define _POSIX_C_SOURCE 200809L
+
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/mman.h>
#include <unistd.h>
#include <fcntl.h>
+#include <errno.h>
#include "zpm.h"
+#include "elf.h"
#include "sha256.h"
+
#if 0
struct zpm {
sqlite3 *db;
return 1;
}
+/* wrapper for sqlite3_exec */
+int zpm_exec(struct zpm *z, const char *sql, int(*callback)(void *, int, char **, char**), void *arg, char **errmsg) {
+ return sqlite3_exec(z->db, sql, callback, arg, errmsg);
+}
+
int zpm_rollback(struct zpm *z) {
char *errstr = 0;
sqlite3_exec(z->db, "rollback;", NULL, NULL, &errstr);
#include "newdb.c"
int zpm_db_initialize(struct zpm *pkg) {
- fprintf(stderr, "initializing zpm database\n");
- switch (sqlite3_exec(pkg->db, createdb, (int (*)(void *,int,char **,char **))0, NULL, NULL)) {
+ //fprintf(stderr, "initializing zpm database\n");
+ char *error;
+ switch (sqlite3_exec(pkg->db, createdb, (int (*)(void *,int,char **,char **))0, NULL, &error)) {
case SQLITE_OK: break;
default:
SQLERROR(sqlite3_errmsg(pkg->db));
+ fprintf(stderr, "error: %s\n", error);
+ sqlite3_free(error);
return 0;
break;
}
appid = zpm_db_pragma(pkg, 1);
dbver = zpm_db_pragma(pkg, 2);
- fprintf(stderr, "db appid = %x, dbver = %d\n", appid, dbver);
+ //fprintf(stderr, "db appid = %x, dbver = %d\n", appid, dbver);
switch (appid) {
case 0: if (!zpm_db_initialize(pkg)) {
sqlite3_close(db);
return 1;
}
+static int zpm_sqlite_vercmp(void *not_used, int lena, const void *a,
+ int lenb, const void *b) {
+ /* not sure what the ints are, possibly string lengths */
+ if (not_used != 0) fprintf(stderr, "sqlite vercmp not_used = %p\n",
+ not_used);
+ if (lena == 0 && lenb > 0) return 1;
+ return zpm_vercmp(a, b);
+}
+
+int zpm_addvercmp(struct zpm *pkg) {
+ return sqlite3_create_collation(
+ pkg->db, "vercmp", SQLITE_UTF8, NULL,
+ zpm_sqlite_vercmp
+ );
+}
+
+/* set package struct variables, database, environment, then command line */
+int zpm_readopts(struct zpm *pkg, int ac, char **av) {
+ char *ev;
+
+ if (!pkg) {
+ return -1;
+ }
+
+ ev = getenv("ZPMPACKAGE");
+ if (ev) {
+ pkg->pkgname = dupstr(ev);
+ }
+ ev = getenv("ZPMPKGREL");
+ if (ev) {
+ pkg->release = strtol(ev, 0, 0);
+ }
+ ev = getenv("ZPMPKGVER");
+ if (ev) {
+ pkg->version = dupstr(ev);
+ }
+
+ /* now, parse the options, return optind so the caller can adjust if needed */
+
+ return av ? ac : 1;
+}
+
int zpm_extract(struct zpm *pkg, char *hash, char *path, int mode) {
int rc;
int blobsize;
- int64_t size;
+ //int64_t size;
void *xzdata;
int type;
FILE *out;
/* didn't find a row */
sqlite3_finalize(ifile);
sqlite3_close(db);
- fprintf(stderr, "no such hash\n");
+ fprintf(stderr, "no such hash: %s\n", hash);
return 0;
}
/* either way we're done with this now */
sqlite3_close(db);
return 0;
}
- size = sqlite3_column_int64(ifile, 0);
+ //size = sqlite3_column_int64(ifile, 0);
xzdata = (void *)sqlite3_column_blob(ifile, 1);
blobsize = sqlite3_column_bytes(ifile, 1);
- out = fopen(path, "w");
+ if (strcmp(path, "-")) {
+ out = fopen(path, "w");
+ } else {
+ out = stdout;
+ }
if (!out) {
fprintf(stderr, "can't open output file %s\n", path);
sqlite3_finalize(ifile);
sqlite3_close(db);
- return 5;
+ return 0;
}
//fwrite(xzdata, blobsize, 1, stdout);
- fprintf(stderr, "uncompressing %d bytes at %p, expect %lld\n", blobsize, xzdata, (long long int)size);
+ //fprintf(stderr, "uncompressing %d bytes at %p, expect %lld\n", blobsize, xzdata, (long long int)size);
uncompresslzma(xzdata, blobsize, out);
fclose(out);
+ chmod(path, mode);
sqlite3_finalize(ifile);
- return 0;
-
+ return 1;
+}
+
+/* flags 0, close mmap, flags 1, return mmap fd */
+int zpm_hash(char *path, char *hash, uint32_t flags) {
+ int fd;
+ void *content;
+ struct stat sbuf;
+ struct sha256_state md;
+ int j;
+ unsigned char tmp[32];
+
+ /* mmap the file */
+ fd = open(path, O_RDONLY);
+ if (fd == -1) {
+ fprintf(stderr, "%s can't open %s: %s\n", __FUNCTION__, path,strerror(errno));
+ return 0;
+ }
+ if (fstat(fd, &sbuf) == -1) {
+ fprintf(stderr, "%s can't fstat %s: %s\n", __FUNCTION__, path,strerror(errno));
+ return 0;
+ }
+ /* not a regular file? */
+ if (!S_ISREG(sbuf.st_mode)) {
+ /* TODO this is ok, just stored differently */
+ fprintf(stderr, "%s non-regular files unsupported %s\n", __FUNCTION__, path);
+ return 0;
+ }
+
+ content = mmap(0, sbuf.st_size, PROT_READ,MAP_PRIVATE, fd, 0);
+ close(fd);
+ if (!content) {
+ fprintf(stderr, "%s can't mmap %s: %s\n", __FUNCTION__, path,strerror(errno));
+ return 0;
+ }
+
+ /* get hash */
+ sha256_init(&md);
+ sha256_process(&md, content, sbuf.st_size);
+ sha256_done(&md, tmp);
+ for (j=0;j<32;j++) {
+ sprintf(hash+j*2, "%02x", (unsigned)tmp[j]);
+ }
+ hash[64] = 0;
+ munmap(content, sbuf.st_size);
+ return flags ? fd : 1;
+}
+
+static sqlite3_stmt *run_for_hash(sqlite3 *db, char *sql, char *hash) {
+ int rc;
+ sqlite3_stmt *ifile;
+
+ rc = sqlite3_prepare_v2(db, sql, -1, &ifile, 0);
+ if (rc != SQLITE_OK) {
+ SQLERROR(sqlite3_errmsg(db));
+ return 0;
+ }
+
+ /* hash, filename */
+
+ sqlite3_bind_text(ifile, 1, hash, 64, SQLITE_STATIC);
+
+ return ifile;
+}
+
+static int set_elf_info(sqlite3 *db, char *hash, char *content, size_t length) {
+ if (length >= sizeof (Elf64_Ehdr) && libelf_iself(content)) {
+ char *strtab;
+ Elf64_Dyn *dyn;
+ int i;
+ Elf64_Phdr *phdr;
+ Elf64_Ehdr *hdr;
+ sqlite3_stmt *ifile;
+ int rc;
+
+ /* go ahead and set up elf information now */
+ /* clear existing for this hash */
+ ifile = run_for_hash(db, "delete from elfinfo where file = ?", hash);
+ do {
+ rc = sqlite3_step(ifile);
+#if 0
+ if (rc == SQLITE_ROW) {
+ int nc;
+ fprintf(stderr, "delete row has %d columns: ", sqlite3_column_count(ifile));
+ nc = sqlite3_column_count(ifile);
+ for (i = 0; i < nc; i++) {
+ char *r;
+ r = sqlite3_column_text(ifile, i);
+ fprintf(stderr, ", %s", r);
+ }
+ fprintf(stderr, "\n");
+ }
+#endif
+ } while (rc == SQLITE_ROW);
+ if (rc != SQLITE_DONE) {
+ SQLERROR(sqlite3_errmsg(db));
+ sqlite3_finalize(ifile);
+ fprintf(stderr, "error clearing elf info: %d\n", rc);
+ return 0;
+ }
+ sqlite3_finalize(ifile);
+ ifile = run_for_hash(db, "delete from elflibraries where file = ?", hash);
+ do {
+ rc = sqlite3_step(ifile);
+ } while (rc == SQLITE_ROW);
+ if (rc != SQLITE_DONE) {
+ SQLERROR(sqlite3_errmsg(db));
+ sqlite3_finalize(ifile);
+ fprintf(stderr, "error clearing elf library: %d\n", rc);
+ return 0;
+ }
+ sqlite3_finalize(ifile);
+ ifile = run_for_hash(db, "delete from elfneeded where file = ?", hash);
+ do {
+ rc = sqlite3_step(ifile);
+ } while (rc == SQLITE_ROW);
+ if (rc != SQLITE_DONE) {
+ SQLERROR(sqlite3_errmsg(db));
+ sqlite3_finalize(ifile);
+ fprintf(stderr, "error clearing elf needed\n");
+ return 0;
+ }
+ sqlite3_finalize(ifile);
+
+ hdr = libelf_header(content);
+ /* if lib, set soname */
+ if (libelf_type(content) == ET_DYN) {
+ char *soname = libelf_soname(content);
+ if (soname) {
+
+ sqlite3_prepare_v2(db, "insert into elflibraries (file,soname) values (?,?)",-1, &ifile, 0);
+ sqlite3_bind_text(ifile,1,hash,64,SQLITE_STATIC);
+ sqlite3_bind_text(ifile,2,soname,-1,SQLITE_STATIC);
+ rc = sqlite3_step(ifile);
+ if (rc != SQLITE_DONE) {
+ SQLERROR(sqlite3_errmsg(db));
+ sqlite3_finalize(ifile);
+ fprintf(stderr, "error setting library soname\n");
+ return 0;
+ }
+ sqlite3_finalize(ifile);
+ } else {
+ fprintf(stderr, "can't find soname\n");
+ }
+ }
+
+ /* if exe, set neededs */
+ if (libelf_type(content) == ET_EXEC) {
+ Elf64_Shdr *dsect;
+ char *elf;
+
+ elf = (char *)content;
+ /* find program header table */
+ for (i = 0; i < hdr->e_phnum; i++) {
+ phdr = (Elf64_Phdr *)(elf + hdr->e_phoff + i * hdr->e_phentsize);
+ if (phdr->p_type == PT_DYNAMIC) {
+ dsect = (Elf64_Shdr *)(elf + phdr->p_offset);
+ }
+ }
+ dyn = (Elf64_Dyn *)(elf + dsect->sh_offset);
+ if (!dyn) {
+ exit(9);
+ }
+ dyn = (Elf64_Dyn *)dsect;
+
+ dsect = libelf_section(elf, SHT_DYNAMIC);
+ Elf64_Shdr *strsect;
+
+ strsect = libelf_section_n(elf, dsect->sh_link);
+ strtab = elf + strsect->sh_offset;
+
+ sqlite3_prepare_v2(db, "insert into elfneeded (file,needed) values (?,?)",-1, &ifile, 0);
+ sqlite3_bind_text(ifile,1,hash,64,SQLITE_STATIC);
+ while (dyn->d_tag != DT_NULL) {
+ if (dyn->d_tag == DT_NEEDED) {
+ char *need;
+ int rc;
+
+ need = strtab + dyn->d_un.d_val;
+ if (strlen(need) == 0) continue;
+ sqlite3_bind_text(ifile,2,need,strlen(need),SQLITE_STATIC);
+ fprintf(stderr, "%s needs %s\n", hash, need);
+ rc = sqlite3_step(ifile);
+ if (rc != SQLITE_DONE) {
+ SQLERROR(sqlite3_errmsg(db));
+ sqlite3_finalize(ifile);
+ fprintf(stderr, "error setting needed library\n");
+ return 0;
+ }
+ sqlite3_reset(ifile);
+ }
+ dyn++;
+ }
+ sqlite3_finalize(ifile);
+ }
+ }
+ return 1;
}
#if 1
void *content;
struct stat sbuf;
unsigned char tmp[32];
- hash_state md;
+ struct sha256_state md;
sqlite3_stmt *ifile;
int haverow = 0,havedata = 0;
int j,rc,type;
hash = hashbuf;
}
+ if (flags) {
+ fprintf(stderr, "zpm_import unused flags = %d\n", flags);
+ }
/* mmap the file */
fd = open(path, O_RDONLY);
if (fd == -1) {
+ pkg->error = errno;
+ fprintf(stderr, "%s can't open %s: %s\n", __FUNCTION__, path,strerror(errno));
return 0;
}
if (fstat(fd, &sbuf) == -1) {
+ pkg->error = errno;
+ fprintf(stderr, "%s can't fstat %s: %s\n", __FUNCTION__, path,strerror(errno));
return 0;
}
/* not a regular file? */
if (!S_ISREG(sbuf.st_mode)) {
+ char *ftype;
+ switch (sbuf.st_mode & S_IFMT) {
+ case S_IFSOCK: ftype = "socket"; break;
+ case S_IFLNK : ftype = "symlink"; break;
+ case S_IFBLK : ftype = "block device"; break;
+ case S_IFDIR : ftype = "directory"; break;
+ case S_IFCHR : ftype = "character device"; break;
+ case S_IFIFO : ftype = "fifo"; break;
+ default: ftype = "unknown file type"; break;
+ }
/* TODO this is ok, just stored differently */
+ fprintf(stderr, "%s can't import %s file: %s\n", __FUNCTION__, ftype, path);
+ pkg->error = EINVAL;
return 0;
}
content = mmap(0, sbuf.st_size, PROT_READ,MAP_PRIVATE, fd, 0);
+ close(fd);
if (!content) {
+ pkg->error = errno;
+ fprintf(stderr, "%s can't mmap %s: %s\n", __FUNCTION__, path,strerror(errno));
return 0;
}
sprintf(hash+j*2, "%02x", (unsigned)tmp[j]);
}
hash[64] = 0;
- fprintf(stderr, "file %s: %s\n", path, hash);
+ //fprintf(stderr, "file %s: %s\n", path, hash);
-
- /* prepare and bind */
/* TODO check null */
sqlite3 *db = pkg->db;
+ /* prepare and bind */
+
rc = sqlite3_prepare_v2(db, "select size, content is not null from files where hash = ?", -1, &ifile,0);
if (rc != SQLITE_OK) {
SQLERROR(sqlite3_errmsg(db));
+ munmap(content, sbuf.st_size);
return 0;
}
/* didn't find a row */
SQLERROR(sqlite3_errmsg(db));
zpm_rollback(pkg);
+ munmap(content, sbuf.st_size);
return 0;
}
haverow = 1;
/* TODO assert, this shouldn't be possible? */
fprintf(stderr, "no file size\n");
sqlite3_finalize(ifile);
+ munmap(content, sbuf.st_size);
return 0;
}
type = sqlite3_column_type(ifile, 1);
/* TODO assert, this shouldn't be possible? */
fprintf(stderr, "no file data\n");
sqlite3_finalize(ifile);
+ munmap(content, sbuf.st_size);
return 0;
/* which is fine, just need to update the row then */
}
outbuf = compresslzma(content, sbuf.st_size, &outlen);
if (!outbuf) {
fprintf(stderr, "compresslzma failed\n");
+ munmap(content, sbuf.st_size);
return 0;
}
- fprintf(stderr, "compressed to %zu\n", outlen);
- /* don't need the original file now */
- munmap(content, sbuf.st_size);
- close(fd);
+ //fprintf(stderr, "compressed to %zu\n", outlen);
/* start a transaction */
// do that outside of here
/* insert */
if (haverow) {
- fprintf(stderr, "adding file data\n");
+ //fprintf(stderr, "adding file data\n");
rc = sqlite3_prepare(db, "update files set size = ?, content = ? where hash = ?", -1, &ifile,0);
} else {
- fprintf(stderr, "creating new data row\n");
+ //fprintf(stderr, "creating new data row\n");
rc = sqlite3_prepare(db, "insert into files (size, content, hash) values (?,?,?)", -1, &ifile,0);
}
if (rc != SQLITE_OK) {
SQLERROR(sqlite3_errmsg(db));
fprintf(stderr, "cant prepare data\n");
zpm_rollback(pkg);
+ munmap(content, sbuf.st_size);
return 0;
}
SQLERROR(sqlite3_errmsg(db));
fprintf(stderr, "cant bind size\n");
zpm_rollback(pkg);
+ munmap(content, sbuf.st_size);
return 0;
}
sqlite3_bind_blob64(ifile, 2, outbuf, (sqlite3_int64)outlen, SQLITE_STATIC);
SQLERROR(sqlite3_errmsg(db));
fprintf(stderr, "cant bind content\n");
zpm_rollback(pkg);
+ munmap(content, sbuf.st_size);
return 0;
}
sqlite3_bind_text(ifile, 3, hash, 64, SQLITE_STATIC);
SQLERROR(sqlite3_errmsg(db));
fprintf(stderr, "cant bind hash\n");
zpm_rollback(pkg);
+ munmap(content, sbuf.st_size);
return 0;
}
rc = sqlite3_step(ifile);
SQLERROR(sqlite3_errmsg(db));
sqlite3_finalize(ifile);
zpm_rollback(pkg);
+ munmap(content, sbuf.st_size);
return 0;
}
sqlite3_finalize(ifile);
/* commit */
//zpm_commit(pkg);
- } else {
/* don't need the original file now */
+
+ }
+
+ if (!set_elf_info(pkg->db, hash, content, sbuf.st_size)) {
+ fprintf(stderr, "setting elf info failed\n");
munmap(content, sbuf.st_size);
- close(fd);
+ return 0;
}
+ munmap(content, sbuf.st_size);
/* if package and not nopackage flag, add to package */
if (pkg->pkgname && (!ZPM_NOPACKAGE)) {
projects / zpackage / blobdiff