summaryrefslogtreecommitdiffstats
path: root/autoirpm.update-all.cc
blob: de96edd482ba021a3f4e5f3c09ef47b43004af79 (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
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <rpmlib.h>
#include <string>
#include <set>
#include <fstream>


typedef set<string>::difference_type diff_type;

set<string> *read_set(const char *file) {
  ifstream input(file);
  if (input) {
    set<string> *myset = new set<string>;
    istream_iterator<string,diff_type> input_set(input),eos;
    copy(input_set, eos, inserter(*myset, myset->begin()));
    return myset;
  }
  return 0;
}

int main(int argc, char **argv) {
  if (argc <= 3) {
    cerr << "usage: " << argv[0] << " <allow progs file> <deny progs file> <hdlist> [<hdlist> ...]\n";
    exit(1);
  }

  set<string> *allow = read_set(argv[1]);
  set<string> *deny  = read_set(argv[2]);

  for (int i = 3; i < argc; i++) {
    Header header;
    FD_t fd = strcmp(argv[i], "-") == 0 ? fdDup(STDIN_FILENO) : fdOpen(argv[i], O_RDONLY, 0);
    if (fdFileno(fd) < 0) {
      fprintf(stderr, "%s: cannot open file %s\n", argv[0], argv[i]);
      exit(1);
    }

    while ((header=headerRead(fd, HEADER_MAGIC_YES))) {
      int_32 type, count;
      unsigned short *p;
      char **f, *name;
      int printed = 0;

      headerGetEntry(header, RPMTAG_NAME, &type, (void **) &name, NULL);
      headerGetEntry(header, RPMTAG_FILEMODES, &type, (void **) &p, &count);
      headerGetEntry(header, RPMTAG_OLDFILENAMES, &type, (void **) &f, NULL);

      char ** baseNames, ** dirNames;
      int_32 * dirIndexes;
      headerGetEntry(header, RPMTAG_BASENAMES, &type, (void **) &baseNames, NULL);
      headerGetEntry(header, RPMTAG_DIRINDEXES, &type, (void **) &dirIndexes, NULL);
      headerGetEntry(header, RPMTAG_DIRNAMES, &type, (void **) &dirNames, NULL);

      for (int i = 0; i < count; i++) {
	if ((p[i] & 040111) == 0111) {
	  int ok;
	  if (f) {
	    char *s = strrchr(f[i], '/');
	    ok = s && s - 3 >= f[i] && strncmp(s - 3, "bin", 3) == 0 &&
	      (!allow || allow->count(s + 1)) &&
	      (!deny || !deny->count(s + 1));
	  } else {
	    char *d = dirNames[dirIndexes[i]];
	    ok = strlen(d) >= 4 && strncmp(d + strlen(d) - 4, "bin/", 4) == 0 &&
	      (!allow || allow->count(baseNames[i])) &&
	      (!deny || !deny->count(baseNames[i]));
	  }
	  if (ok) {
	    if (!printed) {
	      printed = 1;
	      cout << name;
	    }
	    if (f) cout << " " << f[i]; else cout << " " << dirNames[dirIndexes[i]] << baseNames[i];
	  }
	}
      }
      if (printed) cout << "\n";
    }
    fdClose(fd);
  }
  return 0;
}