summaryrefslogtreecommitdiffstats
path: root/autoirpm.update-all.cc
blob: 3d13899e7eb536319ee43cb66b920f73f61ae2f0 (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
#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, *s;
      int printed = 0;

      headerGetEntry(header, RPMTAG_NAME, &type, (void **) &name, &count);
      headerGetEntry(header, RPMTAG_FILEMODES, &type, (void **) &p, &count);
      headerGetEntry(header, RPMTAG_FILENAMES, &type, (void **) &f, &count);
      for (; count--; *p++, *f++)
	if ((*p & 040111) == 0111 && 
	    (s = strrchr(*f, '/')) && 
	    s - 3 >= *f && 
	    strncmp(s - 3, "bin", 3) == 0 &&
	    (!allow || allow->count(s + 1)) &&
	    (!deny || !deny->count(s + 1))) {

	  if (!printed) {
	    printed = 1;
	    cout << name;
	  }
	  cout << " " << *f;
	}
      if (printed) cout << "\n";
    }
    fdClose(fd);
  }
  return 0;
}