source: trollforge/proxy/oven/washingmachine.c @ 606

//#include <stdlib.h>
//#include <stdio.h>
//#include <stdarg.h>
//#include <string.h>
//#include <signal.h>
//#include <sys/socket.h>
//#include <sys/select.h>
//#include <sys/types.h>
//#include <netinet/in.h>

#include "sockslib.h"

// for discarding duplicates
// cat file | sort | uniq > newfile

#define MAX_BUF_LEN 512

int dont_actually_check_proxies;
int verbose_mode;
FILE *check_proxy_output;

int num_checked;
int num_passed;

void sig_handler(int nig) {}

void usage() {
        printf("washingmachine v0.7\n");
        printf("Function:\n");
        printf("    Takes a proxy list from stdin (format [ip:port\\n]), prints working\n");
        printf("    proxies to output, prints verbose information to stderr\n");
        printf("\n");
        printf("Example usage:\n");
        printf("    cat unchecked_proxies.txt | ./washingmachine > checked_proxies.txt\n");
        printf("\n");
        printf("    Command line options\n");
        printf("        -v              toggle verbose mode (off by default)\n");
        printf("        -d              dont check just scan (off by default)\n");
        printf("        -t [seconds]    set a timeout on possibly blocking function\n");
        printf("                            calls (default 3 seconds)\n");
        printf("        -o [file]       set a filename to output proxies\n");
        printf("                            (default stdout)\n");
        printf("        -h              print this help information\n");
}

void report(int nig) {
        if(nig == SIGINT)
                fprintf(stderr, "\n");
        
        fprintf(stderr, "- Report -------------------\n");
        fprintf(stderr, "   Proxies checked -   %d\n", num_checked);
        fprintf(stderr, "   Proxies passed -    %d\n", num_passed);
        fprintf(stderr, "   Ratio -             %d%c\n", (num_passed*100)/num_checked, '%');
        fprintf(stderr, "----------------------------\n");
        
        if(nig == SIGINT)
                exit(0);
}

int main(int argc, char ** argv) {
        int argi;
        char str[MAX_BUF_LEN];
        
        signal(SIGALRM, sig_handler);
        signal(SIGINT, report);
        
        dont_actually_check_proxies = 0;
        verbose_mode = 0;
        check_proxy_output = stdout;
        
        num_checked = 0;
        num_passed = 0;
        
        for(argi = 1; argi < argc; argi++) {
                if(strncmp("-v", argv[argi], 2) == 0) {
                        verbose_mode = !verbose_mode;
                }
                if(strncmp("-d", argv[argi], 2) == 0) {
                        dont_actually_check_proxies = !dont_actually_check_proxies;
                }
                else if(strncmp("-t", argv[argi], 2) == 0) {
                        if(argc > argi+1) {
                                set_signal_timeout(atoi(argv[argi+1]));
                        }
                        else {
                                fprintf(stderr, "Supply a timeout value in seconds\n");
                                return 0;
                        }
                }
                else if(strncmp("-o", argv[argi], 2) == 0) {
                        if(argc > argi+1) {
                                check_proxy_output = fopen(argv[argi+1], "a");
                                if(check_proxy_output == NULL) {
                                        fprintf(stderr, "Cannot open supplied output file\n");
                                        return 0;
                                }
                        }
                        else {
                                fprintf(stderr, "Supply a filename to output proxies too\n");
                                return 0;
                        }
                }
                if(strncmp("-h", argv[argi], 2) == 0) {
                        usage();
                        return 0;
                }
        }
        
        set_verbose_mode(verbose_mode);
        
        while(fgets(str, MAX_BUF_LEN, stdin) != NULL) {
                char *prox;
                int port;
                
                int return_code;
                
                if(split_proxy_port(str, &prox, &port) == 0) {
                        num_checked++;
                        
                        if(dont_actually_check_proxies) {
                                fprintf(check_proxy_output, "%s:%d\n", prox, port);
                                free(prox);
                                continue;
                        }
                        
                        return_code = proxy_test(prox, port);
                        
                        if(return_code == TESTPROXY_RETCODE_SUCCESS) {
                                num_passed++;
                                fprintf(check_proxy_output, "%s:%d\n", prox, port);
                                fflush(check_proxy_output);
                                fprintf(stderr, "Got proxy: [%s:%d]\n", prox, port);
                        }
                        
                        free(prox);
                }
        }
        
        if(check_proxy_output != stdout) {
                fclose(check_proxy_output);
        }
        
        report(0);
        
        return 0;
}