Added a comment in the ChangeLog
[squirrelmail.git] / plugins / filters / bulkquery / bulkquery.c
CommitLineData
b1b8988d 1
2#include <arpa/nameser.h>
3#include <lwres/lwres.h>
4#include <strings.h>
5#include <stdio.h>
6#include <pthread.h>
7
8
9#define BUFLEN 1024
10#define MAXSTR 80
11#define MAXTHREADS 50
12#define MAXRBLS 40
13#define DEFTTL 7200
14
15extern int errno;
16extern int h_errno;
17
18
19struct ipnode;
20typedef struct ipnode *iplist;
21struct ipnode {
22 char *IP;
23 iplist next;
24};
25
26iplist IPs;
27
28pthread_mutex_t *mutexp;
29pthread_mutex_t *mutexoutput;
30
31char *dnsrbls[MAXRBLS];
32int numrbls, numthreads, numqueries;
33
34void do_queries () {
35 iplist tIP;
36 lwres_context_t *ctx = NULL;
37 lwres_grbnresponse_t *response = NULL;
38 int n, i;
39
40
41 pthread_mutex_lock(mutexp);
42 tIP = IPs;
43 if (IPs != NULL) {
44 IPs = tIP->next;
45 }
46 pthread_mutex_unlock(mutexp);
47
48 while (tIP != NULL) {
49//fprintf (stderr, "making query %s\n", tIP->IP); fflush(stderr);
50 if (lwres_context_create(&ctx, NULL, NULL, NULL, 0) != 0) {
51 fprintf (stderr, "Couldn't create context\n");
52 return;
53 } else {
54 lwres_conf_parse(ctx, lwres_resolv_conf);
55 //pthread_mutex_lock(mutexoutput);
56 n = lwres_getrdatabyname(ctx, tIP->IP, ns_c_in, ns_t_a, 0, &response);
57 //pthread_mutex_unlock(mutexoutput);
58 if (n == LWRES_R_SUCCESS) {
59 printf ("%s,%d.%d.%d.%d,%d\n", tIP->IP,
60 response->rdatas[0][0], response->rdatas[0][1],
61 response->rdatas[0][2], response->rdatas[0][3],
62 response->ttl);
63 //fprintf (stderr, "freeing response\n"); fflush(stderr);
64 lwres_grbnresponse_free(ctx, &response);
65 } else {
66 //fprintf (stderr, "Nothing found\n");
67 printf ("%s, %s, %d\n", tIP->IP, tIP->IP, DEFTTL);
68 }
69 //fprintf (stderr, "freeing context\n"); fflush(stderr);
70 lwres_context_destroy(&ctx);
71 //fprintf (stderr, "done freeing\n"); fflush(stderr);
72 }
73
74 pthread_mutex_lock(mutexp);
75 tIP = IPs;
76 if (IPs != NULL) {
77 IPs = tIP->next;
78 }
79 pthread_mutex_unlock(mutexp);
80 }
81}
82
83void GetRBLs() {
84 char instr[MAXSTR];
85 numrbls = 0;
86 while ((fgets(instr, MAXSTR, stdin) != NULL) && (numrbls < MAXRBLS)) {
87 instr[strlen(instr)-1] = 0; // strip off newline
88 if (strncmp(instr, "----------", 10) == 0) {
89 return;
90 }
91 dnsrbls[numrbls] = (char *) malloc(strlen(instr)+1);
92 if (dnsrbls[numrbls] == NULL) {
93 fprintf (stderr, "Couldn't allocate memory for %d DNS RBLs\n", numrbls);
94 exit (10);
95 } else {
96 strcpy (dnsrbls[numrbls], instr);
97 numrbls++;
98 }
99 }
100}
101
102
103main () {
104 pthread_t threads[MAXTHREADS];
105 char instr[MAXSTR];
106 iplist tIP;
107 int loop1;
108
109 GetRBLs();
110
111// for (loop1=0; loop1<numrbls; loop1++)
112// fprintf (stderr, "%s\n", dnsrbls[loop1]);
113// fprintf (stderr, "----------\n");
114
115 numqueries = 0;
116 IPs = NULL;
117 while (fgets(instr, MAXSTR, stdin) != NULL) {
118 instr[strlen(instr)-1] = 0;
119 for (loop1 = 0; loop1 < numrbls; loop1++) {
120 tIP = (iplist)malloc(sizeof(struct ipnode));
121 tIP->IP = (char *)malloc(strlen(instr)+strlen(dnsrbls[loop1])+2);
122 strcpy (tIP->IP, instr);
123 strcat (tIP->IP, dnsrbls[loop1]);
124 tIP->next = IPs;
125 IPs = tIP;
126 numqueries++;
127 }
128 }
129
130// fprintf (stderr, "%d queries to make\n", numqueries);
131// tIP = IPs;
132// while (tIP != NULL) {
133// fprintf (stderr, "%s\n", tIP->IP);
134// tIP = tIP->next;
135// }
136// fprintf (stderr, "done\n");
137// exit (0);
138
139 mutexp=(pthread_mutex_t *) malloc(sizeof(pthread_mutex_t));
140 pthread_mutex_init(mutexp, NULL);
141 mutexoutput=(pthread_mutex_t *) malloc(sizeof(pthread_mutex_t));
142 pthread_mutex_init(mutexoutput, NULL);
143
144 numthreads = 0; // number of threads created successfully
145 for (loop1 = 0; ((loop1<MAXTHREADS) && (loop1<numqueries)); loop1++) {
146 if (pthread_create(&threads[loop1], NULL,
147 (void *) do_queries, NULL) != 0) {
148 fprintf (stderr, "Couldn't make more than %d threads\n", numthreads);
149 break;
150 } else {
151 numthreads++;
152 }
153 }
154
155 for (loop1 = 0; loop1 < numthreads ; loop1++) {
156 pthread_join(threads[loop1], NULL);
157 }
158
159//do_queries();
160
161 exit (0);
162}
163