summaryrefslogtreecommitdiffstats
path: root/common.c
blob: 08dd449a41f5c4efc914423392387342d8295572 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include "common.h"

char *table_name_to_file(const char *name) {

	char *share_path = getenv("SHARE_PATH");
	char *fname;
	if (!share_path || !*share_path) share_path = "/usr/share";

	asprintf(&fname, "%s/ldetect-lst/%s", share_path, name);

	return fname;
}

fh fh_open(const char *name) {
	fh ret;
	char *fname = table_name_to_file(name);

	if (access(fname, R_OK) == 0) {
		/* prefer gzip type when not compressed, more direct than zlib access */
		ret.gztype = GZIP;
		ret.u.gzip_fh.f = fopen(fname, "r");
		ret.u.gzip_fh.pid = 0;
	} else {
		char *fname_gz;
                asprintf(&fname_gz, "%s.gz", fname);
                if (access(GZIP_BIN, R_OK) == 0) {
                        int fdno[2];
                        ret.gztype = GZIP;
                        if (access(fname_gz, R_OK) != 0) {
                                fprintf(stderr, "Missing %s (should be %s)\n", name, fname);
                                exit(1);
                        }
                        if (pipe(fdno)) {
                                perror("pciusb");
                                exit(1);
                        }

                        if ((ret.u.gzip_fh.pid = fork()) != 0) {
                                ret.u.gzip_fh.f = fdopen(fdno[0], "r");
                                close(fdno[1]);
                        } else {
                                char* cmd[5];
                                int ip = 0;
                                char *ld_loader = getenv("LD_LOADER");

                                if (ld_loader && *ld_loader)
                                        cmd[ip++] = ld_loader;

                                cmd[ip++] = GZIP_BIN;
                                cmd[ip++] = "-cd";
                                cmd[ip++] = fname_gz;
                                cmd[ip++] = NULL;

                                dup2(fdno[1], STDOUT_FILENO);
                                close(fdno[0]);
                                close(fdno[1]);
                                execvp(cmd[0], cmd);
                                perror("pciusb"); 
                                exit(2);
                        }
		} else {
                        ret.gztype = ZLIB;
                        ret.u.zlib_fh = gzopen(fname_gz, "r");
                        if (!ret.u.zlib_fh) {
                                perror("pciusb");
                                exit(3);
                        }
                }
	}

	free(fname);
	return ret;
}

char* fh_gets(char *line, int size, fh *f) {
        char *ret;
        switch (f->gztype) {
        case ZLIB:
                ret = gzgets(f->u.zlib_fh, line, size);
                break;
        case GZIP:
                ret = fgets(line, size, f->u.gzip_fh.f);
                break;
        }
        return ret;
}

int fh_close(fh *f) {
        int ret;
        switch (f->gztype) {
        case ZLIB:
                ret = gzclose(f->u.zlib_fh);
                break;
        case GZIP:
                ret = fclose(f->u.gzip_fh.f);
                if (f->u.gzip_fh.pid > 0)
                        waitpid(f->u.gzip_fh.pid, NULL, 0);
                break;
        }
        return ret;
}