]> pd.if.org Git - zpackage/blob - lib/zpm.c
add vercmp at zpm open or init
[zpackage] / lib / zpm.c
1 #define _POSIX_C_SOURCE 200809L
2
3 #include <stdio.h>
4 #include <stdlib.h>
5 #include <string.h>
6 #include <sys/types.h>
7 #include <sys/stat.h>
8 #include <sys/mman.h>
9 #include <unistd.h>
10 #include <fcntl.h>
11 #include <errno.h>
12
13 #include "zpm.h"
14 #include "elf.h"
15
16 #include "sha256.h"
17
18 #if 0
19 struct zpm {
20         sqlite3 *db;
21         char *path; /* path to package file */
22         char *version;
23         int release;
24         char *pkgname;
25         time_t installed; /* install time, 0 for not installed */
26 };
27
28 struct zpm_file {
29         char *path;
30         int mode;
31         uint32_t filetype;
32         char *tags;
33         char *owner;
34         char *group;
35         char *hash; /* could be fixed length */
36         time_t mtime;
37         struct zpm_file *next; /* so you can make a linked list */
38 };
39
40 /* NULL?  Create? */
41 /* associate with a package ? if only one?  first? */
42 int zpm_open(struct zpm *pkg, char *path);
43 int zpm_pkgname(char *base, char *version, int release); /* construct a package file name */
44
45 /* flags for preserving mode, owner, etc */
46 /* puts hash of import in hash */
47 /* path can be a hash, with an "INTERNAL" flag, i.e. internally import */
48 #define ZPM_MODE 0x1
49 #define ZPM_OWNER 0x2
50 #define ZPM_MTIME 0x4
51 #define ZPM_INTERNAL 0x8
52 #define ZPM_NOBLOB 0x10
53 /* don't run scripts on install */
54 #define ZPM_NOSCRIPTS 0x10
55 /* don't associate the file with a package, just do a raw insert */
56 /* otherwise, associate it with the current package */
57 #define ZPM_NOPACKAGE 0x20
58
59 int zpm_import(struct zpm *zp, char *path, uint32_t flags, uint8_t *hash);
60
61 /* link and unlink hashes to packages */
62 int zpm_link(struct zpm *pkg, char *path, char *hash, struct zpm_file *fileinfo);
63 int zpm_unlink(struct zpm *pkg, char *path);
64
65 /* tag a file.  relative to "current package" */
66 int zpm_tag(struct zpm *zp, char *path, char *tags);
67 /* should this be broken up into separage functions ? */
68 int zpm_md(struct zpm *zp, char *path, int mode, char *owner, char *group, time_t mtime);
69
70 /* export hash to dest */
71 int zpm_extract(struct zpm *pkg, char *hash, char *path, int mode);
72
73 /* export path to dest */
74 int zpm_export(struct zpm *zp, char *path, uint32_t flags, char *dest);
75
76 int zpm_close(struct zpm *zp);
77
78 /* attach a signature to a package */
79 int zpm_sign(struct zpm *z, size_t s, void *signature);
80
81 /* set the package info to the nth package, -1 to return count? */
82 /* further import/exports and such will be relative to this package */
83 int zpm_package(struct zpm *zp, int n);
84
85 /* get file information */
86 int zpm_stat(struct zpm *z, struct zpm_file *f, int n);
87
88 /* will also set the package context to the new package */
89 int zpm_newpkg(struct zpm *z, char *base, char *version, int release);
90
91 /* transactions */
92 int zpm_begin(struct zpm *z);
93 int zpm_commit(struct zpm *z);
94 int zpm_rollback(struct zpm *z);
95
96 /* higher level operations */
97
98 /* install or uninstall the package */
99 /* flag for keeping the blobs in local */
100 /* what about tag filtering */
101 int zpm_install(struct zpm *z, struct zpm *local, uint32_t flags);
102 int zpm_uninstall(struct zpm *local);
103
104 /* slurp up the actual blobs */
105 /* what about versioning them if they change */
106 int zpm_preserve(struct zpm *local);
107
108 /* check file integrity */
109 int zpm_checkinstall(struct zpm *local);
110
111 int zpm_merge(struct zpm *z, struct zpm *src, uint32_t flags);
112
113 void uncompresslzma(void *buf, size_t bufsize, FILE *out);
114 #define SQLERROR(x) fprintf(stderr, "%s %d: %s\n", __func__, __LINE__, (x))
115 #endif
116
117 #if 0
118 int zpm_newpkg(struct zpm *z, char *base, char *version, int release) {
119         char *sql = "insert or ignore into packages (package,version,release) values (?,?,?)";
120         int rc;
121         sqlite3_stmt *ifile;
122
123         rc = sqlite3_prepare(db, sql, -1, &ifile,0);
124         if (rc != SQLITE_OK) {
125                 SQLERROR(sqlite3_errmsg(db));
126                 return 0;
127         }
128         rc = sqlite3_bind_text(ifile, 1, base, strlen(base), SQLITE_STATIC);
129         if (rc != SQLITE_OK) {
130                 SQLERROR(sqlite3_errmsg(db));
131                 fprintf(stderr, "cant bind package name\n");
132                 zpm_rollback(pkg);
133                 return 0;
134         }
135         sqlite3_bind_text(ifile, 2, version, strlen(version), SQLITE_STATIC);
136         sqlite3_bind_int(ifile, 3, release)
137
138         rc = sqlite3_step(ifile);
139
140         if (rc != SQLITE_DONE) {
141                 SQLERROR(sqlite3_errmsg(db));
142                 sqlite3_finalize(ifile);
143                 return 0;
144         }
145         sqlite3_finalize(ifile);
146         z->pkg = dupstr(base);
147         z->version = dupstr(version);
148         z->release = release;
149 }
150 #endif
151
152 int zpm_begin(struct zpm *z) {
153         char *errstr = 0;
154         sqlite3_exec(z->db, "begin;", NULL, NULL, &errstr);
155         if (errstr) {
156                 fprintf(stderr, "sqlite begin error: %s\n", errstr);
157                 sqlite3_free(errstr);
158                 return 0;
159         }
160         return 1;
161 }
162
163 int zpm_commit(struct zpm *z) {
164         char *errstr = 0;
165         sqlite3_exec(z->db, "commit;", NULL, NULL, &errstr);
166         if (errstr) {
167                 fprintf(stderr, "sqlite commit error: %s\n", errstr);
168                 sqlite3_free(errstr);
169                 return 0;
170         }
171         return 1;
172 }
173
174 /* wrapper for sqlite3_exec */
175 int zpm_exec(struct zpm *z, const char *sql, int(*callback)(void *, int, char **, char**), void *arg, char **errmsg) {
176         return sqlite3_exec(z->db, sql, callback, arg, errmsg);
177 }
178
179 int zpm_rollback(struct zpm *z) {
180         char *errstr = 0;
181         sqlite3_exec(z->db, "rollback;", NULL, NULL, &errstr);
182         if (errstr) {
183                 fprintf(stderr, "sqlite rollback error: %s\n", errstr);
184                 sqlite3_free(errstr);
185                 return 0;
186         }
187         return 1;
188 }
189
190 int zpm_db_set_pragma(struct zpm *db, int pragma, int value) {
191         int rc;
192         char *sql;
193         sqlite3_stmt *s;
194
195         switch (pragma) {
196                 case 1: sql = "pragma application_id = ?;"; break;
197                 case 2: sql = "pragma user_version = ?;"; break;
198                 default: return -1; break;
199         }
200
201         rc = sqlite3_prepare_v2(db->db, sql, -1, &s, 0);
202
203         if (rc != SQLITE_OK) {
204                 SQLERROR(sqlite3_errmsg(db->db));
205                 return -1;
206         }
207
208         sqlite3_bind_int(s, 1, value);
209         if (rc != SQLITE_OK) {
210                 SQLERROR(sqlite3_errmsg(db->db));
211                 fprintf(stderr, "cant bind pragma value\n");
212                 sqlite3_finalize(s);
213                 return -1;
214         }
215         rc = sqlite3_step(s);
216         if (rc != SQLITE_DONE) {
217                 SQLERROR(sqlite3_errmsg(db->db));
218                 fprintf(stderr, "cant set pragma\n");
219                 sqlite3_finalize(s);
220                 return -1;
221         }
222
223         sqlite3_finalize(s);
224         return value;
225 }
226
227 int zpm_db_pragma(struct zpm *db, int pragma) {
228         int rc;
229         int value = -1;
230         char *sql = 0;
231         sqlite3_stmt *s;
232
233         switch (pragma) {
234                 case 1: sql = "pragma application_id;"; break;
235                 case 2: sql = "pragma user_version;"; break;
236                 default: return -1; break;
237         }
238
239         rc = sqlite3_prepare_v2(db->db, sql, -1, &s, 0);
240
241         if (rc != SQLITE_OK) {
242                 SQLERROR(sqlite3_errmsg(db->db));
243                 fprintf(stderr, "%s, errnum = %d\n", sqlite3_errmsg(db->db), rc);
244                 /* TODO just abort? */
245                 return -1;
246         }
247
248         rc = sqlite3_step(s);
249         if (rc == SQLITE_ROW) {
250                 value = sqlite3_column_int(s, 0);
251         }
252
253         sqlite3_finalize(s);
254         return value;
255 }
256
257 static
258 #include "newdb.c"
259
260 int zpm_db_initialize(struct zpm *pkg) {
261         //fprintf(stderr, "initializing zpm database\n");
262         char *error;
263         switch (sqlite3_exec(pkg->db, createdb, (int (*)(void *,int,char **,char **))0, NULL, &error)) {
264                 case SQLITE_OK: break;
265                 default:
266                         SQLERROR(sqlite3_errmsg(pkg->db));
267                         fprintf(stderr, "error: %s\n", error);
268                         sqlite3_free(error);
269                         return 0;
270                         break;
271         }
272         return 1;
273 }
274
275
276 /* assumes pkg->db is set */
277 static int setupdb(struct zpm *zpm) {
278         char *errstr = 0;
279         int appid, dbver;
280
281         zpm->error = 0;
282
283         appid = zpm_db_pragma(zpm, 1);
284         dbver = zpm_db_pragma(zpm, 2);
285
286         if (appid != 0x5a504442) {
287                 fprintf(stderr, "unknown database type\n");
288                 zpm->error = 1;
289                 return 0;
290         }
291
292         if (dbver > 1) {
293                 fprintf(stderr, "version %d zpm db detected, this program only works with version 1 databases\n", dbver);
294                 zpm->error = 1;
295                 return 0;
296         }
297
298         sqlite3_exec(zpm->db, "pragma foreign_keys = ON;", NULL, NULL, &errstr);
299
300         if (errstr) {
301                 free(zpm->errmsg);
302                 zpm->errmsg = strdup(errstr);
303                 fprintf(stderr, "sqlite foreign key error: %s\n", errstr);
304                 sqlite3_free(errstr);
305                 zpm->error = 1;
306                 return 0;
307         }
308
309         /* TODO add vercmp */
310
311         return 1;
312 }
313
314 struct zpm *zpm_clearmem(struct zpm *zpm) {
315         if (!zpm) {
316                 zpm = malloc(sizeof *zpm);
317         }
318
319         if (zpm) {
320                 *zpm = (struct zpm){0};
321         }
322
323         return zpm;
324 }
325
326 int zpm_init(struct zpm *pkg, char *path) {
327         int rc;
328         sqlite3 *db = 0;
329         int appid;
330
331         zpm_clearmem(pkg);
332
333         rc = sqlite3_open_v2(path, &db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL);
334         if (rc) {
335                 SQLERROR(sqlite3_errmsg(db));
336                 sqlite3_close(db);
337                 return 0;
338         }
339         pkg->db   = db;
340         pkg->path = strdup(path);
341
342         appid = zpm_db_pragma(pkg, 1);
343
344         switch (appid) {
345                 case 0: if (!zpm_db_initialize(pkg)) {
346                                 sqlite3_close(db);
347                                 return 1;
348                         };
349                         break;
350                 case 0x5a504442:
351                         /* already initialized */
352                         break;
353                 default:
354                         fprintf(stderr, "unknown database type\n");
355                         sqlite3_close(db);
356                         return 0;
357                         break;
358         }
359
360         if (!setupdb(pkg)) {
361                 sqlite3_close(db);
362                 pkg->db = 0;
363                 return 0;
364         }
365         zpm_addvercmp(pkg);
366
367         return 1;
368 }
369
370 int zpm_open(struct zpm *zpm, char *path) {
371         int rc;
372         sqlite3 *db = 0;
373
374         zpm_clearmem(zpm);
375
376         rc = sqlite3_open_v2(path, &db, SQLITE_OPEN_READWRITE, NULL);
377         if (rc) {
378                 SQLERROR(sqlite3_errmsg(db));
379                 sqlite3_close(db);
380                 zpm->error = 1;
381                 fprintf(stderr, "path = %s\n", path);
382                 return 0;
383         }
384         zpm->db   = db;
385         zpm->path = strdup(path);
386
387         if (!setupdb(zpm)) {
388                 sqlite3_close(db);
389                 zpm->db = 0;
390                 zpm->error = 1;
391                 return 0;
392         }
393         zpm_addvercmp(zpm);
394
395         return 1;
396 }
397
398 int zpm_close(struct zpm *pkg) {
399         if (pkg) {
400                 sqlite3_close(pkg->db);
401                 free(pkg->path);
402         }
403         /* TODO free any malloced names and such */
404         return 1;
405 }
406
407 static int zpm_sqlite_vercmp(void *not_used, int lena, const void *a,
408                 int lenb, const void *b) {
409         /* not sure what the ints are, possibly string lengths */
410         if (not_used != 0) fprintf(stderr, "sqlite vercmp not_used = %p\n",
411                         not_used);
412         if (lena == 0 && lenb > 0) return 1;
413         return zpm_vercmp(a, b);
414 }
415
416 int zpm_addvercmp(struct zpm *pkg) {
417         return sqlite3_create_collation(
418                         pkg->db, "vercmp", SQLITE_UTF8, NULL,
419                         zpm_sqlite_vercmp
420                         );
421 }
422
423 int zpm_extract(struct zpm *pkg, char *hash, char *path, int mode) {
424         int rc;
425
426         int blobsize;
427         //int64_t size;
428         void *xzdata;
429         int type;
430         FILE *out;
431         sqlite3_stmt *ifile;
432
433         /* TODO check null */
434         sqlite3 *db = pkg->db;
435
436         rc = sqlite3_prepare(db, "select size, content from files where hash = ?", -1, &ifile,0);
437         if (rc != SQLITE_OK) {
438                 SQLERROR(sqlite3_errmsg(db));
439                 return 0;
440         }
441
442         /* hash, filename */
443
444         sqlite3_bind_text(ifile, 1, hash, 64, SQLITE_STATIC);
445
446         rc = sqlite3_step(ifile);
447
448         if (rc == SQLITE_DONE) {
449                 /* didn't find a row */
450                 sqlite3_finalize(ifile);
451                 sqlite3_close(db);
452                 fprintf(stderr, "no such hash: %s\n", hash);
453                 return 0;
454         }
455         /* either way we're done with this now */
456
457         if (rc != SQLITE_ROW) {
458                 SQLERROR(sqlite3_errmsg(db));
459                 sqlite3_finalize(ifile);
460                 sqlite3_close(db);
461                 return 0;
462         }
463
464         type = sqlite3_column_type(ifile, 0);
465         if (type == SQLITE_NULL) {
466                 fprintf(stderr, "no file size\n");
467                 sqlite3_finalize(ifile);
468                 sqlite3_close(db);
469                 return 0;
470         }
471         type = sqlite3_column_type(ifile, 1);
472         if (type == SQLITE_NULL) {
473                 fprintf(stderr, "no file data\n");
474                 sqlite3_finalize(ifile);
475                 sqlite3_close(db);
476                 return 0;
477         }
478         //size = sqlite3_column_int64(ifile, 0);
479         xzdata = (void *)sqlite3_column_blob(ifile, 1);
480         blobsize = sqlite3_column_bytes(ifile, 1);
481
482         if (strcmp(path, "-")) {
483                 out = fopen(path, "w");
484         } else {
485                 out = stdout;
486         }
487         if (!out) {
488                 fprintf(stderr, "can't open output file %s\n", path);
489                 sqlite3_finalize(ifile);
490                 sqlite3_close(db);
491                 return 0;
492         }
493         //fwrite(xzdata, blobsize, 1, stdout);
494
495         //fprintf(stderr, "uncompressing %d bytes at %p, expect %lld\n", blobsize, xzdata, (long long int)size);
496         uncompresslzma(xzdata, blobsize, out);
497         fclose(out);
498         chmod(path, mode);
499
500         sqlite3_finalize(ifile);
501
502         return 1;
503 }
504
505 static int run_for_hash(sqlite3 *db, char *sql, char *hash) {
506         int rc;
507         sqlite3_stmt *ifile;
508
509         rc = sqlite3_prepare_v2(db, sql, -1, &ifile, 0);
510         if (rc != SQLITE_OK) {
511                 SQLERROR(sqlite3_errmsg(db));
512                 return 0;
513         }
514
515         /* hash, filename */
516
517         sqlite3_bind_text(ifile, 1, hash, 64, SQLITE_STATIC);
518         do {
519                 rc = sqlite3_step(ifile);
520         } while (rc == SQLITE_ROW);
521
522         sqlite3_finalize(ifile);
523
524         return rc == SQLITE_DONE;
525 }
526
527 #define SQLERP(db, msg) fprintf(stderr, "%s: %s\n", msg, sqlite3_errmsg(db))
528
529 static int set_elf_info(sqlite3 *db, char *hash, char *content, size_t length) {
530         if (length >= sizeof (Elf64_Ehdr) && libelf_iself(content)) {
531                 char *strtab;
532                 Elf64_Dyn *dyn;
533                 int i;
534                 Elf64_Phdr *phdr;
535                 Elf64_Ehdr *hdr;
536                 sqlite3_stmt *ifile;
537                 int rc;
538
539                 /* clear existing for this hash */
540                 if (!run_for_hash(db, "delete from elfinfo where file = ?", hash)) {
541                         SQLERP(db, "error clearing elf info");
542                         return 0;
543                 }
544
545                 if (!run_for_hash(db, "delete from elflibraries where file = ?", hash)) {
546                         SQLERP(db, "error clearing elf library");
547                         return 0;
548                 }
549
550                 if (!run_for_hash(db, "delete from elfneeded where file = ?", hash)) {
551                         SQLERP(db, "error clearing elf needed");
552                         return 0;
553                 }
554
555                 hdr = libelf_header(content);
556                 /* if lib, set soname */
557                 if (libelf_type(content) == ET_DYN) {
558                         char *soname = libelf_soname(content);
559                         if (soname) {
560
561                                 sqlite3_prepare_v2(db, "insert into elflibraries (file,soname) values (?,?)",-1, &ifile, 0);
562                                 sqlite3_bind_text(ifile,1,hash,64,SQLITE_STATIC);
563                                 sqlite3_bind_text(ifile,2,soname,-1,SQLITE_STATIC);
564                                 rc = sqlite3_step(ifile);
565                                 if (rc != SQLITE_DONE) {
566                                         SQLERROR(sqlite3_errmsg(db));
567                                         sqlite3_finalize(ifile);
568                                         fprintf(stderr, "error setting library soname\n");
569                                         return 0;
570                                 }
571                                 sqlite3_finalize(ifile);
572                         } else {
573                                 fprintf(stderr, "can't find soname\n");
574                         }
575                 }
576
577                 /* if exe, set neededs */
578                 if (libelf_type(content) == ET_EXEC) {
579                         Elf64_Shdr *dsect;
580                         char *elf;
581
582                         elf = (char *)content;
583                         /* find program header table */
584                         for (i = 0; i < hdr->e_phnum; i++) {
585                                 phdr = (Elf64_Phdr *)(elf + hdr->e_phoff + i * hdr->e_phentsize);
586                                 if (phdr->p_type == PT_DYNAMIC) {
587                                         dsect = (Elf64_Shdr *)(elf + phdr->p_offset);
588                                 }
589                         }
590                         dyn = (Elf64_Dyn *)(elf + dsect->sh_offset);
591                         if (!dyn) {
592                                 exit(9);
593                         }
594                         dyn = (Elf64_Dyn *)dsect;
595
596                         dsect = libelf_section(elf, SHT_DYNAMIC);
597                         Elf64_Shdr *strsect;
598
599                         strsect = libelf_section_n(elf, dsect->sh_link);
600                         strtab = elf + strsect->sh_offset;
601
602                         sqlite3_prepare_v2(db, "insert into elfneeded (file,needed) values (?,?)",-1, &ifile, 0);
603                         sqlite3_bind_text(ifile,1,hash,64,SQLITE_STATIC);
604                         while (dyn->d_tag != DT_NULL) {
605                                 if (dyn->d_tag == DT_NEEDED) {
606                                         char *need;
607                                         int rc;
608
609                                         need = strtab + dyn->d_un.d_val;
610                                         if (strlen(need) == 0) continue;
611                                         sqlite3_bind_text(ifile,2,need,strlen(need),SQLITE_STATIC);
612 #if 0
613                                         fprintf(stderr, "%s needs %s\n", hash, need);
614 #endif
615                                         rc = sqlite3_step(ifile);
616                                         if (rc != SQLITE_DONE) {
617                                                 SQLERP(db, "error setting needed library");
618                                                 sqlite3_finalize(ifile);
619                                                 return 0;
620                                         }
621                                         sqlite3_reset(ifile);
622                                 }
623                                 dyn++;
624                         }
625                         sqlite3_finalize(ifile);
626                 }
627         }
628         return 1;
629 }
630
631 #if 1
632 int zpm_import(struct zpm *zpm, char *path, uint32_t flags, char *hash) {
633         int fd;
634         void *content;
635         struct stat sbuf;
636         unsigned char tmp[32];
637         struct sha256_state md;
638         sqlite3_stmt *ifile;
639         int haverow = 0,havedata = 0;
640         int j,rc,type;
641         char hashbuf[65];
642
643         /* xz compress it */
644         size_t outlen = 0;
645         void *outbuf;
646
647         if (!zpm || !zpm->db || !path) {
648                 return 0;
649         }
650
651         /* use local if caller didn't pass in room */
652         if (!hash) {
653                 hash = hashbuf;
654         }
655
656         if (flags) {
657                 fprintf(stderr, "zpm_import unused flags = %d\n", flags);
658         }
659         /* mmap the file */
660         fd = open(path, O_RDONLY);
661         if (fd == -1) {
662                 zpm->error = errno;
663                 fprintf(stderr, "%s can't open %s: %s\n", __FUNCTION__, path,strerror(errno));
664                 return 0;
665         }
666         if (fstat(fd, &sbuf) == -1) {
667                 zpm->error = errno;
668                 fprintf(stderr, "%s can't fstat %s: %s\n", __FUNCTION__, path,strerror(errno));
669                 return 0;
670         }
671         /* not a regular file? */
672         if (!S_ISREG(sbuf.st_mode)) {
673                 char *ftype;
674                 switch (sbuf.st_mode & S_IFMT) {
675                         case S_IFSOCK: ftype = "socket"; break;
676                         case S_IFLNK : ftype = "symlink"; break;
677                         case S_IFBLK : ftype = "block device"; break;
678                         case S_IFDIR : ftype = "directory"; break;
679                         case S_IFCHR : ftype = "character device"; break;
680                         case S_IFIFO : ftype = "fifo"; break;
681                         default: ftype = "unknown file type"; break;
682                 }
683                 /* TODO this is ok, just stored differently */
684                 fprintf(stderr, "%s can't import %s file: %s\n", __FUNCTION__, ftype, path);
685                 zpm->error = EINVAL;
686                 return 0;
687         }
688
689         content = mmap(0, sbuf.st_size, PROT_READ,MAP_PRIVATE, fd, 0);
690         close(fd);
691         if (!content) {
692                 zpm->error = errno;
693                 fprintf(stderr, "%s can't mmap %s: %s\n", __FUNCTION__, path,strerror(errno));
694                 return 0;
695         }
696
697         /* get hash */
698         sha256_init(&md);
699         sha256_process(&md, content, sbuf.st_size);
700         sha256_done(&md, tmp);
701         for (j=0;j<32;j++) {
702                 sprintf(hash+j*2, "%02x", (unsigned)tmp[j]);
703         }
704         hash[64] = 0;
705         //fprintf(stderr, "file %s: %s\n", path, hash);
706
707         /* TODO check null */
708         sqlite3 *db = zpm->db;
709
710         /* prepare and bind */
711
712         rc = sqlite3_prepare_v2(db, "select size, content is not null from files where hash = ?", -1, &ifile,0);
713         if (rc != SQLITE_OK) {
714                 SQLERROR(sqlite3_errmsg(db));
715                 munmap(content, sbuf.st_size);
716                 return 0;
717         }
718
719         /* hash, filename */
720
721         sqlite3_bind_text(ifile, 1, hash, 64, SQLITE_STATIC);
722
723         rc = sqlite3_step(ifile);
724
725         if (rc != SQLITE_DONE) {
726                 if (rc != SQLITE_ROW) {
727                         /* didn't find a row */
728                         SQLERROR(sqlite3_errmsg(db));
729                         zpm_rollback(zpm);
730                 munmap(content, sbuf.st_size);
731                         return 0;
732                 }
733                 haverow = 1;
734 //              fprintf(stderr, "have row for hash\n");
735                 type = sqlite3_column_type(ifile, 0);
736                 if (type == SQLITE_NULL) {
737                         /* TODO assert, this shouldn't be possible? */
738                         fprintf(stderr, "no file size\n");
739                         sqlite3_finalize(ifile);
740                 munmap(content, sbuf.st_size);
741                         return 0;
742                 }
743                 type = sqlite3_column_type(ifile, 1);
744                 if (type == SQLITE_NULL) {
745                         /* TODO assert, this shouldn't be possible? */
746                         fprintf(stderr, "no file data\n");
747                         sqlite3_finalize(ifile);
748                 munmap(content, sbuf.st_size);
749                         return 0;
750                         /* which is fine, just need to update the row then */
751                 }
752                 havedata = sqlite3_column_int(ifile, 1);
753         }
754
755         sqlite3_finalize(ifile);
756
757         if (!havedata) {
758                 /* compress */
759                 outbuf = compresslzma(content, sbuf.st_size, &outlen);
760                 if (!outbuf) {
761                         fprintf(stderr, "compresslzma failed\n");
762                 munmap(content, sbuf.st_size);
763                         return 0;
764                 }
765                 //fprintf(stderr, "compressed to %zu\n", outlen);
766
767                 /* start a transaction */
768                 // do that outside of here 
769                 //zpm_begin(zpm);
770
771                 /* insert */
772                 if (haverow) {
773                         //fprintf(stderr, "adding file data\n");
774                         rc = sqlite3_prepare(db, "update files set size = ?, content = ? where hash = ?", -1, &ifile,0);
775                 } else {
776                         //fprintf(stderr, "creating new data row\n");
777                         rc = sqlite3_prepare(db, "insert into files (size, content, hash) values (?,?,?)", -1, &ifile,0);
778                 }
779                 if (rc != SQLITE_OK) {
780                         SQLERROR(sqlite3_errmsg(db));
781                         fprintf(stderr, "cant prepare data\n");
782                         zpm_rollback(zpm);
783                 munmap(content, sbuf.st_size);
784                         return 0;
785                 }
786
787                 sqlite3_bind_int64(ifile, 1, (sqlite3_int64)sbuf.st_size);
788                 if (rc != SQLITE_OK) {
789                         SQLERROR(sqlite3_errmsg(db));
790                         fprintf(stderr, "cant bind size\n");
791                         zpm_rollback(zpm);
792                 munmap(content, sbuf.st_size);
793                         return 0;
794                 }
795                 sqlite3_bind_blob64(ifile, 2, outbuf, (sqlite3_int64)outlen, SQLITE_STATIC);
796                 if (rc != SQLITE_OK) {
797                         SQLERROR(sqlite3_errmsg(db));
798                         fprintf(stderr, "cant bind content\n");
799                         zpm_rollback(zpm);
800                 munmap(content, sbuf.st_size);
801                         return 0;
802                 }
803                 sqlite3_bind_text(ifile, 3, hash, 64, SQLITE_STATIC);
804                 if (rc != SQLITE_OK) {
805                         SQLERROR(sqlite3_errmsg(db));
806                         fprintf(stderr, "cant bind hash\n");
807                         zpm_rollback(zpm);
808                 munmap(content, sbuf.st_size);
809                         return 0;
810                 }
811                 rc = sqlite3_step(ifile);
812                 if (rc != SQLITE_DONE) {
813                         SQLERROR(sqlite3_errmsg(db));
814                         sqlite3_finalize(ifile);
815                         zpm_rollback(zpm);
816                 munmap(content, sbuf.st_size);
817                         return 0;
818                 }
819                 sqlite3_finalize(ifile);
820
821                 /* commit */
822                 //zpm_commit(zpm);
823
824                 /* don't need the original file now */
825
826         }
827
828         if (!set_elf_info(zpm->db, hash, content, sbuf.st_size)) {
829                 fprintf(stderr, "setting elf info failed\n");
830                 munmap(content, sbuf.st_size);
831                 return 0;
832         }
833
834         munmap(content, sbuf.st_size);
835
836         /* if package and not nopackage flag, add to package */
837         if (zpm->current_package->name && (!ZPM_NOPACKAGE)) {
838                 /* TODO */
839         }
840
841         /* return */
842         return 1;
843 }
844 #endif
845
846 #if 0
847 int main(int ac, char **av){
848         sqlite3 *db = 0;
849         int rc;
850
851         int blobsize;
852         int64_t size;
853         void *xzdata;
854         int type;
855         FILE *out;
856         sqlite3_stmt *ifile;
857
858         char *hash;
859         char *filename;
860
861         if (ac < 3) {
862                 fprintf(stderr, "usage: db hash file\n");
863                 return 1;
864         }
865
866         rc = sqlite3_open(av[1], &db);
867         if (rc) {
868                 SQLERROR(sqlite3_errmsg(db));
869                 sqlite3_close(db);
870                 return 1;
871         }
872
873 }
874 #endif
875
876 #if 0
877 Packages are sqlite databases
878
879 get application id and userver
880
881 Primitive operations:
882
883 add path to repo
884 associate path with package
885 associate blob with path?
886 add blob to repo
887 * extract blob to a path
888 compare blob to filesystem path
889 create package with info
890
891 Extra primitives:
892
893 record elf information about blob
894 compress blob
895 uncompress blob
896 sign a package?  What are we verifying?
897 #endif