static void init_options (int argc, char *argv[]) { tgetopt (argc, argv); if (toptset ('h')) { info (usage, argv[0]); exit (0); } if (toptset ('c')) config_file = toptargs ('c'); if (toptset ('s')) opt_stay++; if (toptset ('t')) test_only++; }
int main(int argc, char **argv) { int i, cur; tgetopt(argc, argv); if ( toptset('h') ) { printf(usage, argv[0]); exit(0); } if ( toptset('c') ) config_file = toptargs('c'); syslog_open("updategroups", LOG_PID, LOG_NEWS); if ( (master = memmap(sizeof(MASTER))) == NULL ) die("Can't allocate master memory"); if ( (groups = memmap(sizeof(ACTIVE) * MAX_GROUPS)) == NULL ) die("Can't allocate groups memory"); loadconfig(); load_servers(); cur = -1; for (i=master->numservers-2; i>=0; i-- ) if ( (master->lservers[i])->Level != cur ) { cur = (master->lservers[i])->Level; update_server( master->lservers[i] ); } write_active(); write_newsgroups(); memunmap(master, sizeof(MASTER)); memunmap(groups, sizeof(ACTIVE) * MAX_GROUPS); syslog_close(); exit(0); }
int tmain(int argc, tchar *argv[]) { struct options options; int opt_char; int i; int ret; program_invocation_name = get_filename(argv[0]); options.to_stdout = false; options.decompress = is_gunzip(); options.force = false; options.keep = false; options.compression_level = 6; options.suffix = T("gz"); while ((opt_char = tgetopt(argc, argv, optstring)) != -1) { switch (opt_char) { case '1': case '2': case '3': case '4': case '5': case '6': case '7': case '8': case '9': options.compression_level = parse_compression_level(opt_char, toptarg); if (options.compression_level == 0) return -1; break; case 'c': options.to_stdout = true; break; case 'd': options.decompress = true; break; case 'f': options.force = true; break; case 'h': show_usage(stdout); return 0; case 'k': options.keep = true; break; case 'S': options.suffix = toptarg; break; case 'V': show_version(); return 0; default: show_usage(stderr); return 1; } } argv += toptind; argc -= toptind; if (argc == 0) { show_usage(stderr); return 1; } for (i = 0; i < argc; i++) { if (argv[i][0] == '-' && argv[i][1] == '\0') { msg("This implementation of gzip does not yet " "support reading from standard input."); return 1; } } ret = 0; if (options.decompress) { struct deflate_decompressor *d; d = alloc_decompressor(); if (d == NULL) return 1; for (i = 0; i < argc; i++) ret |= -decompress_file(d, argv[i], &options); deflate_free_decompressor(d); } else { struct deflate_compressor *c; c = alloc_compressor(options.compression_level); if (c == NULL) return 1; for (i = 0; i < argc; i++) ret |= -compress_file(c, argv[i], &options); deflate_free_compressor(c); } /* * If ret=0, there were no warnings or errors. Exit with status 0. * If ret=2, there was at least one warning. Exit with status 2. * Else, there was at least one error. Exit with status 1. */ if (ret != 0 && ret != 2) ret = 1; return ret; }
int main(int argc, char **argv) { int opt; int o_Sndbuf_set; int num_parms; int test_num; char equiv_cmd[1024]; char *buff; char cmdbuf[512]; SOCKET sock; struct sockaddr_in sin; struct timeval tv = {1,0}; unsigned int wttl; int burst_num; /* number of bursts so far */ int msg_num; /* number of messages so far */ int send_len; /* size of datagram to send */ int sz, default_sndbuf_sz, check_size, i; int send_rtn; #if defined(_WIN32) unsigned long int iface_in; #else struct in_addr iface_in; #endif /* _WIN32 */ msend_opts opts; memset(&opts, sizeof(opts), 0); opts.prog_name = argv[0]; buff = malloc(65536); if (buff == NULL) { mprintf((&opts), "malloc failed\n"); exit(1); } memset(buff, 0, 65536); #if defined(_WIN32) { WSADATA wsadata; int wsstatus; if ((wsstatus = WSAStartup(MAKEWORD(2,2), &wsadata)) != 0) { mprintf((&opts),"%s: WSA startup error - %d\n", argv[0], wsstatus); exit(1); } } #else signal(SIGPIPE, SIG_IGN); #endif /* _WIN32 */ /* Find out what the system default socket buffer size is. */ if((sock = socket(PF_INET,SOCK_DGRAM,0)) == INVALID_SOCKET) { mprintf((&opts), "ERROR: "); perror((&opts), "socket"); exit(1); } sz = sizeof(default_sndbuf_sz); if (getsockopt(sock,SOL_SOCKET,SO_SNDBUF,(char *)&default_sndbuf_sz, (socklen_t *)&sz) == SOCKET_ERROR) { mprintf((&opts), "ERROR: "); perror((&opts), "getsockopt - SO_SNDBUF"); exit(1); } CLOSESOCKET(sock); /* default option values (declared as module globals) */ opts.o_burst_count = 1; /* 1 message per "burst" */ opts.o_decimal = 0; /* hex numbers in message text */ opts.o_loops = 1; /* number of time to loop test */ opts.o_msg_len = 0; /* variable */ opts.o_num_bursts = 0; /* infinite */ opts.o_pause = 1000; /* seconds between bursts */ opts.o_Payload = NULL; opts.o_quiet = 0; opts.o_quiet_equiv_opt = " "; opts.o_stat_pause = 0; /* no stat message */ opts.o_Sndbuf_size = MIN_DEFAULT_SENDBUF_SIZE; o_Sndbuf_set = 0; opts.o_tcp = 0; /* 0 for udp (multicast or unicast) */ opts.o_unicast_udp = 0; /* 0 for multicast or tcp */ /* default values for optional positional parms. */ opts.ttlvar = 2; opts.bind_if = NULL; test_num = -1; while ((opt = tgetopt(argc, argv, "12345b:dhl:m:n:p:P:qs:S:tu")) != EOF) { switch (opt) { case '1': test_num = 1; opts.o_burst_count = 1; opts.o_loops = 1; opts.o_msg_len = 20; opts.o_num_bursts = 600; /* 10 min */ opts.o_pause = 1000; opts.o_quiet = 1; opts.o_quiet_equiv_opt = " -q "; opts.o_stat_pause = 2000; opts.o_Sndbuf_size = MIN_DEFAULT_SENDBUF_SIZE; o_Sndbuf_set = 1; break; case '2': test_num = 2; opts.o_burst_count = 1; opts.o_loops = 1; opts.o_msg_len = 5000; opts.o_num_bursts = 5; /* 5 sec */ opts.o_pause = 1000; opts.o_quiet = 1; opts.o_quiet_equiv_opt = " -q "; opts.o_stat_pause = 2000; opts.o_Sndbuf_size = MIN_DEFAULT_SENDBUF_SIZE; o_Sndbuf_set = 1; break; case '3': test_num = 3; opts.o_burst_count = 100; opts.o_loops = 1; opts.o_msg_len = 8*1024; opts.o_num_bursts = 50; /* 5 sec */ opts.o_pause = 100; opts.o_quiet = 1; opts.o_quiet_equiv_opt = " -q "; opts.o_stat_pause = 2000; opts.o_Sndbuf_size = MIN_DEFAULT_SENDBUF_SIZE; o_Sndbuf_set = 1; break; case '4': test_num = 4; opts.o_burst_count = 5000; opts.o_loops = 1; opts.o_msg_len = 20; opts.o_num_bursts = 1; opts.o_pause = 1000; opts.o_quiet = 1; opts.o_quiet_equiv_opt = " -q "; opts.o_stat_pause = 2000; opts.o_Sndbuf_size = MIN_DEFAULT_SENDBUF_SIZE; o_Sndbuf_set = 1; break; case '5': test_num = 5; opts.o_burst_count = 50000; opts.o_loops = 1; opts.o_msg_len = 800; opts.o_num_bursts = 1; opts.o_pause = 1000; opts.o_quiet = 1; opts.o_quiet_equiv_opt = " -q "; opts.o_stat_pause = 2000; opts.o_Sndbuf_size = default_sndbuf_sz; o_Sndbuf_set = 0; break; case 'b': opts.o_burst_count = atoi(toptarg); break; case 'd': opts.o_decimal = 1; break; case 'h': help((&opts), NULL); exit(0); break; case 'l': opts.o_loops = atoi(toptarg); break; case 'm': opts.o_msg_len = atoi(toptarg); if (opts.o_msg_len > 65536) { opts.o_msg_len = 65536; mprintf((&opts), "warning, msg_len lowered to 65536\n"); } break; case 'n': opts.o_num_bursts = atoi(toptarg); break; case 'p': opts.o_pause = atoi(toptarg); break; case 'P': opts.o_msg_len = strlen(toptarg); if (opts.o_msg_len > 65536) { mprintf((&opts), "Error, payload too big\n"); exit(1); } if (opts.o_msg_len % 2 > 0) { mprintf((&opts), "Error, payload must be even number of hex digits\n"); exit(1); } opts.o_Payload = strdup(toptarg); opts.o_msg_len /= 2; /* num bytes worth of binary payload */ for (i = 0; i < opts.o_msg_len; ++i) { /* convert 2 hex digits to binary byte */ char c = opts.o_Payload[i*2]; if (c >= '0' && c <= '9') buff[i] = c - '0'; else if (c >= 'a' && c <= 'f') buff[i] = c - 'a' + 10; else if (c >= 'A' && c <= 'F') buff[i] = c - 'A' + 10; else { mprintf((&opts), "Error, invalid hex digit in payload\n"); exit(1); } c = opts.o_Payload[i*2 + 1]; if (c >= '0' && c <= '9') buff[i] = 16*buff[i] + c - '0'; else if (c >= 'a' && c <= 'f') buff[i] = 16*buff[i] + c - 'a' + 10; else if (c >= 'A' && c <= 'F') buff[i] = 16*buff[i] + c - 'A' + 10; else { mprintf((&opts), "Error, invalid hex digit in payload\n"); exit(1); } } break; case 'q': ++ opts.o_quiet; if (opts.o_quiet == 1) opts.o_quiet_equiv_opt = " -q "; else if (opts.o_quiet == 2) opts.o_quiet_equiv_opt = " -qq "; else opts.o_quiet = 2; /* never greater than 2 */ break; case 's': opts.o_stat_pause = atoi(toptarg); break; case 'S': opts.o_Sndbuf_size = atoi(toptarg); o_Sndbuf_set = 1; break; case 't': if (opts.o_unicast_udp) { mprintf((&opts), "Error, -t and -u are mutually exclusive\n"); exit(1); } opts.o_tcp = 1; break; case 'u': if (opts.o_tcp) { mprintf((&opts), "Error, -t and -u are mutually exclusive\n"); exit(1); } opts.o_unicast_udp = 1; break; default: usage((&opts), "unrecognized option"); exit(1); break; } /* switch */ } /* while opt */ /* prevent careless usage from killing the network */ if (opts.o_num_bursts == 0 && (opts.o_burst_count > 50 || opts.o_pause < 100)) { usage((&opts), "Danger - heavy traffic chosen with infinite num bursts.\nUse -n to limit execution time"); exit(1); } num_parms = argc - toptind; strcpy(equiv_cmd, "CODE BUG!!! 'equiv_cmd' not initialized"); /* handle positional parameters */ if (num_parms == 2) { opts.groupaddr = inet_addr(argv[toptind]); opts.groupport = (unsigned short)atoi(argv[toptind+1]); if (opts.o_quiet < 2) sprintf(equiv_cmd, "msend -b%d%s-m%d -n%d -p%d%s-s%d -S%d%s%s %s", opts.o_burst_count, (opts.o_decimal)?" -d ":" ", opts.o_msg_len, opts.o_num_bursts, opts.o_pause, opts.o_quiet_equiv_opt, opts.o_stat_pause, opts.o_Sndbuf_size, (opts.o_tcp) ? " -t " : ((opts.o_unicast_udp) ? " -u " : " "), argv[toptind],argv[toptind+1]); mprintf((&opts), "Equiv cmd line: %s\n", equiv_cmd); } else if (num_parms == 3) { char c; int i; opts.groupaddr = inet_addr(argv[toptind]); opts.groupport = (unsigned short)atoi(argv[toptind+1]); for (i = 0; (c = *(argv[toptind+2] + i)) != '\0'; ++i) { if (c < '0' || c > '9') { mprintf((&opts), "ERROR: third positional argument '%s' has non-numeric. Should be TTL.\n", argv[toptind+2]); exit(1); } } opts.ttlvar = (unsigned char)atoi(argv[toptind+2]); if (opts.o_quiet < 2) sprintf(equiv_cmd, "msend -b%d%s-m%d -n%d -p%d%s-s%d -S%d%s%s %s %s", opts.o_burst_count, (opts.o_decimal)?" -d ":" ", opts.o_msg_len, opts.o_num_bursts, opts.o_pause, opts.o_quiet_equiv_opt, opts.o_stat_pause, opts.o_Sndbuf_size, (opts.o_tcp) ? " -t " : ((opts.o_unicast_udp) ? " -u " : " "), argv[toptind],argv[toptind+1],argv[toptind+2]); printf("Equiv cmd line: %s\n", equiv_cmd); fflush(stdout); } else if (num_parms == 4) { opts.groupaddr = inet_addr(argv[toptind]); opts.groupport = (unsigned short)atoi(argv[toptind+1]); opts.ttlvar = (unsigned char)atoi(argv[toptind+2]); opts.bind_if = argv[toptind+3]; if (opts.o_quiet < 2) sprintf(equiv_cmd, "msend -b%d%s-m%d -n%d -p%d%s-s%d -S%d%s%s %s %s %s", opts.o_burst_count, (opts.o_decimal)?" -d ":" ", opts.o_msg_len, opts.o_num_bursts, opts.o_pause, opts.o_quiet_equiv_opt, opts.o_stat_pause, opts.o_Sndbuf_size, (opts.o_tcp) ? " -t " : ((opts.o_unicast_udp) ? " -u " : " "), argv[toptind],argv[toptind+1],argv[toptind+2],opts.bind_if); mprintf((&opts), "Equiv cmd line: %s\n", equiv_cmd); } else { usage((&opts), "need 2-4 positional parameters"); exit(1); } /* Only warn about small default send buf if no sendbuf option supplied */ if (default_sndbuf_sz < MIN_DEFAULT_SENDBUF_SIZE && o_Sndbuf_set == 0) mprintf((&opts), "NOTE: system default SO_SNDBUF only %d (%d preferred)\n", default_sndbuf_sz, MIN_DEFAULT_SENDBUF_SIZE); if (opts.o_tcp) { if((sock = socket(PF_INET,SOCK_STREAM,0)) == INVALID_SOCKET) { mprintf((&opts), "ERROR: "); perror((&opts), "socket"); exit(1); } } else { if((sock = socket(PF_INET,SOCK_DGRAM,0)) == INVALID_SOCKET) { mprintf((&opts), "ERROR: "); perror((&opts), "socket"); exit(1); } } /* Try to set send buf size and check to see if it took */ if (setsockopt(sock,SOL_SOCKET,SO_SNDBUF,(const char *)&opts.o_Sndbuf_size, sizeof(opts.o_Sndbuf_size)) == SOCKET_ERROR) { mprintf((&opts), "WARNING: "); perror((&opts), "setsockopt - SO_SNDBUF"); } sz = sizeof(check_size); if (getsockopt(sock,SOL_SOCKET,SO_SNDBUF,(char *)&check_size, (socklen_t *)&sz) == SOCKET_ERROR) { mprintf((&opts), "ERROR: "); perror((&opts), "getsockopt - SO_SNDBUF"); exit(1); } if (check_size < opts.o_Sndbuf_size) { mprintf((&opts), "WARNING: tried to set SO_SNDBUF to %d, only got %d\n", opts.o_Sndbuf_size, check_size); } memset((char *)&sin,0,sizeof(sin)); sin.sin_family = AF_INET; sin.sin_addr.s_addr = opts.groupaddr; sin.sin_port = htons(opts.groupport); if (! opts.o_unicast_udp && ! opts.o_tcp) { #if defined(_WIN32) wttl = opts.ttlvar; if (setsockopt(sock,IPPROTO_IP,IP_MULTICAST_TTL,(char *)&wttl, sizeof(wttl)) == SOCKET_ERROR) { mprintf((&opts), "ERROR: "); perror((&opts), "setsockopt - TTL"); exit(1); } #else if (setsockopt(sock,IPPROTO_IP,IP_MULTICAST_TTL,(char *)&opts.ttlvar, sizeof(opts.ttlvar)) == SOCKET_ERROR) { mprintf((&opts), "ERROR: "); perror((&opts), "setsockopt - TTL"); exit(1); } #endif /* _WIN32 */ } if (opts.bind_if != NULL) { #if !defined(_WIN32) memset((char *)&iface_in,0,sizeof(iface_in)); iface_in.s_addr = inet_addr(opts.bind_if); #else iface_in = inet_addr(opts.bind_if); #endif /* !_WIN32 */ if(setsockopt(sock, IPPROTO_IP, IP_MULTICAST_IF, (const char*)&iface_in, sizeof(iface_in)) == SOCKET_ERROR) { mprintf((&opts), "ERROR: "); perror((&opts), "setsockopt - IP_MULTICAST_IF"); exit(1); } } if (opts.o_tcp) { if((connect(sock,(struct sockaddr *)&sin,sizeof(sin))) == INVALID_SOCKET) { mprintf((&opts), "ERROR: "); perror((&opts), "connect"); exit(1); } } /* Loop the test "opts.o_loops" times (-l option) */ MAIN_LOOP: if (opts.o_num_bursts != 0) { if (opts.o_msg_len == 0) { if (opts.o_quiet < 2) { printf("Sending %d bursts of %d variable-length messages\n", opts.o_num_bursts, opts.o_burst_count); fflush(stdout); } } else { if (opts.o_quiet < 2) { printf("Sending %d bursts of %d %d-byte messages\n", opts.o_num_bursts, opts.o_burst_count, opts.o_msg_len); fflush(stdout); } } } /* 1st msg: give network hardware time to establish mcast flow */ if (test_num >= 0) sprintf(cmdbuf, "echo test %d, sender equiv cmd %s", test_num, equiv_cmd); else sprintf(cmdbuf, "echo sender equiv cmd: %s", equiv_cmd); if (opts.o_tcp) { send_rtn = send(sock,cmdbuf,strlen(cmdbuf)+1,0); } else { send_rtn = sendto(sock,cmdbuf,strlen(cmdbuf)+1,0,(struct sockaddr *)&sin,sizeof(sin)); } if (send_rtn == SOCKET_ERROR) { mprintf((&opts), "ERROR: "); perror((&opts), "send"); exit(1); } SLEEP_SEC(1); burst_num = 0; msg_num = 0; while (opts.o_num_bursts == 0 || burst_num < opts.o_num_bursts) { if (opts.o_pause > 0 && msg_num > 0) SLEEP_MSEC(opts.o_pause); /* send burst */ for (i = 0; i < opts.o_burst_count; ++i) { send_len = opts.o_msg_len; if (! opts.o_Payload) { if (opts.o_decimal) sprintf(buff,"Message %d",msg_num); else sprintf(buff,"Message %x",msg_num); if (opts.o_msg_len == 0) send_len = strlen(buff); } if (i == 0) { /* first msg in batch */ if (opts.o_quiet == 0) { /* not quiet */ if (opts.o_burst_count == 1) printf("Sending %d bytes\n", send_len); else printf("Sending burst of %d msgs\n", opts.o_burst_count); } else if (opts.o_quiet == 1) { /* pretty quiet */ printf("."); fflush(stdout); } /* else opts.o_quiet > 1; very quiet */ } send_rtn = sendto(sock,buff,send_len,0,(struct sockaddr *)&sin,sizeof(sin)); if (send_rtn == SOCKET_ERROR) { mprintf((&opts), "ERROR: "); perror((&opts), "send"); exit(1); } else if (send_rtn != send_len) { mprintf((&opts), "ERROR: sendto returned %d, expected %d\n", send_rtn, send_len); exit(1); } ++msg_num; } /* for i */ ++ burst_num; } /* while */ if (opts.o_stat_pause > 0) { /* send 'stat' message */ if (opts.o_quiet < 2) printf("Pausing before sending 'stat'\n"); SLEEP_MSEC(opts.o_stat_pause); if (opts.o_quiet < 2) printf("Sending stat\n"); sprintf(cmdbuf, "stat %d", msg_num); send_len = strlen(cmdbuf); send_rtn = sendto(sock,cmdbuf,send_len,0,(struct sockaddr *)&sin,sizeof(sin)); if (send_rtn == SOCKET_ERROR) { mprintf((&opts), "ERROR: "); perror((&opts), "send"); exit(1); } else if (send_rtn != send_len) { mprintf((&opts), "ERROR: sendto returned %d, expected %d\n", send_rtn, send_len); exit(1); } if (opts.o_quiet < 2) printf("%d messages sent (not including 'stat')\n", msg_num); } else { if (opts.o_quiet < 2) printf("%d messages sent\n", msg_num); } /* Loop the test "opts.o_loops" times (-l option) */ -- opts.o_loops; if (opts.o_loops > 0) goto MAIN_LOOP; CLOSESOCKET(sock); return(0); } /* main */