#include "newdb.c"
int zpm_db_initialize(struct zpm *pkg) {
- //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;
SQLERROR(sqlite3_errmsg(pkg->db));
fprintf(stderr, "error: %s\n", error);
sqlite3_free(error);
+ zpm_rollback(pkg);
return 0;
break;
}
return zpm;
}
+static void zpm_set_db_errmsg(struct zpm *zpm, const char *msg) {
+ if (zpm) {
+ if (zpm->dberrmsg) {
+ free(zpm->dberrmsg);
+ }
+ if (msg) {
+ zpm->dberrmsg = strdup(msg);
+ if (!zpm->dberrmsg) {
+ zpm->error = 1;
+ }
+ } else {
+ zpm->dberrmsg = 0;
+ }
+ }
+}
+
int zpm_init(struct zpm *pkg, char *path) {
int rc;
sqlite3 *db = 0;
int appid;
+ if (!pkg) {
+ return 0;
+ }
+
zpm_clearmem(pkg);
rc = sqlite3_open_v2(path, &db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL);
if (rc) {
- SQLERROR(sqlite3_errmsg(db));
- sqlite3_close(db);
+ pkg->error = 1;
+ pkg->dbresult = rc;
+ zpm_set_db_errmsg(pkg, sqlite3_errstr(rc));
+ if (db) {
+ sqlite3_close(db);
+ }
+ fprintf(stderr, "error (%d): %s: %s\n", rc,
+ pkg->dberrmsg ? pkg->dberrmsg : "null", path);
+
return 0;
}
pkg->db = db;
pkg->db = 0;
return 0;
}
+ zpm_addvercmp(pkg);
return 1;
}
zpm->error = 1;
return 0;
}
+ zpm_addvercmp(zpm);
return 1;
}
);
}
-int zpm_extract(struct zpm *pkg, char *hash, char *path, int mode) {
+int zpm_extract(struct zpm *pkg, char *hash, char *path, mode_t mode) {
int rc;
-
int blobsize;
- //int64_t size;
void *xzdata;
int type;
- FILE *out;
+ int out;
sqlite3_stmt *ifile;
+ size_t len;
+ char *tmpfile = 0;
+ sqlite3 *db;
- /* TODO check null */
- sqlite3 *db = pkg->db;
+ if (!pkg || !pkg->db) {
+ return 0;
+ }
+ db = pkg->db;
rc = sqlite3_prepare(db, "select size, content from files where hash = ?", -1, &ifile,0);
if (rc != SQLITE_OK) {
blobsize = sqlite3_column_bytes(ifile, 1);
if (strcmp(path, "-")) {
- out = fopen(path, "w");
+ len = strlen(path);
+ tmpfile = malloc(len+8+1);
+ if (!tmpfile) {
+ fprintf(stderr, "malloc error\n");
+ return 0;
+ }
+ sprintf(tmpfile, "%sXXXXXX", path);
+
+ out = open(tmpfile, O_CREAT|O_WRONLY|O_TRUNC, 0600);
+ if (out == -1) {
+ fprintf(stderr, "can't open output file %s: %s\n",
+ tmpfile, strerror(errno));
+ sqlite3_finalize(ifile);
+ sqlite3_close(db);
+ return 0;
+ }
} else {
- out = stdout;
- }
- if (!out) {
- fprintf(stderr, "can't open output file %s\n", path);
- sqlite3_finalize(ifile);
- sqlite3_close(db);
- return 0;
+ out = 1;
}
- //fwrite(xzdata, blobsize, 1, stdout);
- //fprintf(stderr, "uncompressing %d bytes at %p, expect %lld\n", blobsize, xzdata, (long long int)size);
+#if 0
+ fprintf(stderr, "uncompressing %d bytes at %p, expect %lld\n", blobsize, xzdata, (long long int)size);
+#endif
uncompresslzma(xzdata, blobsize, out);
- fclose(out);
- chmod(path, mode);
-
+ close(out);
sqlite3_finalize(ifile);
- return 1;
+ rc = 1;
+ if (tmpfile) {
+ if (chmod(tmpfile, mode) == -1) {
+ fprintf(stderr, "can't chmod %s: %s\n", tmpfile,
+ strerror(errno));
+ rc = 0;
+ } else if (rename(tmpfile, path) == -1) {
+ fprintf(stderr, "extract rename failed: %s\n",
+ strerror(errno));
+ rc = 0;
+ }
+ }
+
+ if (rc == 0 && tmpfile) {
+ if (unlink(tmpfile) == -1) {
+ fprintf(stderr, "unlink tmpfile %s fail: %s\n",
+ tmpfile,
+ strerror(errno));
+ }
+ }
+
+ return rc;
}
static int run_for_hash(sqlite3 *db, char *sql, char *hash) {
int rc;
/* clear existing for this hash */
- if (!run_for_hash(db, "delete from elfinfo where file = ?", hash)) {
- SQLERP(db, "error clearing elf info");
- return 0;
- }
if (!run_for_hash(db, "delete from elflibraries where file = ?", hash)) {
SQLERP(db, "error clearing elf library");
/* if exe, set neededs */
if (libelf_type(content) == ET_EXEC) {
- Elf64_Shdr *dsect;
+ Elf64_Shdr *dsect = 0;
char *elf;
elf = (char *)content;
dsect = (Elf64_Shdr *)(elf + phdr->p_offset);
}
}
+ if (!dsect) {
+ /* no dynamic section found */
+ return 1;
+ }
+
+#if 0
dyn = (Elf64_Dyn *)(elf + dsect->sh_offset);
if (!dyn) {
exit(9);
}
+#endif
dyn = (Elf64_Dyn *)dsect;
dsect = libelf_section(elf, SHT_DYNAMIC);
return 1;
}
-#if 1
int zpm_import(struct zpm *zpm, char *path, uint32_t flags, char *hash) {
int fd;
- void *content;
+ void *content = 0;
struct stat sbuf;
unsigned char tmp[32];
struct sha256_state md;
- sqlite3_stmt *ifile;
+ sqlite3_stmt *ifile = 0;
int haverow = 0,havedata = 0;
int j,rc,type;
char hashbuf[65];
sprintf(hash+j*2, "%02x", (unsigned)tmp[j]);
}
hash[64] = 0;
- //fprintf(stderr, "file %s: %s\n", path, hash);
/* TODO check null */
sqlite3 *db = zpm->db;
if (rc != SQLITE_ROW) {
/* didn't find a row */
SQLERROR(sqlite3_errmsg(db));
- zpm_rollback(zpm);
munmap(content, sbuf.st_size);
return 0;
}
haverow = 1;
-// fprintf(stderr, "have row for hash\n");
type = sqlite3_column_type(ifile, 0);
if (type == SQLITE_NULL) {
/* TODO assert, this shouldn't be possible? */
outbuf = compresslzma(content, sbuf.st_size, &outlen);
if (!outbuf) {
fprintf(stderr, "compresslzma failed\n");
- munmap(content, sbuf.st_size);
+ munmap(content, sbuf.st_size);
return 0;
}
- //fprintf(stderr, "compressed to %zu\n", outlen);
-
- /* start a transaction */
- // do that outside of here
- //zpm_begin(zpm);
/* insert */
if (haverow) {
- //fprintf(stderr, "adding file data\n");
- rc = sqlite3_prepare(db, "update files set size = ?, content = ? where hash = ?", -1, &ifile,0);
+ rc = sqlite3_prepare_v2(db, "update files set size = ?, content = ? where hash = ?", -1, &ifile,0);
} else {
- //fprintf(stderr, "creating new data row\n");
- rc = sqlite3_prepare(db, "insert into files (size, content, hash) values (?,?,?)", -1, &ifile,0);
+ rc = sqlite3_prepare_v2(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(zpm);
- munmap(content, sbuf.st_size);
+ munmap(content, sbuf.st_size);
return 0;
}
- sqlite3_bind_int64(ifile, 1, (sqlite3_int64)sbuf.st_size);
+ rc = sqlite3_bind_int64(ifile, 1, (sqlite3_int64)sbuf.st_size);
if (rc != SQLITE_OK) {
SQLERROR(sqlite3_errmsg(db));
+ sqlite3_finalize(ifile);
fprintf(stderr, "cant bind size\n");
- zpm_rollback(zpm);
- munmap(content, sbuf.st_size);
+ munmap(content, sbuf.st_size);
return 0;
}
- sqlite3_bind_blob64(ifile, 2, outbuf, (sqlite3_int64)outlen, SQLITE_STATIC);
+
+ rc = sqlite3_bind_blob64(ifile, 2, outbuf,
+ (sqlite3_int64)outlen, SQLITE_STATIC);
if (rc != SQLITE_OK) {
SQLERROR(sqlite3_errmsg(db));
+ sqlite3_finalize(ifile);
fprintf(stderr, "cant bind content\n");
- zpm_rollback(zpm);
- munmap(content, sbuf.st_size);
+ munmap(content, sbuf.st_size);
return 0;
}
- sqlite3_bind_text(ifile, 3, hash, 64, SQLITE_STATIC);
+
+ rc = sqlite3_bind_text(ifile, 3, hash, 64, SQLITE_STATIC);
if (rc != SQLITE_OK) {
SQLERROR(sqlite3_errmsg(db));
fprintf(stderr, "cant bind hash\n");
- zpm_rollback(zpm);
- munmap(content, sbuf.st_size);
+ sqlite3_finalize(ifile);
+ munmap(content, sbuf.st_size);
return 0;
}
+
rc = sqlite3_step(ifile);
if (rc != SQLITE_DONE) {
SQLERROR(sqlite3_errmsg(db));
sqlite3_finalize(ifile);
- zpm_rollback(zpm);
- munmap(content, sbuf.st_size);
+ munmap(content, sbuf.st_size);
return 0;
}
sqlite3_finalize(ifile);
- /* commit */
- //zpm_commit(zpm);
-
/* don't need the original file now */
-
}
if (!set_elf_info(zpm->db, hash, content, sbuf.st_size)) {
munmap(content, sbuf.st_size);
- /* if package and not nopackage flag, add to package */
- if (zpm->current_package->name && (!ZPM_NOPACKAGE)) {
- /* TODO */
- }
-
/* return */
return 1;
}
-#endif
-
-#if 0
-int main(int ac, char **av){
- sqlite3 *db = 0;
- int rc;
-
- int blobsize;
- int64_t size;
- void *xzdata;
- int type;
- FILE *out;
- sqlite3_stmt *ifile;
-
- char *hash;
- char *filename;
-
- if (ac < 3) {
- fprintf(stderr, "usage: db hash file\n");
- return 1;
- }
-
- rc = sqlite3_open(av[1], &db);
- if (rc) {
- SQLERROR(sqlite3_errmsg(db));
- sqlite3_close(db);
- return 1;
- }
-
-}
-#endif
-
-#if 0
-Packages are sqlite databases
-
-get application id and userver
-
-Primitive operations:
-
-add path to repo
-associate path with package
-associate blob with path?
-add blob to repo
-* extract blob to a path
-compare blob to filesystem path
-create package with info
-
-Extra primitives:
-
-record elf information about blob
-compress blob
-uncompress blob
-sign a package? What are we verifying?
-#endif
projects / zpackage / blobdiff