commit 490054d23955ff9f423bf645121653f3c72bde8b Author: Zakir Durumeric Date: Fri Aug 16 11:12:47 2013 -0400 inital public release diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..16cfc29 --- /dev/null +++ b/.gitignore @@ -0,0 +1,6 @@ +*.o +*.a +*.pyc +*~ +\#* +src/zmap diff --git a/AUTHORS b/AUTHORS new file mode 100644 index 0000000..e5a0436 --- /dev/null +++ b/AUTHORS @@ -0,0 +1,3 @@ +Zakir Durumeric +J. Alex Halderman +Eric Wustrow \ No newline at end of file diff --git a/CHANGELOG b/CHANGELOG new file mode 100644 index 0000000..0adce2d --- /dev/null +++ b/CHANGELOG @@ -0,0 +1,2 @@ +1.0.0 2013-8-16 +Initial public release. diff --git a/INSTALL b/INSTALL new file mode 100644 index 0000000..2fca593 --- /dev/null +++ b/INSTALL @@ -0,0 +1,25 @@ +ZMap is designed to run on GNU/Linux systems and can be built with +most recent versions of gcc. ZMap requires GMP, a free library for +arbitrary precision arithmetic, gengetopt, and libpcap. + +These packages can be installed on Debian-based systems by running: + + sudo apt-get install libgmp3-dev gengetopt libpcap-dev + +or on RHEL- and Fedora-based systems by running: + + sudo yum install gmp gmp-devel gengetopt libpcap-devel + +Once these prerequisites have been installed, ZMap can be installed by running: + + cd src + make + +followed by: + + sudo make install + +Redis support is not enabled by default. If you are want to use ZMap +with Redis, you will first need to install Hiredis. Then, rebuild +ZMap with the command "make REDIS=true". + diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..f433b1a --- /dev/null +++ b/LICENSE @@ -0,0 +1,177 @@ + + Apache License + Version 2.0, January 2004 + http://www.apache.org/licenses/ + + TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION + + 1. Definitions. + + "License" shall mean the terms and conditions for use, reproduction, + and distribution as defined by Sections 1 through 9 of this document. + + "Licensor" shall mean the copyright owner or entity authorized by + the copyright owner that is granting the License. + + "Legal Entity" shall mean the union of the acting entity and all + other entities that control, are controlled by, or are under common + control with that entity. For the purposes of this definition, + "control" means (i) the power, direct or indirect, to cause the + direction or management of such entity, whether by contract or + otherwise, or (ii) ownership of fifty percent (50%) or more of the + outstanding shares, or (iii) beneficial ownership of such entity. + + "You" (or "Your") shall mean an individual or Legal Entity + exercising permissions granted by this License. + + "Source" form shall mean the preferred form for making modifications, + including but not limited to software source code, documentation + source, and configuration files. + + "Object" form shall mean any form resulting from mechanical + transformation or translation of a Source form, including but + not limited to compiled object code, generated documentation, + and conversions to other media types. + + "Work" shall mean the work of authorship, whether in Source or + Object form, made available under the License, as indicated by a + copyright notice that is included in or attached to the work + (an example is provided in the Appendix below). + + "Derivative Works" shall mean any work, whether in Source or Object + form, that is based on (or derived from) the Work and for which the + editorial revisions, annotations, elaborations, or other modifications + represent, as a whole, an original work of authorship. For the purposes + of this License, Derivative Works shall not include works that remain + separable from, or merely link (or bind by name) to the interfaces of, + the Work and Derivative Works thereof. + + "Contribution" shall mean any work of authorship, including + the original version of the Work and any modifications or additions + to that Work or Derivative Works thereof, that is intentionally + submitted to Licensor for inclusion in the Work by the copyright owner + or by an individual or Legal Entity authorized to submit on behalf of + the copyright owner. For the purposes of this definition, "submitted" + means any form of electronic, verbal, or written communication sent + to the Licensor or its representatives, including but not limited to + communication on electronic mailing lists, source code control systems, + and issue tracking systems that are managed by, or on behalf of, the + Licensor for the purpose of discussing and improving the Work, but + excluding communication that is conspicuously marked or otherwise + designated in writing by the copyright owner as "Not a Contribution." + + "Contributor" shall mean Licensor and any individual or Legal Entity + on behalf of whom a Contribution has been received by Licensor and + subsequently incorporated within the Work. + + 2. Grant of Copyright License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + copyright license to reproduce, prepare Derivative Works of, + publicly display, publicly perform, sublicense, and distribute the + Work and such Derivative Works in Source or Object form. + + 3. Grant of Patent License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + (except as stated in this section) patent license to make, have made, + use, offer to sell, sell, import, and otherwise transfer the Work, + where such license applies only to those patent claims licensable + by such Contributor that are necessarily infringed by their + Contribution(s) alone or by combination of their Contribution(s) + with the Work to which such Contribution(s) was submitted. If You + institute patent litigation against any entity (including a + cross-claim or counterclaim in a lawsuit) alleging that the Work + or a Contribution incorporated within the Work constitutes direct + or contributory patent infringement, then any patent licenses + granted to You under this License for that Work shall terminate + as of the date such litigation is filed. + + 4. Redistribution. You may reproduce and distribute copies of the + Work or Derivative Works thereof in any medium, with or without + modifications, and in Source or Object form, provided that You + meet the following conditions: + + (a) You must give any other recipients of the Work or + Derivative Works a copy of this License; and + + (b) You must cause any modified files to carry prominent notices + stating that You changed the files; and + + (c) You must retain, in the Source form of any Derivative Works + that You distribute, all copyright, patent, trademark, and + attribution notices from the Source form of the Work, + excluding those notices that do not pertain to any part of + the Derivative Works; and + + (d) If the Work includes a "NOTICE" text file as part of its + distribution, then any Derivative Works that You distribute must + include a readable copy of the attribution notices contained + within such NOTICE file, excluding those notices that do not + pertain to any part of the Derivative Works, in at least one + of the following places: within a NOTICE text file distributed + as part of the Derivative Works; within the Source form or + documentation, if provided along with the Derivative Works; or, + within a display generated by the Derivative Works, if and + wherever such third-party notices normally appear. The contents + of the NOTICE file are for informational purposes only and + do not modify the License. You may add Your own attribution + notices within Derivative Works that You distribute, alongside + or as an addendum to the NOTICE text from the Work, provided + that such additional attribution notices cannot be construed + as modifying the License. + + You may add Your own copyright statement to Your modifications and + may provide additional or different license terms and conditions + for use, reproduction, or distribution of Your modifications, or + for any such Derivative Works as a whole, provided Your use, + reproduction, and distribution of the Work otherwise complies with + the conditions stated in this License. + + 5. Submission of Contributions. Unless You explicitly state otherwise, + any Contribution intentionally submitted for inclusion in the Work + by You to the Licensor shall be under the terms and conditions of + this License, without any additional terms or conditions. + Notwithstanding the above, nothing herein shall supersede or modify + the terms of any separate license agreement you may have executed + with Licensor regarding such Contributions. + + 6. Trademarks. This License does not grant permission to use the trade + names, trademarks, service marks, or product names of the Licensor, + except as required for reasonable and customary use in describing the + origin of the Work and reproducing the content of the NOTICE file. + + 7. Disclaimer of Warranty. Unless required by applicable law or + agreed to in writing, Licensor provides the Work (and each + Contributor provides its Contributions) on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or + implied, including, without limitation, any warranties or conditions + of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A + PARTICULAR PURPOSE. You are solely responsible for determining the + appropriateness of using or redistributing the Work and assume any + risks associated with Your exercise of permissions under this License. + + 8. Limitation of Liability. In no event and under no legal theory, + whether in tort (including negligence), contract, or otherwise, + unless required by applicable law (such as deliberate and grossly + negligent acts) or agreed to in writing, shall any Contributor be + liable to You for damages, including any direct, indirect, special, + incidental, or consequential damages of any character arising as a + result of this License or out of the use or inability to use the + Work (including but not limited to damages for loss of goodwill, + work stoppage, computer failure or malfunction, or any and all + other commercial damages or losses), even if such Contributor + has been advised of the possibility of such damages. + + 9. Accepting Warranty or Additional Liability. While redistributing + the Work or Derivative Works thereof, You may choose to offer, + and charge a fee for, acceptance of support, warranty, indemnity, + or other liability obligations and/or rights consistent with this + License. However, in accepting such obligations, You may act only + on Your own behalf and on Your sole responsibility, not on behalf + of any other Contributor, and only if You agree to indemnify, + defend, and hold each Contributor harmless for any liability + incurred by, or claims asserted against, such Contributor by reason + of your accepting any such warranty or additional liability. + + END OF TERMS AND CONDITIONS diff --git a/README b/README new file mode 100644 index 0000000..87d118c --- /dev/null +++ b/README @@ -0,0 +1,25 @@ +ZMap is a fast network scanner designed for Internet-wide network surveys. On a +typical desktop computer with a gigabit Ethernet connection, ZMap is capable +scanning the entire public IPv4 address space in under 45 minutes. + +While previous network tools have been designed to scan small network segments, +ZMap is specifically architected to scan the entire address space. It is built +in a modular manner in order to allow incorporation with other network survey +tools. ZMap operates on GNU/Linux and supports TCP SYN and ICMP echo request +scanning out of the box. + +Documentation and examples can be found at https://zmap.io/. + +-- + +ZMap Copyright 2013 Regents of the University of Michigan + +Licensed under the Apache License, Version 2.0 (the "License"); you may not use +this file except in compliance with the License. You may obtain a copy of the +License at http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software distributed +under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR +CONDITIONS OF ANY KIND, either express or implied. See LICENSE for the specific +language governing permissions and limitations under the License. + diff --git a/conf/blacklist.conf b/conf/blacklist.conf new file mode 100644 index 0000000..6b44d53 --- /dev/null +++ b/conf/blacklist.conf @@ -0,0 +1,25 @@ +# From IANA IPv4 Special-Purpose Address Registry +# http://www.iana.org/assignments/iana-ipv4-special-registry/iana-ipv4-special-registry.xhtml +# Updated 2013-05-22 + +0.0.0.0/8 # RFC1122: "This host on this network" +10.0.0.0/8 # RFC1918: Private-Use +100.64.0.0/10 # RFC6598: Shared Address Space +127.0.0.0/8 # RFC1122: Loopback +169.254.0.0/16 # RFC3927: Link Local +172.16.0.0/12 # RFC1918: Private-Use +192.0.0.0/24 # RFC6890: IETF Protocol Assignments +192.0.2.0/24 # RFC5737: Documentation (TEST-NET-1) +192.88.99.0/24 # RFC3068: 6to4 Relay Anycast +192.168.0.0/16 # RFC1918: Private-Use +192.18.0.0/15 # RFC2544: Benchmarking +198.51.100.0/24 # RFC5737: Documentation (TEST-NET-2) +203.0.113.0/24 # RFC5737: Documentation (TEST-NET-3) +240.0.0.0/4 # RFC1112: Reserved +255.255.255.255/32 # RFC0919: Limited Broadcast + +# From IANA Multicast Address Space Registry +# http://www.iana.org/assignments/multicast-addresses/multicast-addresses.xhtml +# Updated 2013-06-25 + +224.0.0.0/4 # RFC5771: Multicast/Reserved diff --git a/conf/zmap.conf b/conf/zmap.conf new file mode 100644 index 0000000..54de33b --- /dev/null +++ b/conf/zmap.conf @@ -0,0 +1,22 @@ +### Probe Module to use +#probe-module tcp_synscan + + +### Destination port to scan +#target-port 443 + +### Scan rate in packets/sec +#rate 10000 + +### Scan rate in bandwidth (bits/sec); overrides `rate` +#bandwidth 1M # 1mbps + + +### Blacklist file to use. We encourage you to exclude +### RFC1918, IANA reserved, and multicast networks, +### in addition to those who have opted out of your +### network scans. +blacklist-file "/etc/zmap/blacklist.conf" + +### Optionally print a summary at the end +#summary diff --git a/examples/banner-grab/Makefile b/examples/banner-grab/Makefile new file mode 100644 index 0000000..1c5feea --- /dev/null +++ b/examples/banner-grab/Makefile @@ -0,0 +1,22 @@ + + +CFLAGS+=-I../../lib/ -Wall +LDFLAGS+=-lpcap -levent -levent_extra -lm +VPATH=../../lib/ + +# from dpkg-buildflags --get CFLAGS, but use stack-protector-all and fPIC +GCCHARDENING=-g -O2 -fstack-protector-all --param=ssp-buffer-size=4 -Wformat -Wformat-security -Werror=format-security -fPIC +# from gpkg-buildflags --get LDFLAGS, + -z,now +LDHARDENING=-Wl,-Bsymbolic-functions -Wl,-z,relro,-z,now + +CFLAGS+=$(GCCHARDENING) +LDFLAGS+=$(LDHARDENING) + + +all: banner-grab-tcp + +banner-grab-tcp: banner-grab-tcp.o logger.o + $(CC) $(CFLAGS) $^ -o $@ $(LDFLAGS) + +clean: + rm -f banner-grab-tcp *.o diff --git a/examples/banner-grab/README b/examples/banner-grab/README new file mode 100644 index 0000000..430d1b9 --- /dev/null +++ b/examples/banner-grab/README @@ -0,0 +1,54 @@ + +TCP Banner Grab +====== + +This utility will connect (TCP) to ip addresses provide over stdin, optionally +send them a small message, and wait for their response. The response is then +printed along with their IP address on stdout. Status messages appear on stderr. + + +USING: +----- +make +#echo -e -n "GET / HTTP/1.1\r\nHost: %s\r\n\r\n" > http-req +zmap -p 80 -N 1000 -o - | ./banner-grab-tcp -p 80 -c 100 -d http-req > http-banners.out + + +OPTIONS: +----- +-c, --concurent Number of connections that can be going on at once. + This, combined with timeouts, will decide the maximum + rate at which banners are grabbed. If this value + is set higher than 1000, you should use + `ulimit -SSn 1000000` and `ulimit -SHn 1000000` to + avoid running out of file descriptors (typically capped + at 1024). + +-p, --port The port which to connect to hosts on + +-t, --conn-timeout Connection timeout (seconds). Give up on a host if connect + has not completed by this time. Default: 4 seconds. + +-r, --read-timeout Read timeout (seconds). Give up on a host if after + connecting (and optionally sending data), it does + not send any response by this time. Default: 4 seconds. + +-v, --verbosity Set status verbosity. Status/error messages are outputed + on stderr. This value can be 0-5, with 5 being the most + verbose (LOG_TRACE). Default: 3 (LOG_INFO) + +-f, --format Format to output banner responses. One of 'hex', 'ascii', + or 'base64'. + 'hex' outputs ascii hex characters, e.g. 48656c6c6f. + 'ascii' outputs ascii, without separators, e.g. Hello + 'base64' outputs base64 encoding, e.g. SGVsbG8= + Default is base64. + +-d, --data Optional data file. This data will be sent to each host + upon successful connection. Currently, this file does + not allow null characters, but supports up to 4 + occurances of the current host's IP address, by replacing + %s with the string (inet_ntoa) of that host's IP address. + + + diff --git a/examples/banner-grab/banner-grab-tcp.c b/examples/banner-grab/banner-grab-tcp.c new file mode 100644 index 0000000..8f8ad64 --- /dev/null +++ b/examples/banner-grab/banner-grab-tcp.c @@ -0,0 +1,440 @@ +/* + * Banner Grab Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include "logger.h" + +#include +#include + +#include +#include +#include +#include +#include +#include + +#define MAX_BANNER_LEN 1024 +#define BASE64_ALPHABET "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/" + +struct config { + uint16_t port; + int connect_timeout; // how long to wait for connecting (seconds) + int read_timeout; // how long to wait once connected for the banner (seconds) + int current_running; + int max_concurrent; + struct event_base *base; + struct bufferevent *stdin_bev; + int stdin_closed; + enum {FORMAT_HEX, FORMAT_BASE64, FORMAT_ASCII} format; + char *send_str; + long send_str_size; + + struct stats_st { + int init_connected_hosts; // Number of hosts we have even tried to connect to + int connected_hosts; // # hosts that picked up + int conn_timed_out; // # hosts that timed out during connection + int read_timed_out; // # hosts that connected, but sent no data (banner) + int timed_out; // # hosts that timed out at all (conn_timed_out+read_timed_out)? + int completed_hosts; // # hosts that presented a banner + } stats; +}; + + +struct state { + struct config *conf; + uint32_t ip; + enum {CONNECTING, CONNECTED, RECEIVED} state; +}; + +void stdin_readcb(struct bufferevent *bev, void *arg); + +void print_status(evutil_socket_t fd, short events, void *arg) +{ + struct event *ev; + struct config *conf = arg; + struct event_base *base = conf->base; + struct timeval status_timeout = {1, 0}; + ev = evtimer_new(base, print_status, conf); + evtimer_add(ev, &status_timeout); + (void)fd; (void)events; + + log_info("banner-grab-tcp", "(%d/%d in use) - Totals: %d inited, %d connected, %d conn timeout, %d read timeout %d completed", + conf->current_running, conf->max_concurrent, + conf->stats.init_connected_hosts, + conf->stats.connected_hosts, conf->stats.conn_timed_out, + conf->stats.read_timed_out, conf->stats.completed_hosts); +} + +void decrement_cur_running(struct state *st) +{ + struct config *conf = st->conf; + conf->current_running--; + log_debug("banner-grab-tcp", "done, down to %d", + conf->current_running); + if (evbuffer_get_length(bufferevent_get_input(conf->stdin_bev)) > 0) { + stdin_readcb(conf->stdin_bev, conf); + } + free(st); + + if (conf->stdin_closed && conf->current_running == 0) { + // Done + log_info("banner-grab-tcp", "done"); + print_status(0, 0, conf); + exit(0); + } + +} + +void connect_cb(struct bufferevent *bev, short events, void *arg) +{ + struct state *st = arg; + struct config *conf = st->conf; + struct in_addr addr; + addr.s_addr = st->ip; + if (events & BEV_EVENT_CONNECTED) { + struct timeval tv = {st->conf->read_timeout, 0}; + log_debug("banner-grab-tcp", "%s connected", inet_ntoa(addr)); + + // Send data + if (conf->send_str) { + struct evbuffer *evout = bufferevent_get_output(bev); + // HACK!!! TODO: make some messy parser that replaces ${IP} with IP etc + // and allow null characters + evbuffer_add_printf(evout, conf->send_str, + inet_ntoa(addr), inet_ntoa(addr), inet_ntoa(addr), inet_ntoa(addr)); + } + + // Change from connect timeout to read timeout + bufferevent_set_timeouts(bev, &tv, &tv); + + // Update state/stats + st->state = CONNECTED; + st->conf->stats.connected_hosts++; + } else { + if (st->state == CONNECTED) { + // Print out that we just didn't receive data + printf("%s X\n", inet_ntoa(addr)); + fflush(stdout); + + st->conf->stats.read_timed_out++; + } else { + st->conf->stats.conn_timed_out++; + } + log_debug("banner-grab-tcp", "%s bailing..", inet_ntoa(addr)); + bufferevent_free(bev); + st->conf->stats.timed_out++; + decrement_cur_running(st); + } +} + +// Grab these bytes, and close the connection. +// Even if we don't need to read any bytes, +// we have to have this so that libevent thinks we have +// a read event, so that it can timeout TCP connects +// (as a read timeout) +void read_cb(struct bufferevent *bev, void *arg) +{ + struct evbuffer *in = bufferevent_get_input(bev); + struct state *st = arg; + size_t len = evbuffer_get_length(in); + struct in_addr addr; + addr.s_addr = st->ip; + + log_debug("banner-grab-tcp", "read_cb for %s", inet_ntoa(addr)); + + if (len > MAX_BANNER_LEN) { + len = MAX_BANNER_LEN; + } + + if (len > 0) { + // Grab the banner + unsigned int i; + unsigned char *buf = malloc(len+1); + + st->state = RECEIVED; + + if (!buf) { + log_fatal("banner-grab-tcp", "cannot alloc %d byte buf", len+1); + return; + } + evbuffer_remove(in, buf, len); + + printf("%s ", inet_ntoa(addr)); + + if (st->conf->format == FORMAT_ASCII) { + // Ascii + buf[len] = '\0'; + printf("%s\n", buf); + } else if (st->conf->format == FORMAT_HEX) { + // Hex output + for (i=0; iconf->format == FORMAT_BASE64) { + // Base64 + int i=0; + char out[4] = {0,0,0,0}; + while (i < len) { + uint32_t value = 0; + value += (i < len) ? buf[i++] << 16 : 0; + value += (i < len) ? buf[i++] << 8 : 0; + value += (i < len) ? buf[i++] : 0; + out[0] = BASE64_ALPHABET[(value >> 18) & 0x3F]; + out[1] = BASE64_ALPHABET[(value >> 12) & 0x3F]; + out[2] = BASE64_ALPHABET[(value >> 6) & 0x3F]; + out[3] = BASE64_ALPHABET[(value ) & 0x3F]; + if (i < len) { + printf("%c%c%c%c", out[0], out[1], out[2], out[3]); + } + } + if (len > 0) { + switch (len % 3) { + case 1: + out[2] = '='; + case 2: + out[3] = '='; + default: + break; + } + printf("%c%c%c%c\n", out[0], out[1], out[2], out[3]); + } + } + fflush(stdout); + + free(buf); + st->conf->stats.completed_hosts++; + } + bufferevent_free(bev); + decrement_cur_running(st); +} + +void grab_banner(struct state *st) +{ + struct sockaddr_in addr; + struct bufferevent *bev; + struct timeval read_to = {st->conf->connect_timeout, 0}; + + addr.sin_family = AF_INET; + addr.sin_port = htons(st->conf->port); + addr.sin_addr.s_addr = st->ip; + log_debug("banner-grab-tcp", "connect %s:%d", + inet_ntoa(addr.sin_addr), st->conf->port); + + bev = bufferevent_socket_new(st->conf->base, -1, BEV_OPT_CLOSE_ON_FREE); + + bufferevent_set_timeouts(bev, &read_to, &read_to); + + bufferevent_setcb(bev, read_cb, NULL, connect_cb, st); + bufferevent_enable(bev, EV_READ); + + st->state = CONNECTING; + + st->conf->stats.init_connected_hosts++; + + if (bufferevent_socket_connect(bev, + (struct sockaddr *)&addr, sizeof(addr)) < 0) { + log_warn("banner-grab-tcp", "could not connect socket %d (%d open) %d %d", + bufferevent_getfd(bev), st->conf->current_running, errno, ENFILE); + perror("connect"); + + + bufferevent_free(bev); + decrement_cur_running(st); + return; + } +} + +void stdin_eventcb(struct bufferevent *bev, short events, void *ptr) { + struct config *conf = ptr; + + if (events & BEV_EVENT_EOF) { + log_debug("banner-grab-tcp", + "received EOF; quitting after buffer empties"); + conf->stdin_closed = 1; + if (conf->current_running == 0) { + log_info("banner-grab-tcp", "done"); + print_status(0, 0, conf); + exit(0); + } + } +} + +void stdin_readcb(struct bufferevent *bev, void *arg) +{ + struct evbuffer *in = bufferevent_get_input(bev); + struct config *conf = arg; + + log_debug("banner-grab-tcp", "stdin cb %d < %d ?", + conf->current_running, conf->max_concurrent); + + while (conf->current_running < conf->max_concurrent && + evbuffer_get_length(in) > 0) { + char *ip_str; + size_t line_len; + char *line = evbuffer_readln(in, &line_len, EVBUFFER_EOL_LF); + struct state *st; + if (!line) + break; + log_debug("banner-grab-tcp", "line: %s", line); + + ip_str = line; + /* + port_str = strstr(line, ":") + 1; + if (!port_str) + port_str = strstr(line, " ") + 1; + if (!port_str) + break; + + *(port_str-1) = '\0'; + port = atoi(port_str); + */ + //printf("scanning %s:%d\n", ip + + conf->current_running++; + st = malloc(sizeof(*st)); + st->conf = conf; + st->ip = inet_addr(ip_str); + grab_banner(st); + } +} + +int main(int argc, char *argv[]) +{ + struct event_base *base; + struct event *status_timer; + struct timeval status_timeout = {1, 0}; + int c; + struct option long_options[] = { + {"concurrent", required_argument, 0, 'c'}, + {"port", required_argument, 0, 'p'}, + {"conn-timeout", required_argument, 0, 't'}, + {"read-timeout", required_argument, 0, 'r'}, + {"verbosity", required_argument, 0, 'v'}, + {"format", no_argument, 0, 'f'}, + {"data", required_argument, 0, 'd'}, + {0, 0, 0, 0} }; + + struct config conf; + int ret; + FILE *fp; + + log_init(stderr, LOG_INFO); + + ret = ulimit(4, 1000000); // Allow us to open 1 million fds (instead of 1024) + if (ret < 0) { + log_fatal("banner-grab-tcp", "cannot set ulimit"); + perror("ulimit"); + exit(1); + } + + base = event_base_new(); + conf.base = base; + + // buffer stdin as an event + conf.stdin_bev = bufferevent_socket_new(base, 0, BEV_OPT_DEFER_CALLBACKS); + bufferevent_setcb(conf.stdin_bev, stdin_readcb, NULL, stdin_eventcb, &conf); + bufferevent_enable(conf.stdin_bev, EV_READ); + + // Status timer + status_timer = evtimer_new(base, print_status, &conf); + evtimer_add(status_timer, &status_timeout); + + // Defaults + conf.max_concurrent = 1; + conf.current_running = 0; + memset(&conf.stats, 0, sizeof(conf.stats)); + conf.connect_timeout = 4; + conf.read_timeout = 4; + conf.stdin_closed = 0; + conf.format = FORMAT_BASE64; + conf.send_str = NULL; + + // Parse command line args + while (1) { + int option_index = 0; + c = getopt_long(argc, argv, "c:p:t:r:v:f:d:", + long_options, &option_index); + + if (c < 0) { + break; + } + + switch (c) { + case 'c': + conf.max_concurrent = atoi(optarg); + break; + case 'p': + conf.port = atoi(optarg); + break; + case 't': + conf.connect_timeout = atoi(optarg); + break; + case 'r': + conf.read_timeout = atoi(optarg); + break; + case 'v': + if (atoi(optarg) >= 0 && atoi(optarg) <= 5) { + log_init(stderr, atoi(optarg)); + } + break; + case 'f': + if (strcmp(optarg, "hex") == 0) { + conf.format = FORMAT_HEX; + } else if (strcmp(optarg, "base64") == 0) { + conf.format = FORMAT_BASE64; + } else if (strcmp(optarg, "ascii") == 0) { + conf.format = FORMAT_ASCII; + } else { + log_fatal("banner-grab-tcp", "Unknown format '%s'; use 'hex', 'base64', or 'ascii'", + optarg); + } + break; + case 'd': + fp = fopen(optarg, "r"); + if (!fp) { + log_error("banner-grab-tcp", "Could not open send data file '%s':", optarg); + perror("fopen"); + exit(-1); + } + fseek(fp, 0L, SEEK_END); + conf.send_str_size = ftell(fp); + fseek(fp, 0L, SEEK_SET); + //assert(conf.send_str_size < 10000); // jumbo frames? + conf.send_str = malloc(conf.send_str_size+1); + if (!conf.send_str) { + log_fatal("banner-grab-tcp", "Could not malloc %d bytes", conf.send_str_size+1); + } + if (fread(conf.send_str, conf.send_str_size, 1, fp) != 1) { + log_fatal("banner-grab-tcp", "Couldn't read from send data file '%s':", optarg); + } + conf.send_str[conf.send_str_size] = '\0'; + fclose(fp); + break; + case '?': + printf("Usage:\n"); + printf("\t%s [-c max_concurrency] [-t connect_timeout] [-r read_timeout] \n\t" + "[-v verbosity=0-5] [-d send_data_file] [-f ascii|hex|base64] -p port\n", argv[0]); + exit(1); + default: + log_info("banner-grab-tcp", "hmmm.."); + break; + } + } + + log_info("banner-grab-tcp", "Using port %d with max_concurrency %d, %d s conn timeout, %d s read timeout", + conf.port, conf.max_concurrent, conf.connect_timeout, conf.read_timeout); + + event_base_dispatch(base); + + return 0; +} + diff --git a/examples/banner-grab/http-req b/examples/banner-grab/http-req new file mode 100644 index 0000000..8370bb1 --- /dev/null +++ b/examples/banner-grab/http-req @@ -0,0 +1,3 @@ +GET / HTTP/1.1 +Host: %s + diff --git a/examples/forge-socket/Makefile b/examples/forge-socket/Makefile new file mode 100644 index 0000000..be14950 --- /dev/null +++ b/examples/forge-socket/Makefile @@ -0,0 +1,22 @@ + + +CFLAGS+=-I../../lib/ -I../../forge_socket -Wall +LDFLAGS+=-lpcap -levent -levent_extra -lm +VPATH=../../lib/ + +# from dpkg-buildflags --get CFLAGS, but use stack-protector-all and fPIC +GCCHARDENING=-g -O2 -fstack-protector-all --param=ssp-buffer-size=4 -Wformat -Wformat-security -Werror=format-security -fPIC +# from gpkg-buildflags --get LDFLAGS, + -z,now +LDHARDENING=-Wl,-Bsymbolic-functions -Wl,-z,relro,-z,now + +CFLAGS+=$(GCCHARDENING) +LDFLAGS+=$(LDHARDENING) + + +all: forge-socket + +forge-socket: forge-socket.o logger.o + $(CC) $(CFLAGS) $^ -o $@ $(LDFLAGS) + +clean: + rm -f forge-socket *.o diff --git a/examples/forge-socket/README b/examples/forge-socket/README new file mode 100644 index 0000000..45a926a --- /dev/null +++ b/examples/forge-socket/README @@ -0,0 +1,73 @@ +Forge-socket banner grab +====== + +This utility, in combination with a kernel module +(https://github.com/ewust/forge_socket/) will complete the half-open connection +created by ZMap during a TCP-scan, optionally send a small message, and wait +for the hosts response. The response is then printed along with their IP +address on stdout. Periodic status messages appear on stderr. + +This utility is functionally equivalent to banner-grab-tcp, however, instead of +having the kernel send a RST packet for the server's SYN+ACK, and +banner-grab-tcp attempting to start a fresh TCP connection with the host, +forge-socket will take the parameters of the SYN+ACK packet, and use a kernel +module to add it as an ESTABLISHED TCP connection socket. Then, the +forge-socket user-space program can use this socket to send() and recv() as +normal, and completes the banner-grab process (optionally send a small message, +and receive the server's response). + + + +USING: +----- +# Install forge-socket to the ZMap root directory: +cd ./zmap/ +git clone git@github.com:ewust/forge_socket.git +cd forge_socket +make +sudo insmod forge_socket.ko + +# Don't send RST packets (forge-socket will complete these connections instead) +sudo iptables -A OUTPUT -p tcp -m tcp --tcp-flags RST,RST RST,RST -j DROP + +# Use ZMap + forge-socket simultaneously: +make +#echo -e -n "GET / HTTP/1.1\r\nHost: %s\r\n\r\n" > http-req +sudo su +ulimit -SHn 1000000 && ulimit -SSn 1000000 +zmap -p 80 -B 50M -N 1000 -O extended_file -o - | ./forge-socket -c 8000 -d http-req > http-banners.out + + +The options are similar to banner-grab-tcp, except there is no connection timeout :) + +OPTIONS: +----- +-c, --concurent Number of connections that can be going on at once. + This, combined with timeouts, will decide the maximum + rate at which banners are grabbed. If this value + is set higher than 1000, you should use + `ulimit -SSn 1000000` and `ulimit -SHn 1000000` to + avoid running out of file descriptors (typically capped + at 1024). + +-r, --read-timeout Read timeout (seconds). Give up on a host if after + connecting (and optionally sending data), it does + not send any response by this time. Default: 4 seconds. + +-v, --verbosity Set status verbosity. Status/error messages are outputed + on stderr. This value can be 0-5, with 5 being the most + verbose (LOG_TRACE). Default: 3 (LOG_INFO) + +-f, --format Format to output banner responses. One of 'hex', 'ascii', + or 'base64'. + 'hex' outputs ascii hex characters, e.g. 48656c6c6f. + 'ascii' outputs ascii, without separators, e.g. Hello + 'base64' outputs base64 encoding, e.g. SGVsbG8= + Default is base64. + +-d, --data Optional data file. This data will be sent to each host + upon successful connection. Currently, this file does + not allow null characters, but supports up to 4 + occurances of the current host's IP address, by replacing + %s with the string (inet_ntoa) of that host's IP address. + diff --git a/examples/forge-socket/forge-socket.c b/examples/forge-socket/forge-socket.c new file mode 100644 index 0000000..2795d4b --- /dev/null +++ b/examples/forge-socket/forge-socket.c @@ -0,0 +1,496 @@ +/* + * Forge Socket Banner Grab Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include "logger.h" + +#include +#include + +#include +#include +#include +#include +#include +#include + +#include "forge_socket.h" + +#define MAX_BANNER_LEN 1024 +#define BASE64_ALPHABET "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/" + +struct config { + int read_timeout; // how long to wait once connected for the banner (seconds) + int current_running; + int max_concurrent; + struct event_base *base; + struct bufferevent *stdin_bev; + int stdin_closed; + enum {FORMAT_HEX, FORMAT_BASE64, FORMAT_ASCII} format; + char *send_str; + long send_str_size; + + struct stats_st { + int init_connected_hosts; // Number of hosts we have even tried to connect to + int connected_hosts; // # hosts that picked up + int conn_timed_out; // # hosts that timed out during connection + int read_timed_out; // # hosts that connected, but sent no data (banner) + int timed_out; // # hosts that timed out at all (conn_timed_out+read_timed_out)? + int completed_hosts; // # hosts that presented a banner + } stats; +}; + + +struct state { + struct config *conf; + uint32_t src_ip; + uint32_t dst_ip; + uint16_t sport; + uint16_t dport; + uint32_t seq; + uint32_t seq_ack; + enum {CONNECTING, CONNECTED, RECEIVED} state; +}; + +void stdin_readcb(struct bufferevent *bev, void *arg); + +void print_status(evutil_socket_t fd, short events, void *arg) +{ + struct event *ev; + struct config *conf = arg; + struct event_base *base = conf->base; + struct timeval status_timeout = {1, 0}; + ev = evtimer_new(base, print_status, conf); + evtimer_add(ev, &status_timeout); + (void)fd; (void)events; + + log_info("forge-socket", "(%d/%d in use) - Totals: %d inited, %d connected, %d conn timeout, %d read timeout %d completed", + conf->current_running, conf->max_concurrent, + conf->stats.init_connected_hosts, + conf->stats.connected_hosts, conf->stats.conn_timed_out, + conf->stats.read_timed_out, conf->stats.completed_hosts); +} + +void decrement_cur_running(struct state *st) +{ + struct config *conf = st->conf; + conf->current_running--; + log_debug("forge-socket", "done, down to %d", + conf->current_running); + if (evbuffer_get_length(bufferevent_get_input(conf->stdin_bev)) > 0) { + stdin_readcb(conf->stdin_bev, conf); + } + free(st); + + if (conf->stdin_closed && conf->current_running == 0) { + // Done + log_info("forge-socket", "done"); + print_status(0, 0, conf); + exit(0); + } + +} + +void event_cb(struct bufferevent *bev, short events, void *arg) +{ + struct state *st = arg; + struct config *conf = st->conf; + struct in_addr addr; + addr.s_addr = st->src_ip; + if (events & BEV_EVENT_CONNECTED) { + log_error("forge-socket", "%s connected - wat?", inet_ntoa(addr)); + + } else { + if (st->state == CONNECTED) { + // Print out that we just didn't receive data + printf("%s X\n", inet_ntoa(addr)); + fflush(stdout); + + conf->stats.read_timed_out++; + } else { + conf->stats.conn_timed_out++; + } + log_debug("forge-socket", "%s bailing..", inet_ntoa(addr)); + bufferevent_free(bev); + conf->stats.timed_out++; + decrement_cur_running(st); + } +} + +// Grab these bytes, and close the connection. +// Even if we don't need to read any bytes, +// we have to have this so that libevent thinks we have +// a read event, so that it can timeout TCP connects +// (as a read timeout) +void read_cb(struct bufferevent *bev, void *arg) +{ + struct evbuffer *in = bufferevent_get_input(bev); + struct state *st = arg; + size_t len = evbuffer_get_length(in); + struct in_addr addr; + addr.s_addr = st->src_ip; + + log_debug("forge-socket", "read_cb for %s", inet_ntoa(addr)); + + if (len > MAX_BANNER_LEN) { + len = MAX_BANNER_LEN; + } + + if (len > 0) { + // Grab the banner + unsigned int i; + unsigned char *buf = malloc(len+1); + + st->state = RECEIVED; + + if (!buf) { + log_fatal("forge-socket", "cannot alloc %d byte buf", len+1); + return; + } + evbuffer_remove(in, buf, len); + + printf("%s ", inet_ntoa(addr)); + + if (st->conf->format == FORMAT_ASCII) { + // Ascii + buf[len] = '\0'; + printf("%s\n", buf); + } else if (st->conf->format == FORMAT_HEX) { + // Hex output + for (i=0; iconf->format == FORMAT_BASE64) { + // Base64 + int i=0; + char out[4] = {0,0,0,0}; + while (i < len) { + uint32_t value = 0; + value += (i < len) ? buf[i++] << 16 : 0; + value += (i < len) ? buf[i++] << 8 : 0; + value += (i < len) ? buf[i++] : 0; + out[0] = BASE64_ALPHABET[(value >> 18) & 0x3F]; + out[1] = BASE64_ALPHABET[(value >> 12) & 0x3F]; + out[2] = BASE64_ALPHABET[(value >> 6) & 0x3F]; + out[3] = BASE64_ALPHABET[(value ) & 0x3F]; + if (i < len) { + printf("%c%c%c%c", out[0], out[1], out[2], out[3]); + } + } + if (len > 0) { + switch (len % 3) { + case 1: + out[2] = '='; + case 2: + out[3] = '='; + default: + break; + } + printf("%c%c%c%c\n", out[0], out[1], out[2], out[3]); + } + } + fflush(stdout); + + free(buf); + st->conf->stats.completed_hosts++; + } + bufferevent_free(bev); + decrement_cur_running(st); +} + +int set_sock_state(int sock, struct tcp_state *st) +{ + struct sockaddr_in sin; + sin.sin_family = AF_INET; + sin.sin_addr.s_addr = st->src_ip; + sin.sin_port = st->sport; + + int value = 1; + if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &value, sizeof(value)) < 0) { + perror("setsockopt SO_REUSEADDR"); + return -1; + } + + if (setsockopt(sock, SOL_IP, IP_TRANSPARENT, &value, sizeof(value)) < 0) { + perror("setsockopt IP_TRANSPARENT"); + return -1; + } + + if (bind(sock, (struct sockaddr *)&sin, sizeof(sin)) < 0) { + perror("bind"); + return -1; + } + + if (setsockopt(sock, IPPROTO_TCP, TCP_STATE, st, sizeof(struct tcp_state)) < 0) { + perror("setsockopt TCP_STATE"); + return -1; + } + + return 0; +} + + + + +void grab_banner(struct state *st) +{ + struct sockaddr_in addr; + struct bufferevent *bev; + struct timeval read_to = {st->conf->read_timeout, 0}; + struct tcp_state tcp_st; + int sock = socket(AF_INET, SOCK_FORGE, 0); + + addr.sin_addr.s_addr = st->src_ip; + + if (sock < 0) { + perror("SOCK_FORGE socket"); + log_fatal("forge_socket", "(did you insmod forge_socket.ko?)"); + return; + } + + memset(&tcp_st, 0, sizeof(tcp_st)); + + // These need to be in network order for forge socket" + tcp_st.src_ip = st->dst_ip; + tcp_st.dst_ip = st->src_ip; + tcp_st.sport = htons(st->dport); + tcp_st.dport = htons(st->sport); + + // This should be in ??? + tcp_st.seq = st->seq_ack; + tcp_st.ack = (st->seq + 1); + + tcp_st.snd_wnd = 0x1000; + tcp_st.rcv_wnd = 0x1000; + + tcp_st.snd_una = tcp_st.seq; + st->state = CONNECTING; + st->conf->stats.init_connected_hosts++; + + // consider this a non-blocking, but completed "connect()". heh. + if (set_sock_state(sock, &tcp_st) != 0) { + log_error("forge_socket", "set_sock_state failed\n"); + decrement_cur_running(st); + return; + } + + evutil_make_socket_nonblocking(sock); + + bev = bufferevent_socket_new(st->conf->base, sock, BEV_OPT_CLOSE_ON_FREE); + + bufferevent_set_timeouts(bev, &read_to, &read_to); + + bufferevent_setcb(bev, read_cb, NULL, event_cb, st); + bufferevent_enable(bev, EV_READ); + + + // Send data + if (st->conf->send_str) { + struct evbuffer *evout = bufferevent_get_output(bev); + // HACK!!! TODO: make some messy parser that replaces ${IP} with IP etc + // and allow null characters + evbuffer_add_printf(evout, st->conf->send_str, + inet_ntoa(addr.sin_addr), inet_ntoa(addr.sin_addr), + inet_ntoa(addr.sin_addr), inet_ntoa(addr.sin_addr)); + log_trace("forge-socket", "sent str to %s", inet_ntoa(addr.sin_addr)); + } + + // Update state/stats + st->state = CONNECTED; + st->conf->stats.connected_hosts++; + + log_trace("forge-socket", "go %s go! read a byte!!", inet_ntoa(addr.sin_addr)); +} + +void stdin_eventcb(struct bufferevent *bev, short events, void *ptr) { + struct config *conf = ptr; + + if (events & BEV_EVENT_EOF) { + log_debug("forge-socket", + "received EOF; quitting after buffer empties"); + conf->stdin_closed = 1; + if (conf->current_running == 0) { + log_info("forge-socket", "done"); + print_status(0, 0, conf); + exit(0); + } + } +} + +void stdin_readcb(struct bufferevent *bev, void *arg) +{ + struct evbuffer *in = bufferevent_get_input(bev); + struct config *conf = arg; + + log_debug("forge-socket", "stdin cb %d < %d ?", + conf->current_running, conf->max_concurrent); + + while (conf->current_running < conf->max_concurrent && + evbuffer_get_length(in) > 0) { + size_t line_len; + char *line = evbuffer_readln(in, &line_len, EVBUFFER_EOL_LF); + struct state *st; + if (!line) + break; + log_debug("forge-socket", "line: '%s'", line); + + //synack, 77.176.116.205, 141.212.121.125, 443, 49588, 3628826326, 3441755636, 0, 0,2013-08-11 19:16:05.799 + char synack[12]; + char srcip[INET_ADDRSTRLEN], dstip[INET_ADDRSTRLEN]; + uint32_t seq, seq_ack; + uint16_t sport, dport; + int cooldown, repeat=1; + + + int ret = sscanf(line, "%11[^,], %15[^,], %15[^,], %hu, %hu, %u, %u, %d, %d,%*s", + synack, srcip, dstip, &sport, &dport, &seq, &seq_ack, &cooldown, &repeat); + + log_trace("forge-socket", "%d '%s' sip: '%s', dip: '%s', sport: %d, dport: %d, seq: %d, seq_ack: %d", + ret, synack, srcip, dstip, sport, dport, seq, seq_ack); + + if (ret==9 && !repeat && strcmp(synack, "synack") == 0) { + st = malloc(sizeof(*st)); + st->conf = conf; + st->src_ip = inet_addr(srcip); + st->dst_ip = inet_addr(dstip); + st->sport = sport; + st->dport = dport; + st->seq = seq; + st->seq_ack = seq_ack; + + conf->current_running++; + grab_banner(st); + } + } +} + +int main(int argc, char *argv[]) +{ + struct event_base *base; + struct event *status_timer; + struct timeval status_timeout = {1, 0}; + int c; + struct option long_options[] = { + {"concurrent", required_argument, 0, 'c'}, + {"read-timeout", required_argument, 0, 'r'}, + {"verbosity", required_argument, 0, 'v'}, + {"format", no_argument, 0, 'f'}, + {"data", required_argument, 0, 'd'}, + {0, 0, 0, 0} }; + + struct config conf; + int ret; + FILE *fp; + + log_init(stderr, LOG_INFO); + + ret = ulimit(4, 1000000); // Allow us to open 1 million fds (instead of 1024) + if (ret < 0) { + log_fatal("forge-socket", "cannot set ulimit"); + perror("ulimit"); + exit(1); + } + + base = event_base_new(); + conf.base = base; + + // buffer stdin as an event + conf.stdin_bev = bufferevent_socket_new(base, 0, BEV_OPT_DEFER_CALLBACKS); + bufferevent_setcb(conf.stdin_bev, stdin_readcb, NULL, stdin_eventcb, &conf); + bufferevent_enable(conf.stdin_bev, EV_READ); + + // Status timer + status_timer = evtimer_new(base, print_status, &conf); + evtimer_add(status_timer, &status_timeout); + + // Defaults + conf.max_concurrent = 1; + conf.current_running = 0; + memset(&conf.stats, 0, sizeof(conf.stats)); + conf.read_timeout = 4; + conf.stdin_closed = 0; + conf.format = FORMAT_BASE64; + conf.send_str = NULL; + + // Parse command line args + while (1) { + int option_index = 0; + c = getopt_long(argc, argv, "c:t:r:v:f:d:", + long_options, &option_index); + + if (c < 0) { + break; + } + + switch (c) { + case 'c': + conf.max_concurrent = atoi(optarg); + break; + case 'r': + conf.read_timeout = atoi(optarg); + break; + case 'v': + if (atoi(optarg) >= 0 && atoi(optarg) <= 5) { + log_init(stderr, atoi(optarg)); + } + break; + case 'f': + if (strcmp(optarg, "hex") == 0) { + conf.format = FORMAT_HEX; + } else if (strcmp(optarg, "base64") == 0) { + conf.format = FORMAT_BASE64; + } else if (strcmp(optarg, "ascii") == 0) { + conf.format = FORMAT_ASCII; + } else { + log_fatal("forge-socket", "Unknown format '%s'; use 'hex', 'base64', or 'ascii'", + optarg); + } + break; + case 'd': + fp = fopen(optarg, "r"); + if (!fp) { + log_error("forge-socket", "Could not open send data file '%s':", optarg); + perror("fopen"); + exit(-1); + } + fseek(fp, 0L, SEEK_END); + conf.send_str_size = ftell(fp); + fseek(fp, 0L, SEEK_SET); + //assert(conf.send_str_size < 10000); // jumbo frames? + conf.send_str = malloc(conf.send_str_size+1); + if (!conf.send_str) { + log_fatal("forge-socket", "Could not malloc %d bytes", conf.send_str_size+1); + } + if (fread(conf.send_str, conf.send_str_size, 1, fp) != 1) { + log_fatal("forge-socket", "Couldn't read from send data file '%s':", optarg); + } + conf.send_str[conf.send_str_size] = '\0'; + fclose(fp); + break; + case '?': + printf("Usage:\n"); + printf("\t%s [-c max_concurrency] [-r read_timeout] \n\t" + "[-v verbosity=0-5] [-d send_data_file] [-f ascii|hex|base64]\n", argv[0]); + exit(1); + default: + log_info("forge-socket", "hmmm.."); + break; + } + } + + log_info("forge-socket", "Using max_concurrency %d, %d s read timeout", + conf.max_concurrent, conf.read_timeout); + + event_base_dispatch(base); + + return 0; +} + diff --git a/lib/blacklist.c b/lib/blacklist.c new file mode 100644 index 0000000..9647e4b --- /dev/null +++ b/lib/blacklist.c @@ -0,0 +1,129 @@ +/* + * Blacklist Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include "blacklist.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include "constraint.h" +#include "logger.h" + +#define ADDR_DISALLOWED 0 +#define ADDR_ALLOWED 1 + +static constraint_t *constraint = NULL; + +// check whether a single IP address is allowed to be scanned. +// 1 => is allowed +// 0 => is not allowed +int blacklist_is_allowed(uint32_t s_addr) { + return constraint_lookup_ip(constraint, ntohl(s_addr)) == ADDR_ALLOWED; +} + +// blacklist a CIDR network allocation +// e.g. blacklist_add("128.255.134.0", 24) +void blacklist_prefix(char *ip, int prefix_len) +{ + assert(constraint); + constraint_set(constraint, ntohl(inet_addr(ip)), prefix_len, ADDR_DISALLOWED); +} + +// whitelist a CIDR network allocation +void whitelist_prefix(char *ip, int prefix_len) +{ + assert(constraint); + constraint_set(constraint, ntohl(inet_addr(ip)), prefix_len, ADDR_ALLOWED); +} + +static int init(char *file, const char *name, int value) +{ + FILE *fp; + char line[1000]; + + fp = fopen(file, "r"); + if (fp == NULL) { + log_fatal(name, "Unable to open %s file: %s: %s", name, file, strerror(errno)); + } + + while (fgets(line, sizeof(line), fp) != NULL) { + char *comment = strchr(line, '#'); + if (comment) { + *comment = '\0'; + } + char ip[33]; + if ((sscanf(line, "%32s", ip)) == EOF) { + continue; + } + int prefix_len = 32; + char *slash = strchr(ip, '/'); + if (slash) { // split apart network and prefix length + *slash = '\0'; + char *end; + char *len = slash+1; + errno = 0; + prefix_len = strtol(len, &end, 10); + if (end == len || errno != 0 || prefix_len < 0 || prefix_len > 32) { + log_fatal(name, "Unable to parse %s file: %s ('%s' is not a valid prefix length)", name, file, len); + } + } + struct in_addr addr; + if (inet_aton(ip, &addr) == 0) { + log_fatal(name, "Unable to parse %s file: %s ('%s' is not a valid IP address)", name, file, ip); + } + constraint_set(constraint, ntohl(addr.s_addr), prefix_len, value); + log_trace(name, "%sing %s/%i", + name, ip, prefix_len); + } + fclose(fp); + + return 0; +} + +uint64_t blacklist_count_allowed() +{ + assert(constraint); + return constraint_count_ips(constraint, ADDR_ALLOWED); +} + +uint64_t blacklist_count_not_allowed() +{ + assert(constraint); + return constraint_count_ips(constraint, ADDR_DISALLOWED); +} + + +// Initialize address constraints from whitelist and blacklist files. +// Either can be set to NULL to omit. +int blacklist_init_from_files(char *whitelist_filename, char *blacklist_filename) +{ + assert(!constraint); + if (whitelist_filename) { + // using a whitelist, so default to allowing nothing + constraint = constraint_init(ADDR_DISALLOWED); + log_trace("whitelist", "blacklisting 0.0.0.0/0"); + init(whitelist_filename, "whitelist", ADDR_ALLOWED); + } else { + // no whitelist, so default to allowing everything + constraint = constraint_init(ADDR_ALLOWED); + } + if (blacklist_filename) { + init(blacklist_filename, "blacklist", ADDR_DISALLOWED); + } + constraint_optimize(constraint); + uint64_t allowed = blacklist_count_allowed(); + log_debug("blacklist", "%lu addresses allowed to be scanned (%0.0f%% of address space)", + allowed, allowed*100./(1L << 32)); + return 0; +} diff --git a/lib/blacklist.h b/lib/blacklist.h new file mode 100644 index 0000000..5604171 --- /dev/null +++ b/lib/blacklist.h @@ -0,0 +1,13 @@ +#include + +#ifndef _BLACKLIST_H +#define _BLACKLIST_H + +int blacklist_is_allowed(uint32_t s_addr); +void blacklist_prefix(char *ip, int prefix_len); +void whitelist_prefix(char *ip, int prefix_len); +int blacklist_init_from_files(char *whitelist, char*blacklist); +uint64_t blacklist_count_allowed(); +uint64_t blacklist_count_not_allowed(); + +#endif diff --git a/lib/constraint.c b/lib/constraint.c new file mode 100644 index 0000000..aa8d0b9 --- /dev/null +++ b/lib/constraint.c @@ -0,0 +1,384 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "../lib/constraint.h" +#include "../lib/logger.h" + +// +// Efficient address-space constraints (AH 7/2013) +// +// This module uses a tree-based representation to efficiently +// manipulate and query constraints on the address space to be +// scanned. It provides a value for every IP address, and these +// values are applied by setting them for network prefixes. Order +// matters: setting a value replaces any existing value for that +// prefix or subsets of it. We use this to implement network +// whitelisting and blacklisting. +// +// Think of setting values in this structure like painting +// subnets with different colors. We can paint subnets black to +// exclude them and white to allow them. Only the top color shows. +// This makes for potentially very powerful constraint specifications. +// +// Internally, this is implemented using a binary tree, where each +// node corresponds to a network prefix. (E.g., the root is +// 0.0.0.0/0, and its children, if present, are 0.0.0.0/1 and +// 128.0.0.0/1.) Each leaf of the tree stores the value that applies +// to every address within the leaf's portion of the prefix space. +// +// As an optimization, after all values are set, we look up the +// value or subtree for every /16 prefix and cache them as an array. +// This lets subsequent lookups bypass the bottom half of the tree. +// + +/* + * Constraint Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +typedef struct node { + struct node *l; + struct node *r; + value_t value; +} node_t; + +// As an optimization, we precompute lookups for every prefix of this +// length: +#define RADIX_LENGTH 16 + +struct _constraint { + node_t *root; // root node of the tree + node_t **radix; // array of nodes for every RADIX_LENGTH prefix + int optimized; // is radix populated and up-to-date? +}; + +// Tree operations respect the invariant that every node that isn't a +// leaf has exactly two children. +#define IS_LEAF(node) ((node)->l == NULL) + +// Allocate a new leaf with the given value +static node_t* _create_leaf(value_t value) +{ + node_t *node = malloc(sizeof(node_t)); + assert(node); + node->l = NULL; + node->r = NULL; + node->value = value; + return node; +} + +// Free the subtree rooted at node. +static void _destroy_subtree(node_t *node) +{ + if (node == NULL) + return; + _destroy_subtree(node->l); + _destroy_subtree(node->r); + free(node); +} + +// Convert from an internal node to a leaf. +static void _convert_to_leaf(node_t *node) +{ + assert(node); + assert(!IS_LEAF(node)); + _destroy_subtree(node->l); + _destroy_subtree(node->r); + node->l = NULL; + node->r = NULL; +} + +// Recursive function to set value for a given network prefix within +// the tree. (Note: prefix must be in host byte order.) +static void _set_recurse(node_t *node, uint32_t prefix, int len, value_t value) +{ + assert(node); + assert(0 <= len && len <= 32); + + if (len == 0) { + // We're at the end of the prefix; make this a leaf and set the value. + if (!IS_LEAF(node)) { + _convert_to_leaf(node); + } + node->value = value; + return; + } + + if (IS_LEAF(node)) { + // We're not at the end of the prefix, but we hit a leaf. + if (node->value == value) { + // A larger prefix has the same value, so we're done. + return; + } + // The larger prefix has a different value, so we need to convert it + // into an internal node and continue processing on one of the leaves. + node->l = _create_leaf(node->value); + node->r = _create_leaf(node->value); + } + + // We're not at the end of the prefix, and we're at an internal + // node. Recurse on the left or right subtree. + if (prefix & 0x80000000) { + _set_recurse(node->r, prefix << 1, len - 1, value); + } else { + _set_recurse(node->l, prefix << 1, len - 1, value); + } + + // At this point, we're an internal node, and the value is set + // by one of our children or its descendent. If both children are + // leaves with the same value, we can discard them and become a left. + if (IS_LEAF(node->r) && IS_LEAF(node->l) && node->r->value == node->l->value) { + node->value = node->l->value; + _convert_to_leaf(node); + } +} + +// Set the value for a given network prefix, overwriting any existing +// values on that prefix or subsets of it. +// (Note: prefix must be in host byte order.) +void constraint_set(constraint_t *con, uint32_t prefix, int len, value_t value) +{ + assert(con); + _set_recurse(con->root, prefix, len, value); + con->optimized = 0; +} + +// Return the value pertaining to an address, according to the tree +// starting at given root. (Note: address must be in host byte order.) +static int _lookup_ip(node_t *root, uint32_t address) +{ + assert(root); + node_t *node = root; + uint32_t mask = 0x80000000; + for (;;) { + if (IS_LEAF(node)) { + return node->value; + } + if (address & mask) { + node = node->r; + } else { + node = node->l; + } + mask >>= 1; + } +} + + +// Return the value pertaining to an address. +// (Note: address must be in host byte order.) +int constraint_lookup_ip(constraint_t *con, uint32_t address) +{ + assert(con); + if (con->optimized) { + // Use radix optimization + node_t *node = con->radix[address >> (32 - RADIX_LENGTH)]; + if (IS_LEAF(node)) { + return node->value; + } + return _lookup_ip(node, address << RADIX_LENGTH); + } else { + // Do a full lookup using the tree + log_trace("constraint", "Unoptimized lookup"); + return _lookup_ip(con->root, address); + } +} + +// Implement count_ips by recursing on halves of the tree. Size represents +// the number of addresses in a prefix at the current level of the tree. +static uint64_t _count_ips_recurse(node_t *node, value_t value, uint64_t size) +{ + assert(node); + if (IS_LEAF(node)) { + if (node->value == value) { + return size; + } else { + return 0; + } + } + return _count_ips_recurse(node->l, value, size >> 1) + + _count_ips_recurse(node->r, value, size >> 1); +} + +// Return the number of addresses that have a given value. +uint64_t constraint_count_ips(constraint_t *con, value_t value) +{ + assert(con); + return _count_ips_recurse(con->root, value, (uint64_t)1 << 32); +} + +// Initialize the tree. +// All addresses will initally have the given value. +constraint_t* constraint_init(value_t value) +{ + log_trace("constraint", "Initializing"); + constraint_t* con = malloc(sizeof(constraint_t)); + con->root = _create_leaf(value); + con->radix = calloc(sizeof(node_t *), 1 << RADIX_LENGTH); + assert(con->radix); + con->optimized = 0; + return con; +} + +// Deinitialize and free the tree. +void constraint_free(constraint_t *con) +{ + assert(con); + log_trace("constraint", "Cleaning up"); + _destroy_subtree(con->root); + free(con->radix); + free(con); +} + +// Return a node that determines the values for the addresses with +// the given prefix. This is either the internal node that +// corresponds to the end of the prefix or a leaf node that +// encompasses the prefix. (Note: prefix must be in host byte order.) +static node_t* _lookup_node(node_t *root, uint32_t prefix, int len) +{ + assert(root); + assert(0 <= len && len <= 32); + + node_t *node = root; + uint32_t mask = 0x80000000; + + for (int i=0; i < len; i++) { + if (IS_LEAF(node)) { + return node; + } + if (prefix & mask) { + node = node->r; + } else { + node = node->l; + } + mask >>= 1; + } + return node; +} + +// After values have been set, precompute prefix lookups. +void constraint_optimize(constraint_t *con) +{ + assert(con); + if (con->optimized) { + return; + } + log_trace("constraint", "Optimizing constraints"); + for (uint32_t i=0; i < (1 << RADIX_LENGTH); i++) { + uint32_t prefix = i << (32 - RADIX_LENGTH); + con->radix[i] = _lookup_node(con->root, prefix, RADIX_LENGTH); + } + con->optimized = 1; +} + + +/* +int main(void) +{ + log_init(stderr, LOG_DEBUG); + + constraint_t *con = constraint_init(0); + constraint_set(con, ntohl(inet_addr("128.128.0.0")), 1, 22); + constraint_set(con, ntohl(inet_addr("128.128.0.0")), 1, 1); + constraint_set(con, ntohl(inet_addr("128.0.0.0")), 1, 1); + constraint_set(con, ntohl(inet_addr("10.0.0.0")), 24, 1); + constraint_set(con, ntohl(inet_addr("10.0.0.0")), 24, 0); + constraint_set(con, ntohl(inet_addr("10.11.12.0")), 24, 1); + constraint_set(con, ntohl(inet_addr("141.212.0.0")), 16, 0); + + for (int x=1; x < 2; x++) { + if (x == 1) { + constraint_optimize(con); + } + + printf("count(0)=%ld\n", constraint_count_ips(con, 0)); + printf("count(1)=%ld\n", constraint_count_ips(con, 1)); + printf("%d\n", constraint_lookup_ip(con,ntohl(inet_addr("10.11.12.0")))); + assert(constraint_count_ips(con, 0) + constraint_count_ips(con, 1) == (uint64_t)1 << 32); + + uint32_t i=0, count=0; + do { + if (constraint_lookup_ip(con, i)) + count++; + } while (++i != 0); + printf("derived count(1)=%u\n", count); + } + + constraint_free(con); +} +*/ + +/* +static int init(constraint_t *con, char *file, const char *name, value_t value) +{ + FILE *fp; + char line[1000]; + int blocked = 0; + + fp = fopen(file, "r"); + if (fp == NULL) { + log_fatal(name, "Unable to open %s file: %s: %s", + name, file, strerror(errno)); + } + + while (fgets(line, sizeof(line), fp) != NULL) { + char *comment = strchr(line, '#'); + if (comment) { + *comment = '\0'; + } + char ip[33]; + if ((sscanf(line, "%32s", ip)) == EOF) { + continue; + } + int prefix_len; + char *slash = strchr(ip, '/'); + if (slash == NULL) { + log_fatal(name, + "Unable to parse %s file: %s", + name, file); + } + // split apart network and prefix length + *slash = '\0'; + prefix_len = atoi(&slash[1]); + constraint_set(con, ntohl(inet_addr(ip)), prefix_len, value); + + blocked++; + } + fclose(fp); + return 0; +} + + + +void main() +{ + log_init(stderr, LOG_TRACE); + + constraint_t *con = constraint_init(1); + init(con, "blacklist.prefixes", "blacklist", 0); + //constraint_optimize(con); + + printf("count(0)=%lu\n", constraint_count_ips(con, 0)); + printf("count(1)=%lu\n", constraint_count_ips(con, 1)); + + uint32_t i=0, count=0; + do { + if (constraint_lookup_ip(con, i)) + count++; + } while (++i != 0); + printf("derived count(1)=%u\n", count); + + constraint_free(con); + +} + + */ diff --git a/lib/constraint.h b/lib/constraint.h new file mode 100644 index 0000000..3e914a3 --- /dev/null +++ b/lib/constraint.h @@ -0,0 +1,14 @@ +#ifndef _CONSTRAINT_H +#define _CONSTRAINT_H + +typedef struct _constraint constraint_t; +typedef int value_t; + +constraint_t* constraint_init(value_t value); +void constraint_free(constraint_t *con); +void constraint_set(constraint_t *con, uint32_t prefix, int len, value_t value); +void constraint_optimize(constraint_t *con); +int constraint_lookup_ip(constraint_t *con, uint32_t address); +uint64_t constraint_count_ips(constraint_t *con, value_t value); + +#endif //_CONSTRAINT_H diff --git a/lib/logger.c b/lib/logger.c new file mode 100644 index 0000000..8a797e7 --- /dev/null +++ b/lib/logger.c @@ -0,0 +1,126 @@ +/* + * Logger Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include +#include +#include +#include + +#include "logger.h" + +#ifndef HEADER_ZUTIL_LOGGER_H +#define HEADER_ZUTIL_LOGGER_H + +static enum LogLevel log_output_level = LOG_INFO; + +static FILE *log_output_stream = NULL; + +static const char *log_level_name[] = { + "FATAL", "ERROR", "WARN", "INFO", "DEBUG", "TRACE" }; + +static int LogLogVA(enum LogLevel level, const char *loggerName, + const char *logMessage, va_list args) +{ + if (!log_output_stream) { + log_output_stream = stdout; + } + if (level <= log_output_level) { + const char *levelName = log_level_name[level]; + assert(level < sizeof(log_level_name)); + + struct timeval now; + char timestamp[256]; + gettimeofday(&now, NULL); + time_t sec = now.tv_sec; + struct tm* ptm = localtime(&sec); + strftime(timestamp, 20, "%b %d %H:%M:%S", ptm); + fprintf(log_output_stream, "%s.%03ld [%s] ", + timestamp, now.tv_usec/1000, levelName); + if (loggerName) { + fprintf(log_output_stream, "%s: ", loggerName); + } + if (logMessage) { + vfprintf(log_output_stream, logMessage, args); + } + if (loggerName || logMessage) { + fputs("\n", log_output_stream); + } + } + return 0; +} + +int log_fatal(const char *name, const char *message, ...) { + va_list va; va_start(va, message); + LogLogVA(LOG_FATAL, name, message, va); + va_end(va); + exit(EXIT_FAILURE); +} + +int log_error(const char *name, const char *message, ...) { + va_list va; va_start(va, message); + int ret = LogLogVA(LOG_ERROR, name, message, va); + va_end(va); + return ret; +} + +int log_warn(const char *name, const char *message, ...) { + va_list va; va_start(va, message); + int ret = LogLogVA(LOG_WARN, name, message, va); + va_end(va); + return ret; +} + +int log_info(const char *name, const char *message, ...) { + va_list va; va_start(va, message); + int ret = LogLogVA(LOG_INFO, name, message, va); + va_end(va); + return ret; +} + +int log_debug(const char *name, const char *message, ...) { + va_list va; va_start(va, message); + int ret = LogLogVA(LOG_DEBUG, name, message, va); + va_end(va); + return ret; +} + +extern int log_trace(const char *name, const char *message, ...) { + va_list va; va_start(va, message); + int ret = LogLogVA(LOG_TRACE, name, message, va); + va_end(va); + return ret; +} + +int log_init(FILE *stream, enum LogLevel level) +{ + log_output_stream = stream; + log_output_level = level; + return 0; +} + +double now(void) +{ + struct timeval now; + gettimeofday(&now, NULL); + return (double)now.tv_sec + (double)now.tv_usec/1000000.; +} + +size_t dstrftime(char *buf, size_t maxsize, const char *format, double tm) +{ + struct timeval tv; + double tm_floor; + tm_floor = floor(tm); + tv.tv_sec = (long) tm_floor; + tv.tv_usec = (long) (tm - floor(tm)) * 1000000; + return strftime(buf, maxsize, format, localtime((const time_t*) &tv)); +} + +#endif diff --git a/lib/logger.h b/lib/logger.h new file mode 100644 index 0000000..8cf2a9f --- /dev/null +++ b/lib/logger.h @@ -0,0 +1,29 @@ +#include +#include + +#ifndef _LOGGER_H +#define _LOGGER_H + +enum LogLevel { LOG_FATAL, LOG_ERROR, LOG_WARN, LOG_INFO, LOG_DEBUG, LOG_TRACE, + NUM_LOGLEVELS }; + +int log_fatal(const char *loggerName, const char *logMessage, ...) __attribute__((noreturn)); + +int log_error(const char *loggerName, const char *logMessage, ...); + +int log_warn(const char *loggerName, const char *logMessage, ...); + +int log_info(const char *loggerName, const char *logMessage, ...); + +int log_debug(const char *loggerName, const char *logMessage, ...); + +int log_trace(const char *loggerName, const char *logMessage, ...); + +int log_init(FILE *stream, enum LogLevel level); + +size_t dstrftime(char *, size_t, const char *, double); + +double now(); + +#endif // _LOGGER_H + diff --git a/lib/random.c b/lib/random.c new file mode 100644 index 0000000..e2e6404 --- /dev/null +++ b/lib/random.c @@ -0,0 +1,28 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include "random.h" +#include +#include +#include +#include +#include + +#define RANDSRC "/dev/urandom" + +int random_bytes(void *dst, size_t n) +{ + FILE *f = fopen(RANDSRC, "rb"); + assert(f); + size_t r = fread(dst, n, 1, f); + fclose(f); + if (r < 1) { + return 0; + } + return 1; +} diff --git a/lib/random.h b/lib/random.h new file mode 100644 index 0000000..d83b14d --- /dev/null +++ b/lib/random.h @@ -0,0 +1,9 @@ +#include +#include + +#ifndef _RANDOM_H +#define _RANDOM_H + +int random_bytes(void *dst, size_t n); + +#endif diff --git a/lib/redis.c b/lib/redis.c new file mode 100644 index 0000000..20bf4e7 --- /dev/null +++ b/lib/redis.c @@ -0,0 +1,261 @@ +/* + * ZMap Redis Helpers Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include "redis.h" +#include +#include +#include "assert.h" +#include "logger.h" +#include +#include + +#define REDIS_UNIX_PATH "/tmp/redis.sock" +#define REDIS_TIMEOUT 2 + +#undef MIN +#define MIN(X,Y) ((X) < (Y) ? (X) : (Y)) + +static redisContext *rctx; + +static redisContext* redis_connect(void) +{ + struct timeval timeout; + timeout.tv_sec = REDIS_TIMEOUT; + timeout.tv_usec = 0; + return (redisContext*) redisConnectUnixWithTimeout(REDIS_UNIX_PATH, + timeout); +} + +static int chkerr(redisReply *reply) +{ + assert(rctx); + if (reply == NULL || reply->type == REDIS_REPLY_ERROR) { + log_error("redis", "an error occurred when " + "retreiving item from redis: %s", + rctx->errstr); + if (reply) { + freeReplyObject(reply); + } + return -1; + } + return 0; +} + +int redis_init(void) +{ + rctx = redis_connect(); + if (!rctx) { + return -1; + } + return 0; +} + +int redis_close(void) +{ + redisFree(rctx); + return 0; +} + +redisContext* redis_get_context(void) +{ + return rctx; +} + +int redis_flush(void) +{ + redisReply *reply = (redisReply*) redisCommand(rctx, "FLUSHDB"); + if (chkerr(reply)) { + return -1; + } + freeReplyObject(reply); + return 0; +} + +int redis_existconf(const char *name) +{ + assert(rctx); + redisReply *reply = (redisReply*) redisCommand(rctx, "EXISTS %s", name); + if (chkerr(reply)) { + return -1; + } + int v = reply->integer; + freeReplyObject(reply); + return v; +} + +int redis_delconf(const char *name) +{ + assert(rctx); + redisReply *reply = (redisReply*) redisCommand(rctx, "DEL %s", name); + if (chkerr(reply)) { + return -1; + } + freeReplyObject(reply); + return 0; +} + +int redis_setconf(const char *name, char *value) +{ + assert(rctx); + redisReply *reply = (redisReply*) redisCommand(rctx, "SET %s %s", + name, value); + if (chkerr(reply)) { + return -1; + } + freeReplyObject(reply); + return 0; +} + +int redis_getconf(const char *name, char *buf, size_t maxlen) +{ + assert(rctx); + redisReply *reply = (redisReply*) redisCommand(rctx, "GET %s", name); + if (chkerr(reply)) { + return -1; + } + strncpy(buf, reply->str, maxlen); + freeReplyObject(reply); + return 0; +} + +uint32_t redis_getconf_uint32_t(const char *key) +{ + assert(rctx); + char buf[50]; + redis_getconf(key, buf, 50); + return atoi(buf); +} + +int redis_setconf_uint32_t(const char *key, uint32_t value) +{ + assert(rctx); + char buf[50]; + sprintf(buf, "%u", value); + return redis_setconf(key, buf); +} + + +static long redis_get_sizeof(const char *cmd, const char *name) +{ + assert(rctx); + redisReply *reply; + reply = (redisReply*) redisCommand(rctx, "%s %s", cmd, name); + assert(reply); + assert(reply->type == REDIS_REPLY_INTEGER); + long rtr = reply->integer; + freeReplyObject(reply); + return rtr; +} + +long redis_get_sizeof_list(const char *name) +{ + return redis_get_sizeof("LLEN", name); +} + +long redis_get_sizeof_set(const char *name) +{ + return redis_get_sizeof("SCARD", name); +} + +int redis_pull(char *redisqueuename, void *buf, + int maxload, size_t obj_size, int *numloaded, const char* cmd) +{ + assert(rctx); + long elems_in_redis = redis_get_sizeof_list(redisqueuename); + long num_to_add = MIN(elems_in_redis, maxload); + log_info("redis", "INFO: redis load called on %s. Transfering %li " + "of %li elements to in-memory queue.", + redisqueuename, + num_to_add, elems_in_redis); + for(int i=0; i < num_to_add; i++) { + redisAppendCommand(rctx, "%s %s", cmd, redisqueuename); + } + for(int i=0; i < num_to_add; i++) { + redisReply *reply; + int rc = redisGetReply(rctx, (void**) &reply); + if (rc != REDIS_OK) { + log_fatal("redis", "response from redis != REDIS_OK"); + return -1; + } + if (!reply) { + log_fatal("redis", "no reply provided by redis."); + return -1; + } + if (reply->type != REDIS_REPLY_STRING) { + log_fatal("redis", + "unxpected reply type from redis."); + return -1; + } + if ((size_t)reply->len != obj_size) { + log_fatal("redis", "ERROR: unexpected lengthed " + "object provided by redis.\n"); + return -1; + } + memcpy((void*)((intptr_t)buf+i*obj_size), reply->str, obj_size); + freeReplyObject(reply); + } + *numloaded = num_to_add; + return 0; +} + +int redis_lpull(char *redisqueuename, void *buf, + int maxload, size_t obj_size, int *numloaded) +{ + return redis_pull(redisqueuename, buf, + maxload, obj_size, numloaded, "LPOP"); +} + +int redis_spull(char *redisqueuename, void *buf, + int maxload, size_t obj_size, int *numloaded) +{ + return redis_pull(redisqueuename, buf, + maxload, obj_size, numloaded, "SRAND"); +} + +static int redis_push(char *redisqueuename, + void *buf, int num, size_t len, const char *cmd) +{ + assert(rctx); + for (int i=0; i < num; i++) { + void* load = (void*)((intptr_t)buf + i*len); + int rc = redisAppendCommand(rctx, "%s %s %b", + cmd, redisqueuename, load, len); + if (rc != REDIS_OK || rctx->err) { + log_fatal("redis", "%s", rctx->errstr); + return -1; + } + } + redisReply *reply; + for (int i=0; i < num; i++) { + if (redisGetReply(rctx, (void**) &reply) != REDIS_OK + || rctx->err) { + log_fatal("redis","%s", rctx->errstr); + return -1; + } + if (reply->type == REDIS_REPLY_ERROR) { + log_fatal("redis", "%s", rctx->errstr); + return -1; + } + freeReplyObject(reply); + } + return 0; +} + +int redis_lpush(char *redisqueuename, + void *buf, int num, size_t len) +{ + return redis_push(redisqueuename, buf, num, len, "RPUSH"); +} + +int redis_spush(char *redisqueuename, + void *buf, int num, size_t len) +{ + return redis_push(redisqueuename, buf, num, len, "SADD"); +} + diff --git a/lib/redis.h b/lib/redis.h new file mode 100644 index 0000000..51a1a7f --- /dev/null +++ b/lib/redis.h @@ -0,0 +1,40 @@ +#include +#include +#include + +#ifndef _REDIS_ZHELPERS_H +#define _REDIS_ZHELPERS_H + +int redis_init(void); + +int redis_close(void); + +int redis_existconf(const char*); + +int redis_flush(void); + +int redis_delconf(const char*); + +int redis_setconf(const char*, char*); + +int redis_getconf(const char*, char*, size_t); + +long redis_get_sizeof_list(const char*); + +long redis_get_sizeof_set(const char*); + +int redis_lpush(char*, void*, int, size_t); + +int redis_lpull(char*, void*, int, size_t, int*); + +int redis_spull(char*, void*, int, size_t, int*); + +int redis_spush(char*, void*, int, size_t); + +redisContext* redis_get_context(void); + +uint32_t redis_getconf_uint32_t(const char*); + +int redis_setconf_uint32_t(const char*, uint32_t); + +#endif // _REDIS_ZHELPERS_H diff --git a/lib/rijndael-alg-fst.c b/lib/rijndael-alg-fst.c new file mode 100644 index 0000000..2bd35b3 --- /dev/null +++ b/lib/rijndael-alg-fst.c @@ -0,0 +1,1400 @@ +/** + * rijndael-alg-fst.c + * + * @version 3.0 (December 2000) + * + * Optimised ANSI C code for the Rijndael cipher (now AES) + * + * @author Vincent Rijmen + * @author Antoon Bosselaers + * @author Paulo Barreto + * + * This code is hereby placed in the public domain. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHORS ''AS IS'' AND ANY EXPRESS + * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR + * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, + * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE + * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, + * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ +#include +#include + +#include "rijndael-alg-fst.h" + +/* +Te0[x] = S [x].[02, 01, 01, 03]; +Te1[x] = S [x].[03, 02, 01, 01]; +Te2[x] = S [x].[01, 03, 02, 01]; +Te3[x] = S [x].[01, 01, 03, 02]; +Te4[x] = S [x].[01, 01, 01, 01]; + +Td0[x] = Si[x].[0e, 09, 0d, 0b]; +Td1[x] = Si[x].[0b, 0e, 09, 0d]; +Td2[x] = Si[x].[0d, 0b, 0e, 09]; +Td3[x] = Si[x].[09, 0d, 0b, 0e]; +Td4[x] = Si[x].[01, 01, 01, 01]; +*/ + +static const u32 Te0[256] = { + 0xc66363a5U, 0xf87c7c84U, 0xee777799U, 0xf67b7b8dU, + 0xfff2f20dU, 0xd66b6bbdU, 0xde6f6fb1U, 0x91c5c554U, + 0x60303050U, 0x02010103U, 0xce6767a9U, 0x562b2b7dU, + 0xe7fefe19U, 0xb5d7d762U, 0x4dababe6U, 0xec76769aU, + 0x8fcaca45U, 0x1f82829dU, 0x89c9c940U, 0xfa7d7d87U, + 0xeffafa15U, 0xb25959ebU, 0x8e4747c9U, 0xfbf0f00bU, + 0x41adadecU, 0xb3d4d467U, 0x5fa2a2fdU, 0x45afafeaU, + 0x239c9cbfU, 0x53a4a4f7U, 0xe4727296U, 0x9bc0c05bU, + 0x75b7b7c2U, 0xe1fdfd1cU, 0x3d9393aeU, 0x4c26266aU, + 0x6c36365aU, 0x7e3f3f41U, 0xf5f7f702U, 0x83cccc4fU, + 0x6834345cU, 0x51a5a5f4U, 0xd1e5e534U, 0xf9f1f108U, + 0xe2717193U, 0xabd8d873U, 0x62313153U, 0x2a15153fU, + 0x0804040cU, 0x95c7c752U, 0x46232365U, 0x9dc3c35eU, + 0x30181828U, 0x379696a1U, 0x0a05050fU, 0x2f9a9ab5U, + 0x0e070709U, 0x24121236U, 0x1b80809bU, 0xdfe2e23dU, + 0xcdebeb26U, 0x4e272769U, 0x7fb2b2cdU, 0xea75759fU, + 0x1209091bU, 0x1d83839eU, 0x582c2c74U, 0x341a1a2eU, + 0x361b1b2dU, 0xdc6e6eb2U, 0xb45a5aeeU, 0x5ba0a0fbU, + 0xa45252f6U, 0x763b3b4dU, 0xb7d6d661U, 0x7db3b3ceU, + 0x5229297bU, 0xdde3e33eU, 0x5e2f2f71U, 0x13848497U, + 0xa65353f5U, 0xb9d1d168U, 0x00000000U, 0xc1eded2cU, + 0x40202060U, 0xe3fcfc1fU, 0x79b1b1c8U, 0xb65b5bedU, + 0xd46a6abeU, 0x8dcbcb46U, 0x67bebed9U, 0x7239394bU, + 0x944a4adeU, 0x984c4cd4U, 0xb05858e8U, 0x85cfcf4aU, + 0xbbd0d06bU, 0xc5efef2aU, 0x4faaaae5U, 0xedfbfb16U, + 0x864343c5U, 0x9a4d4dd7U, 0x66333355U, 0x11858594U, + 0x8a4545cfU, 0xe9f9f910U, 0x04020206U, 0xfe7f7f81U, + 0xa05050f0U, 0x783c3c44U, 0x259f9fbaU, 0x4ba8a8e3U, + 0xa25151f3U, 0x5da3a3feU, 0x804040c0U, 0x058f8f8aU, + 0x3f9292adU, 0x219d9dbcU, 0x70383848U, 0xf1f5f504U, + 0x63bcbcdfU, 0x77b6b6c1U, 0xafdada75U, 0x42212163U, + 0x20101030U, 0xe5ffff1aU, 0xfdf3f30eU, 0xbfd2d26dU, + 0x81cdcd4cU, 0x180c0c14U, 0x26131335U, 0xc3ecec2fU, + 0xbe5f5fe1U, 0x359797a2U, 0x884444ccU, 0x2e171739U, + 0x93c4c457U, 0x55a7a7f2U, 0xfc7e7e82U, 0x7a3d3d47U, + 0xc86464acU, 0xba5d5de7U, 0x3219192bU, 0xe6737395U, + 0xc06060a0U, 0x19818198U, 0x9e4f4fd1U, 0xa3dcdc7fU, + 0x44222266U, 0x542a2a7eU, 0x3b9090abU, 0x0b888883U, + 0x8c4646caU, 0xc7eeee29U, 0x6bb8b8d3U, 0x2814143cU, + 0xa7dede79U, 0xbc5e5ee2U, 0x160b0b1dU, 0xaddbdb76U, + 0xdbe0e03bU, 0x64323256U, 0x743a3a4eU, 0x140a0a1eU, + 0x924949dbU, 0x0c06060aU, 0x4824246cU, 0xb85c5ce4U, + 0x9fc2c25dU, 0xbdd3d36eU, 0x43acacefU, 0xc46262a6U, + 0x399191a8U, 0x319595a4U, 0xd3e4e437U, 0xf279798bU, + 0xd5e7e732U, 0x8bc8c843U, 0x6e373759U, 0xda6d6db7U, + 0x018d8d8cU, 0xb1d5d564U, 0x9c4e4ed2U, 0x49a9a9e0U, + 0xd86c6cb4U, 0xac5656faU, 0xf3f4f407U, 0xcfeaea25U, + 0xca6565afU, 0xf47a7a8eU, 0x47aeaee9U, 0x10080818U, + 0x6fbabad5U, 0xf0787888U, 0x4a25256fU, 0x5c2e2e72U, + 0x381c1c24U, 0x57a6a6f1U, 0x73b4b4c7U, 0x97c6c651U, + 0xcbe8e823U, 0xa1dddd7cU, 0xe874749cU, 0x3e1f1f21U, + 0x964b4bddU, 0x61bdbddcU, 0x0d8b8b86U, 0x0f8a8a85U, + 0xe0707090U, 0x7c3e3e42U, 0x71b5b5c4U, 0xcc6666aaU, + 0x904848d8U, 0x06030305U, 0xf7f6f601U, 0x1c0e0e12U, + 0xc26161a3U, 0x6a35355fU, 0xae5757f9U, 0x69b9b9d0U, + 0x17868691U, 0x99c1c158U, 0x3a1d1d27U, 0x279e9eb9U, + 0xd9e1e138U, 0xebf8f813U, 0x2b9898b3U, 0x22111133U, + 0xd26969bbU, 0xa9d9d970U, 0x078e8e89U, 0x339494a7U, + 0x2d9b9bb6U, 0x3c1e1e22U, 0x15878792U, 0xc9e9e920U, + 0x87cece49U, 0xaa5555ffU, 0x50282878U, 0xa5dfdf7aU, + 0x038c8c8fU, 0x59a1a1f8U, 0x09898980U, 0x1a0d0d17U, + 0x65bfbfdaU, 0xd7e6e631U, 0x844242c6U, 0xd06868b8U, + 0x824141c3U, 0x299999b0U, 0x5a2d2d77U, 0x1e0f0f11U, + 0x7bb0b0cbU, 0xa85454fcU, 0x6dbbbbd6U, 0x2c16163aU, +}; +static const u32 Te1[256] = { + 0xa5c66363U, 0x84f87c7cU, 0x99ee7777U, 0x8df67b7bU, + 0x0dfff2f2U, 0xbdd66b6bU, 0xb1de6f6fU, 0x5491c5c5U, + 0x50603030U, 0x03020101U, 0xa9ce6767U, 0x7d562b2bU, + 0x19e7fefeU, 0x62b5d7d7U, 0xe64dababU, 0x9aec7676U, + 0x458fcacaU, 0x9d1f8282U, 0x4089c9c9U, 0x87fa7d7dU, + 0x15effafaU, 0xebb25959U, 0xc98e4747U, 0x0bfbf0f0U, + 0xec41adadU, 0x67b3d4d4U, 0xfd5fa2a2U, 0xea45afafU, + 0xbf239c9cU, 0xf753a4a4U, 0x96e47272U, 0x5b9bc0c0U, + 0xc275b7b7U, 0x1ce1fdfdU, 0xae3d9393U, 0x6a4c2626U, + 0x5a6c3636U, 0x417e3f3fU, 0x02f5f7f7U, 0x4f83ccccU, + 0x5c683434U, 0xf451a5a5U, 0x34d1e5e5U, 0x08f9f1f1U, + 0x93e27171U, 0x73abd8d8U, 0x53623131U, 0x3f2a1515U, + 0x0c080404U, 0x5295c7c7U, 0x65462323U, 0x5e9dc3c3U, + 0x28301818U, 0xa1379696U, 0x0f0a0505U, 0xb52f9a9aU, + 0x090e0707U, 0x36241212U, 0x9b1b8080U, 0x3ddfe2e2U, + 0x26cdebebU, 0x694e2727U, 0xcd7fb2b2U, 0x9fea7575U, + 0x1b120909U, 0x9e1d8383U, 0x74582c2cU, 0x2e341a1aU, + 0x2d361b1bU, 0xb2dc6e6eU, 0xeeb45a5aU, 0xfb5ba0a0U, + 0xf6a45252U, 0x4d763b3bU, 0x61b7d6d6U, 0xce7db3b3U, + 0x7b522929U, 0x3edde3e3U, 0x715e2f2fU, 0x97138484U, + 0xf5a65353U, 0x68b9d1d1U, 0x00000000U, 0x2cc1ededU, + 0x60402020U, 0x1fe3fcfcU, 0xc879b1b1U, 0xedb65b5bU, + 0xbed46a6aU, 0x468dcbcbU, 0xd967bebeU, 0x4b723939U, + 0xde944a4aU, 0xd4984c4cU, 0xe8b05858U, 0x4a85cfcfU, + 0x6bbbd0d0U, 0x2ac5efefU, 0xe54faaaaU, 0x16edfbfbU, + 0xc5864343U, 0xd79a4d4dU, 0x55663333U, 0x94118585U, + 0xcf8a4545U, 0x10e9f9f9U, 0x06040202U, 0x81fe7f7fU, + 0xf0a05050U, 0x44783c3cU, 0xba259f9fU, 0xe34ba8a8U, + 0xf3a25151U, 0xfe5da3a3U, 0xc0804040U, 0x8a058f8fU, + 0xad3f9292U, 0xbc219d9dU, 0x48703838U, 0x04f1f5f5U, + 0xdf63bcbcU, 0xc177b6b6U, 0x75afdadaU, 0x63422121U, + 0x30201010U, 0x1ae5ffffU, 0x0efdf3f3U, 0x6dbfd2d2U, + 0x4c81cdcdU, 0x14180c0cU, 0x35261313U, 0x2fc3ececU, + 0xe1be5f5fU, 0xa2359797U, 0xcc884444U, 0x392e1717U, + 0x5793c4c4U, 0xf255a7a7U, 0x82fc7e7eU, 0x477a3d3dU, + 0xacc86464U, 0xe7ba5d5dU, 0x2b321919U, 0x95e67373U, + 0xa0c06060U, 0x98198181U, 0xd19e4f4fU, 0x7fa3dcdcU, + 0x66442222U, 0x7e542a2aU, 0xab3b9090U, 0x830b8888U, + 0xca8c4646U, 0x29c7eeeeU, 0xd36bb8b8U, 0x3c281414U, + 0x79a7dedeU, 0xe2bc5e5eU, 0x1d160b0bU, 0x76addbdbU, + 0x3bdbe0e0U, 0x56643232U, 0x4e743a3aU, 0x1e140a0aU, + 0xdb924949U, 0x0a0c0606U, 0x6c482424U, 0xe4b85c5cU, + 0x5d9fc2c2U, 0x6ebdd3d3U, 0xef43acacU, 0xa6c46262U, + 0xa8399191U, 0xa4319595U, 0x37d3e4e4U, 0x8bf27979U, + 0x32d5e7e7U, 0x438bc8c8U, 0x596e3737U, 0xb7da6d6dU, + 0x8c018d8dU, 0x64b1d5d5U, 0xd29c4e4eU, 0xe049a9a9U, + 0xb4d86c6cU, 0xfaac5656U, 0x07f3f4f4U, 0x25cfeaeaU, + 0xafca6565U, 0x8ef47a7aU, 0xe947aeaeU, 0x18100808U, + 0xd56fbabaU, 0x88f07878U, 0x6f4a2525U, 0x725c2e2eU, + 0x24381c1cU, 0xf157a6a6U, 0xc773b4b4U, 0x5197c6c6U, + 0x23cbe8e8U, 0x7ca1ddddU, 0x9ce87474U, 0x213e1f1fU, + 0xdd964b4bU, 0xdc61bdbdU, 0x860d8b8bU, 0x850f8a8aU, + 0x90e07070U, 0x427c3e3eU, 0xc471b5b5U, 0xaacc6666U, + 0xd8904848U, 0x05060303U, 0x01f7f6f6U, 0x121c0e0eU, + 0xa3c26161U, 0x5f6a3535U, 0xf9ae5757U, 0xd069b9b9U, + 0x91178686U, 0x5899c1c1U, 0x273a1d1dU, 0xb9279e9eU, + 0x38d9e1e1U, 0x13ebf8f8U, 0xb32b9898U, 0x33221111U, + 0xbbd26969U, 0x70a9d9d9U, 0x89078e8eU, 0xa7339494U, + 0xb62d9b9bU, 0x223c1e1eU, 0x92158787U, 0x20c9e9e9U, + 0x4987ceceU, 0xffaa5555U, 0x78502828U, 0x7aa5dfdfU, + 0x8f038c8cU, 0xf859a1a1U, 0x80098989U, 0x171a0d0dU, + 0xda65bfbfU, 0x31d7e6e6U, 0xc6844242U, 0xb8d06868U, + 0xc3824141U, 0xb0299999U, 0x775a2d2dU, 0x111e0f0fU, + 0xcb7bb0b0U, 0xfca85454U, 0xd66dbbbbU, 0x3a2c1616U, +}; +static const u32 Te2[256] = { + 0x63a5c663U, 0x7c84f87cU, 0x7799ee77U, 0x7b8df67bU, + 0xf20dfff2U, 0x6bbdd66bU, 0x6fb1de6fU, 0xc55491c5U, + 0x30506030U, 0x01030201U, 0x67a9ce67U, 0x2b7d562bU, + 0xfe19e7feU, 0xd762b5d7U, 0xabe64dabU, 0x769aec76U, + 0xca458fcaU, 0x829d1f82U, 0xc94089c9U, 0x7d87fa7dU, + 0xfa15effaU, 0x59ebb259U, 0x47c98e47U, 0xf00bfbf0U, + 0xadec41adU, 0xd467b3d4U, 0xa2fd5fa2U, 0xafea45afU, + 0x9cbf239cU, 0xa4f753a4U, 0x7296e472U, 0xc05b9bc0U, + 0xb7c275b7U, 0xfd1ce1fdU, 0x93ae3d93U, 0x266a4c26U, + 0x365a6c36U, 0x3f417e3fU, 0xf702f5f7U, 0xcc4f83ccU, + 0x345c6834U, 0xa5f451a5U, 0xe534d1e5U, 0xf108f9f1U, + 0x7193e271U, 0xd873abd8U, 0x31536231U, 0x153f2a15U, + 0x040c0804U, 0xc75295c7U, 0x23654623U, 0xc35e9dc3U, + 0x18283018U, 0x96a13796U, 0x050f0a05U, 0x9ab52f9aU, + 0x07090e07U, 0x12362412U, 0x809b1b80U, 0xe23ddfe2U, + 0xeb26cdebU, 0x27694e27U, 0xb2cd7fb2U, 0x759fea75U, + 0x091b1209U, 0x839e1d83U, 0x2c74582cU, 0x1a2e341aU, + 0x1b2d361bU, 0x6eb2dc6eU, 0x5aeeb45aU, 0xa0fb5ba0U, + 0x52f6a452U, 0x3b4d763bU, 0xd661b7d6U, 0xb3ce7db3U, + 0x297b5229U, 0xe33edde3U, 0x2f715e2fU, 0x84971384U, + 0x53f5a653U, 0xd168b9d1U, 0x00000000U, 0xed2cc1edU, + 0x20604020U, 0xfc1fe3fcU, 0xb1c879b1U, 0x5bedb65bU, + 0x6abed46aU, 0xcb468dcbU, 0xbed967beU, 0x394b7239U, + 0x4ade944aU, 0x4cd4984cU, 0x58e8b058U, 0xcf4a85cfU, + 0xd06bbbd0U, 0xef2ac5efU, 0xaae54faaU, 0xfb16edfbU, + 0x43c58643U, 0x4dd79a4dU, 0x33556633U, 0x85941185U, + 0x45cf8a45U, 0xf910e9f9U, 0x02060402U, 0x7f81fe7fU, + 0x50f0a050U, 0x3c44783cU, 0x9fba259fU, 0xa8e34ba8U, + 0x51f3a251U, 0xa3fe5da3U, 0x40c08040U, 0x8f8a058fU, + 0x92ad3f92U, 0x9dbc219dU, 0x38487038U, 0xf504f1f5U, + 0xbcdf63bcU, 0xb6c177b6U, 0xda75afdaU, 0x21634221U, + 0x10302010U, 0xff1ae5ffU, 0xf30efdf3U, 0xd26dbfd2U, + 0xcd4c81cdU, 0x0c14180cU, 0x13352613U, 0xec2fc3ecU, + 0x5fe1be5fU, 0x97a23597U, 0x44cc8844U, 0x17392e17U, + 0xc45793c4U, 0xa7f255a7U, 0x7e82fc7eU, 0x3d477a3dU, + 0x64acc864U, 0x5de7ba5dU, 0x192b3219U, 0x7395e673U, + 0x60a0c060U, 0x81981981U, 0x4fd19e4fU, 0xdc7fa3dcU, + 0x22664422U, 0x2a7e542aU, 0x90ab3b90U, 0x88830b88U, + 0x46ca8c46U, 0xee29c7eeU, 0xb8d36bb8U, 0x143c2814U, + 0xde79a7deU, 0x5ee2bc5eU, 0x0b1d160bU, 0xdb76addbU, + 0xe03bdbe0U, 0x32566432U, 0x3a4e743aU, 0x0a1e140aU, + 0x49db9249U, 0x060a0c06U, 0x246c4824U, 0x5ce4b85cU, + 0xc25d9fc2U, 0xd36ebdd3U, 0xacef43acU, 0x62a6c462U, + 0x91a83991U, 0x95a43195U, 0xe437d3e4U, 0x798bf279U, + 0xe732d5e7U, 0xc8438bc8U, 0x37596e37U, 0x6db7da6dU, + 0x8d8c018dU, 0xd564b1d5U, 0x4ed29c4eU, 0xa9e049a9U, + 0x6cb4d86cU, 0x56faac56U, 0xf407f3f4U, 0xea25cfeaU, + 0x65afca65U, 0x7a8ef47aU, 0xaee947aeU, 0x08181008U, + 0xbad56fbaU, 0x7888f078U, 0x256f4a25U, 0x2e725c2eU, + 0x1c24381cU, 0xa6f157a6U, 0xb4c773b4U, 0xc65197c6U, + 0xe823cbe8U, 0xdd7ca1ddU, 0x749ce874U, 0x1f213e1fU, + 0x4bdd964bU, 0xbddc61bdU, 0x8b860d8bU, 0x8a850f8aU, + 0x7090e070U, 0x3e427c3eU, 0xb5c471b5U, 0x66aacc66U, + 0x48d89048U, 0x03050603U, 0xf601f7f6U, 0x0e121c0eU, + 0x61a3c261U, 0x355f6a35U, 0x57f9ae57U, 0xb9d069b9U, + 0x86911786U, 0xc15899c1U, 0x1d273a1dU, 0x9eb9279eU, + 0xe138d9e1U, 0xf813ebf8U, 0x98b32b98U, 0x11332211U, + 0x69bbd269U, 0xd970a9d9U, 0x8e89078eU, 0x94a73394U, + 0x9bb62d9bU, 0x1e223c1eU, 0x87921587U, 0xe920c9e9U, + 0xce4987ceU, 0x55ffaa55U, 0x28785028U, 0xdf7aa5dfU, + 0x8c8f038cU, 0xa1f859a1U, 0x89800989U, 0x0d171a0dU, + 0xbfda65bfU, 0xe631d7e6U, 0x42c68442U, 0x68b8d068U, + 0x41c38241U, 0x99b02999U, 0x2d775a2dU, 0x0f111e0fU, + 0xb0cb7bb0U, 0x54fca854U, 0xbbd66dbbU, 0x163a2c16U, +}; +static const u32 Te3[256] = { + + 0x6363a5c6U, 0x7c7c84f8U, 0x777799eeU, 0x7b7b8df6U, + 0xf2f20dffU, 0x6b6bbdd6U, 0x6f6fb1deU, 0xc5c55491U, + 0x30305060U, 0x01010302U, 0x6767a9ceU, 0x2b2b7d56U, + 0xfefe19e7U, 0xd7d762b5U, 0xababe64dU, 0x76769aecU, + 0xcaca458fU, 0x82829d1fU, 0xc9c94089U, 0x7d7d87faU, + 0xfafa15efU, 0x5959ebb2U, 0x4747c98eU, 0xf0f00bfbU, + 0xadadec41U, 0xd4d467b3U, 0xa2a2fd5fU, 0xafafea45U, + 0x9c9cbf23U, 0xa4a4f753U, 0x727296e4U, 0xc0c05b9bU, + 0xb7b7c275U, 0xfdfd1ce1U, 0x9393ae3dU, 0x26266a4cU, + 0x36365a6cU, 0x3f3f417eU, 0xf7f702f5U, 0xcccc4f83U, + 0x34345c68U, 0xa5a5f451U, 0xe5e534d1U, 0xf1f108f9U, + 0x717193e2U, 0xd8d873abU, 0x31315362U, 0x15153f2aU, + 0x04040c08U, 0xc7c75295U, 0x23236546U, 0xc3c35e9dU, + 0x18182830U, 0x9696a137U, 0x05050f0aU, 0x9a9ab52fU, + 0x0707090eU, 0x12123624U, 0x80809b1bU, 0xe2e23ddfU, + 0xebeb26cdU, 0x2727694eU, 0xb2b2cd7fU, 0x75759feaU, + 0x09091b12U, 0x83839e1dU, 0x2c2c7458U, 0x1a1a2e34U, + 0x1b1b2d36U, 0x6e6eb2dcU, 0x5a5aeeb4U, 0xa0a0fb5bU, + 0x5252f6a4U, 0x3b3b4d76U, 0xd6d661b7U, 0xb3b3ce7dU, + 0x29297b52U, 0xe3e33eddU, 0x2f2f715eU, 0x84849713U, + 0x5353f5a6U, 0xd1d168b9U, 0x00000000U, 0xeded2cc1U, + 0x20206040U, 0xfcfc1fe3U, 0xb1b1c879U, 0x5b5bedb6U, + 0x6a6abed4U, 0xcbcb468dU, 0xbebed967U, 0x39394b72U, + 0x4a4ade94U, 0x4c4cd498U, 0x5858e8b0U, 0xcfcf4a85U, + 0xd0d06bbbU, 0xefef2ac5U, 0xaaaae54fU, 0xfbfb16edU, + 0x4343c586U, 0x4d4dd79aU, 0x33335566U, 0x85859411U, + 0x4545cf8aU, 0xf9f910e9U, 0x02020604U, 0x7f7f81feU, + 0x5050f0a0U, 0x3c3c4478U, 0x9f9fba25U, 0xa8a8e34bU, + 0x5151f3a2U, 0xa3a3fe5dU, 0x4040c080U, 0x8f8f8a05U, + 0x9292ad3fU, 0x9d9dbc21U, 0x38384870U, 0xf5f504f1U, + 0xbcbcdf63U, 0xb6b6c177U, 0xdada75afU, 0x21216342U, + 0x10103020U, 0xffff1ae5U, 0xf3f30efdU, 0xd2d26dbfU, + 0xcdcd4c81U, 0x0c0c1418U, 0x13133526U, 0xecec2fc3U, + 0x5f5fe1beU, 0x9797a235U, 0x4444cc88U, 0x1717392eU, + 0xc4c45793U, 0xa7a7f255U, 0x7e7e82fcU, 0x3d3d477aU, + 0x6464acc8U, 0x5d5de7baU, 0x19192b32U, 0x737395e6U, + 0x6060a0c0U, 0x81819819U, 0x4f4fd19eU, 0xdcdc7fa3U, + 0x22226644U, 0x2a2a7e54U, 0x9090ab3bU, 0x8888830bU, + 0x4646ca8cU, 0xeeee29c7U, 0xb8b8d36bU, 0x14143c28U, + 0xdede79a7U, 0x5e5ee2bcU, 0x0b0b1d16U, 0xdbdb76adU, + 0xe0e03bdbU, 0x32325664U, 0x3a3a4e74U, 0x0a0a1e14U, + 0x4949db92U, 0x06060a0cU, 0x24246c48U, 0x5c5ce4b8U, + 0xc2c25d9fU, 0xd3d36ebdU, 0xacacef43U, 0x6262a6c4U, + 0x9191a839U, 0x9595a431U, 0xe4e437d3U, 0x79798bf2U, + 0xe7e732d5U, 0xc8c8438bU, 0x3737596eU, 0x6d6db7daU, + 0x8d8d8c01U, 0xd5d564b1U, 0x4e4ed29cU, 0xa9a9e049U, + 0x6c6cb4d8U, 0x5656faacU, 0xf4f407f3U, 0xeaea25cfU, + 0x6565afcaU, 0x7a7a8ef4U, 0xaeaee947U, 0x08081810U, + 0xbabad56fU, 0x787888f0U, 0x25256f4aU, 0x2e2e725cU, + 0x1c1c2438U, 0xa6a6f157U, 0xb4b4c773U, 0xc6c65197U, + 0xe8e823cbU, 0xdddd7ca1U, 0x74749ce8U, 0x1f1f213eU, + 0x4b4bdd96U, 0xbdbddc61U, 0x8b8b860dU, 0x8a8a850fU, + 0x707090e0U, 0x3e3e427cU, 0xb5b5c471U, 0x6666aaccU, + 0x4848d890U, 0x03030506U, 0xf6f601f7U, 0x0e0e121cU, + 0x6161a3c2U, 0x35355f6aU, 0x5757f9aeU, 0xb9b9d069U, + 0x86869117U, 0xc1c15899U, 0x1d1d273aU, 0x9e9eb927U, + 0xe1e138d9U, 0xf8f813ebU, 0x9898b32bU, 0x11113322U, + 0x6969bbd2U, 0xd9d970a9U, 0x8e8e8907U, 0x9494a733U, + 0x9b9bb62dU, 0x1e1e223cU, 0x87879215U, 0xe9e920c9U, + 0xcece4987U, 0x5555ffaaU, 0x28287850U, 0xdfdf7aa5U, + 0x8c8c8f03U, 0xa1a1f859U, 0x89898009U, 0x0d0d171aU, + 0xbfbfda65U, 0xe6e631d7U, 0x4242c684U, 0x6868b8d0U, + 0x4141c382U, 0x9999b029U, 0x2d2d775aU, 0x0f0f111eU, + 0xb0b0cb7bU, 0x5454fca8U, 0xbbbbd66dU, 0x16163a2cU, +}; +static const u32 Te4[256] = { + 0x63636363U, 0x7c7c7c7cU, 0x77777777U, 0x7b7b7b7bU, + 0xf2f2f2f2U, 0x6b6b6b6bU, 0x6f6f6f6fU, 0xc5c5c5c5U, + 0x30303030U, 0x01010101U, 0x67676767U, 0x2b2b2b2bU, + 0xfefefefeU, 0xd7d7d7d7U, 0xababababU, 0x76767676U, + 0xcacacacaU, 0x82828282U, 0xc9c9c9c9U, 0x7d7d7d7dU, + 0xfafafafaU, 0x59595959U, 0x47474747U, 0xf0f0f0f0U, + 0xadadadadU, 0xd4d4d4d4U, 0xa2a2a2a2U, 0xafafafafU, + 0x9c9c9c9cU, 0xa4a4a4a4U, 0x72727272U, 0xc0c0c0c0U, + 0xb7b7b7b7U, 0xfdfdfdfdU, 0x93939393U, 0x26262626U, + 0x36363636U, 0x3f3f3f3fU, 0xf7f7f7f7U, 0xccccccccU, + 0x34343434U, 0xa5a5a5a5U, 0xe5e5e5e5U, 0xf1f1f1f1U, + 0x71717171U, 0xd8d8d8d8U, 0x31313131U, 0x15151515U, + 0x04040404U, 0xc7c7c7c7U, 0x23232323U, 0xc3c3c3c3U, + 0x18181818U, 0x96969696U, 0x05050505U, 0x9a9a9a9aU, + 0x07070707U, 0x12121212U, 0x80808080U, 0xe2e2e2e2U, + 0xebebebebU, 0x27272727U, 0xb2b2b2b2U, 0x75757575U, + 0x09090909U, 0x83838383U, 0x2c2c2c2cU, 0x1a1a1a1aU, + 0x1b1b1b1bU, 0x6e6e6e6eU, 0x5a5a5a5aU, 0xa0a0a0a0U, + 0x52525252U, 0x3b3b3b3bU, 0xd6d6d6d6U, 0xb3b3b3b3U, + 0x29292929U, 0xe3e3e3e3U, 0x2f2f2f2fU, 0x84848484U, + 0x53535353U, 0xd1d1d1d1U, 0x00000000U, 0xededededU, + 0x20202020U, 0xfcfcfcfcU, 0xb1b1b1b1U, 0x5b5b5b5bU, + 0x6a6a6a6aU, 0xcbcbcbcbU, 0xbebebebeU, 0x39393939U, + 0x4a4a4a4aU, 0x4c4c4c4cU, 0x58585858U, 0xcfcfcfcfU, + 0xd0d0d0d0U, 0xefefefefU, 0xaaaaaaaaU, 0xfbfbfbfbU, + 0x43434343U, 0x4d4d4d4dU, 0x33333333U, 0x85858585U, + 0x45454545U, 0xf9f9f9f9U, 0x02020202U, 0x7f7f7f7fU, + 0x50505050U, 0x3c3c3c3cU, 0x9f9f9f9fU, 0xa8a8a8a8U, + 0x51515151U, 0xa3a3a3a3U, 0x40404040U, 0x8f8f8f8fU, + 0x92929292U, 0x9d9d9d9dU, 0x38383838U, 0xf5f5f5f5U, + 0xbcbcbcbcU, 0xb6b6b6b6U, 0xdadadadaU, 0x21212121U, + 0x10101010U, 0xffffffffU, 0xf3f3f3f3U, 0xd2d2d2d2U, + 0xcdcdcdcdU, 0x0c0c0c0cU, 0x13131313U, 0xececececU, + 0x5f5f5f5fU, 0x97979797U, 0x44444444U, 0x17171717U, + 0xc4c4c4c4U, 0xa7a7a7a7U, 0x7e7e7e7eU, 0x3d3d3d3dU, + 0x64646464U, 0x5d5d5d5dU, 0x19191919U, 0x73737373U, + 0x60606060U, 0x81818181U, 0x4f4f4f4fU, 0xdcdcdcdcU, + 0x22222222U, 0x2a2a2a2aU, 0x90909090U, 0x88888888U, + 0x46464646U, 0xeeeeeeeeU, 0xb8b8b8b8U, 0x14141414U, + 0xdedededeU, 0x5e5e5e5eU, 0x0b0b0b0bU, 0xdbdbdbdbU, + 0xe0e0e0e0U, 0x32323232U, 0x3a3a3a3aU, 0x0a0a0a0aU, + 0x49494949U, 0x06060606U, 0x24242424U, 0x5c5c5c5cU, + 0xc2c2c2c2U, 0xd3d3d3d3U, 0xacacacacU, 0x62626262U, + 0x91919191U, 0x95959595U, 0xe4e4e4e4U, 0x79797979U, + 0xe7e7e7e7U, 0xc8c8c8c8U, 0x37373737U, 0x6d6d6d6dU, + 0x8d8d8d8dU, 0xd5d5d5d5U, 0x4e4e4e4eU, 0xa9a9a9a9U, + 0x6c6c6c6cU, 0x56565656U, 0xf4f4f4f4U, 0xeaeaeaeaU, + 0x65656565U, 0x7a7a7a7aU, 0xaeaeaeaeU, 0x08080808U, + 0xbabababaU, 0x78787878U, 0x25252525U, 0x2e2e2e2eU, + 0x1c1c1c1cU, 0xa6a6a6a6U, 0xb4b4b4b4U, 0xc6c6c6c6U, + 0xe8e8e8e8U, 0xddddddddU, 0x74747474U, 0x1f1f1f1fU, + 0x4b4b4b4bU, 0xbdbdbdbdU, 0x8b8b8b8bU, 0x8a8a8a8aU, + 0x70707070U, 0x3e3e3e3eU, 0xb5b5b5b5U, 0x66666666U, + 0x48484848U, 0x03030303U, 0xf6f6f6f6U, 0x0e0e0e0eU, + 0x61616161U, 0x35353535U, 0x57575757U, 0xb9b9b9b9U, + 0x86868686U, 0xc1c1c1c1U, 0x1d1d1d1dU, 0x9e9e9e9eU, + 0xe1e1e1e1U, 0xf8f8f8f8U, 0x98989898U, 0x11111111U, + 0x69696969U, 0xd9d9d9d9U, 0x8e8e8e8eU, 0x94949494U, + 0x9b9b9b9bU, 0x1e1e1e1eU, 0x87878787U, 0xe9e9e9e9U, + 0xcecececeU, 0x55555555U, 0x28282828U, 0xdfdfdfdfU, + 0x8c8c8c8cU, 0xa1a1a1a1U, 0x89898989U, 0x0d0d0d0dU, + 0xbfbfbfbfU, 0xe6e6e6e6U, 0x42424242U, 0x68686868U, + 0x41414141U, 0x99999999U, 0x2d2d2d2dU, 0x0f0f0f0fU, + 0xb0b0b0b0U, 0x54545454U, 0xbbbbbbbbU, 0x16161616U, +}; +static const u32 Td0[256] = { + 0x51f4a750U, 0x7e416553U, 0x1a17a4c3U, 0x3a275e96U, + 0x3bab6bcbU, 0x1f9d45f1U, 0xacfa58abU, 0x4be30393U, + 0x2030fa55U, 0xad766df6U, 0x88cc7691U, 0xf5024c25U, + 0x4fe5d7fcU, 0xc52acbd7U, 0x26354480U, 0xb562a38fU, + 0xdeb15a49U, 0x25ba1b67U, 0x45ea0e98U, 0x5dfec0e1U, + 0xc32f7502U, 0x814cf012U, 0x8d4697a3U, 0x6bd3f9c6U, + 0x038f5fe7U, 0x15929c95U, 0xbf6d7aebU, 0x955259daU, + 0xd4be832dU, 0x587421d3U, 0x49e06929U, 0x8ec9c844U, + 0x75c2896aU, 0xf48e7978U, 0x99583e6bU, 0x27b971ddU, + 0xbee14fb6U, 0xf088ad17U, 0xc920ac66U, 0x7dce3ab4U, + 0x63df4a18U, 0xe51a3182U, 0x97513360U, 0x62537f45U, + 0xb16477e0U, 0xbb6bae84U, 0xfe81a01cU, 0xf9082b94U, + 0x70486858U, 0x8f45fd19U, 0x94de6c87U, 0x527bf8b7U, + 0xab73d323U, 0x724b02e2U, 0xe31f8f57U, 0x6655ab2aU, + 0xb2eb2807U, 0x2fb5c203U, 0x86c57b9aU, 0xd33708a5U, + 0x302887f2U, 0x23bfa5b2U, 0x02036abaU, 0xed16825cU, + 0x8acf1c2bU, 0xa779b492U, 0xf307f2f0U, 0x4e69e2a1U, + 0x65daf4cdU, 0x0605bed5U, 0xd134621fU, 0xc4a6fe8aU, + 0x342e539dU, 0xa2f355a0U, 0x058ae132U, 0xa4f6eb75U, + 0x0b83ec39U, 0x4060efaaU, 0x5e719f06U, 0xbd6e1051U, + 0x3e218af9U, 0x96dd063dU, 0xdd3e05aeU, 0x4de6bd46U, + 0x91548db5U, 0x71c45d05U, 0x0406d46fU, 0x605015ffU, + 0x1998fb24U, 0xd6bde997U, 0x894043ccU, 0x67d99e77U, + 0xb0e842bdU, 0x07898b88U, 0xe7195b38U, 0x79c8eedbU, + 0xa17c0a47U, 0x7c420fe9U, 0xf8841ec9U, 0x00000000U, + 0x09808683U, 0x322bed48U, 0x1e1170acU, 0x6c5a724eU, + 0xfd0efffbU, 0x0f853856U, 0x3daed51eU, 0x362d3927U, + 0x0a0fd964U, 0x685ca621U, 0x9b5b54d1U, 0x24362e3aU, + 0x0c0a67b1U, 0x9357e70fU, 0xb4ee96d2U, 0x1b9b919eU, + 0x80c0c54fU, 0x61dc20a2U, 0x5a774b69U, 0x1c121a16U, + 0xe293ba0aU, 0xc0a02ae5U, 0x3c22e043U, 0x121b171dU, + 0x0e090d0bU, 0xf28bc7adU, 0x2db6a8b9U, 0x141ea9c8U, + 0x57f11985U, 0xaf75074cU, 0xee99ddbbU, 0xa37f60fdU, + 0xf701269fU, 0x5c72f5bcU, 0x44663bc5U, 0x5bfb7e34U, + 0x8b432976U, 0xcb23c6dcU, 0xb6edfc68U, 0xb8e4f163U, + 0xd731dccaU, 0x42638510U, 0x13972240U, 0x84c61120U, + 0x854a247dU, 0xd2bb3df8U, 0xaef93211U, 0xc729a16dU, + 0x1d9e2f4bU, 0xdcb230f3U, 0x0d8652ecU, 0x77c1e3d0U, + 0x2bb3166cU, 0xa970b999U, 0x119448faU, 0x47e96422U, + 0xa8fc8cc4U, 0xa0f03f1aU, 0x567d2cd8U, 0x223390efU, + 0x87494ec7U, 0xd938d1c1U, 0x8ccaa2feU, 0x98d40b36U, + 0xa6f581cfU, 0xa57ade28U, 0xdab78e26U, 0x3fadbfa4U, + 0x2c3a9de4U, 0x5078920dU, 0x6a5fcc9bU, 0x547e4662U, + 0xf68d13c2U, 0x90d8b8e8U, 0x2e39f75eU, 0x82c3aff5U, + 0x9f5d80beU, 0x69d0937cU, 0x6fd52da9U, 0xcf2512b3U, + 0xc8ac993bU, 0x10187da7U, 0xe89c636eU, 0xdb3bbb7bU, + 0xcd267809U, 0x6e5918f4U, 0xec9ab701U, 0x834f9aa8U, + 0xe6956e65U, 0xaaffe67eU, 0x21bccf08U, 0xef15e8e6U, + 0xbae79bd9U, 0x4a6f36ceU, 0xea9f09d4U, 0x29b07cd6U, + 0x31a4b2afU, 0x2a3f2331U, 0xc6a59430U, 0x35a266c0U, + 0x744ebc37U, 0xfc82caa6U, 0xe090d0b0U, 0x33a7d815U, + 0xf104984aU, 0x41ecdaf7U, 0x7fcd500eU, 0x1791f62fU, + 0x764dd68dU, 0x43efb04dU, 0xccaa4d54U, 0xe49604dfU, + 0x9ed1b5e3U, 0x4c6a881bU, 0xc12c1fb8U, 0x4665517fU, + 0x9d5eea04U, 0x018c355dU, 0xfa877473U, 0xfb0b412eU, + 0xb3671d5aU, 0x92dbd252U, 0xe9105633U, 0x6dd64713U, + 0x9ad7618cU, 0x37a10c7aU, 0x59f8148eU, 0xeb133c89U, + 0xcea927eeU, 0xb761c935U, 0xe11ce5edU, 0x7a47b13cU, + 0x9cd2df59U, 0x55f2733fU, 0x1814ce79U, 0x73c737bfU, + 0x53f7cdeaU, 0x5ffdaa5bU, 0xdf3d6f14U, 0x7844db86U, + 0xcaaff381U, 0xb968c43eU, 0x3824342cU, 0xc2a3405fU, + 0x161dc372U, 0xbce2250cU, 0x283c498bU, 0xff0d9541U, + 0x39a80171U, 0x080cb3deU, 0xd8b4e49cU, 0x6456c190U, + 0x7bcb8461U, 0xd532b670U, 0x486c5c74U, 0xd0b85742U, +}; +static const u32 Td1[256] = { + 0x5051f4a7U, 0x537e4165U, 0xc31a17a4U, 0x963a275eU, + 0xcb3bab6bU, 0xf11f9d45U, 0xabacfa58U, 0x934be303U, + 0x552030faU, 0xf6ad766dU, 0x9188cc76U, 0x25f5024cU, + 0xfc4fe5d7U, 0xd7c52acbU, 0x80263544U, 0x8fb562a3U, + 0x49deb15aU, 0x6725ba1bU, 0x9845ea0eU, 0xe15dfec0U, + 0x02c32f75U, 0x12814cf0U, 0xa38d4697U, 0xc66bd3f9U, + 0xe7038f5fU, 0x9515929cU, 0xebbf6d7aU, 0xda955259U, + 0x2dd4be83U, 0xd3587421U, 0x2949e069U, 0x448ec9c8U, + 0x6a75c289U, 0x78f48e79U, 0x6b99583eU, 0xdd27b971U, + 0xb6bee14fU, 0x17f088adU, 0x66c920acU, 0xb47dce3aU, + 0x1863df4aU, 0x82e51a31U, 0x60975133U, 0x4562537fU, + 0xe0b16477U, 0x84bb6baeU, 0x1cfe81a0U, 0x94f9082bU, + 0x58704868U, 0x198f45fdU, 0x8794de6cU, 0xb7527bf8U, + 0x23ab73d3U, 0xe2724b02U, 0x57e31f8fU, 0x2a6655abU, + 0x07b2eb28U, 0x032fb5c2U, 0x9a86c57bU, 0xa5d33708U, + 0xf2302887U, 0xb223bfa5U, 0xba02036aU, 0x5ced1682U, + 0x2b8acf1cU, 0x92a779b4U, 0xf0f307f2U, 0xa14e69e2U, + 0xcd65daf4U, 0xd50605beU, 0x1fd13462U, 0x8ac4a6feU, + 0x9d342e53U, 0xa0a2f355U, 0x32058ae1U, 0x75a4f6ebU, + 0x390b83ecU, 0xaa4060efU, 0x065e719fU, 0x51bd6e10U, + 0xf93e218aU, 0x3d96dd06U, 0xaedd3e05U, 0x464de6bdU, + 0xb591548dU, 0x0571c45dU, 0x6f0406d4U, 0xff605015U, + 0x241998fbU, 0x97d6bde9U, 0xcc894043U, 0x7767d99eU, + 0xbdb0e842U, 0x8807898bU, 0x38e7195bU, 0xdb79c8eeU, + 0x47a17c0aU, 0xe97c420fU, 0xc9f8841eU, 0x00000000U, + 0x83098086U, 0x48322bedU, 0xac1e1170U, 0x4e6c5a72U, + 0xfbfd0effU, 0x560f8538U, 0x1e3daed5U, 0x27362d39U, + 0x640a0fd9U, 0x21685ca6U, 0xd19b5b54U, 0x3a24362eU, + 0xb10c0a67U, 0x0f9357e7U, 0xd2b4ee96U, 0x9e1b9b91U, + 0x4f80c0c5U, 0xa261dc20U, 0x695a774bU, 0x161c121aU, + 0x0ae293baU, 0xe5c0a02aU, 0x433c22e0U, 0x1d121b17U, + 0x0b0e090dU, 0xadf28bc7U, 0xb92db6a8U, 0xc8141ea9U, + 0x8557f119U, 0x4caf7507U, 0xbbee99ddU, 0xfda37f60U, + 0x9ff70126U, 0xbc5c72f5U, 0xc544663bU, 0x345bfb7eU, + 0x768b4329U, 0xdccb23c6U, 0x68b6edfcU, 0x63b8e4f1U, + 0xcad731dcU, 0x10426385U, 0x40139722U, 0x2084c611U, + 0x7d854a24U, 0xf8d2bb3dU, 0x11aef932U, 0x6dc729a1U, + 0x4b1d9e2fU, 0xf3dcb230U, 0xec0d8652U, 0xd077c1e3U, + 0x6c2bb316U, 0x99a970b9U, 0xfa119448U, 0x2247e964U, + 0xc4a8fc8cU, 0x1aa0f03fU, 0xd8567d2cU, 0xef223390U, + 0xc787494eU, 0xc1d938d1U, 0xfe8ccaa2U, 0x3698d40bU, + 0xcfa6f581U, 0x28a57adeU, 0x26dab78eU, 0xa43fadbfU, + 0xe42c3a9dU, 0x0d507892U, 0x9b6a5fccU, 0x62547e46U, + 0xc2f68d13U, 0xe890d8b8U, 0x5e2e39f7U, 0xf582c3afU, + 0xbe9f5d80U, 0x7c69d093U, 0xa96fd52dU, 0xb3cf2512U, + 0x3bc8ac99U, 0xa710187dU, 0x6ee89c63U, 0x7bdb3bbbU, + 0x09cd2678U, 0xf46e5918U, 0x01ec9ab7U, 0xa8834f9aU, + 0x65e6956eU, 0x7eaaffe6U, 0x0821bccfU, 0xe6ef15e8U, + 0xd9bae79bU, 0xce4a6f36U, 0xd4ea9f09U, 0xd629b07cU, + 0xaf31a4b2U, 0x312a3f23U, 0x30c6a594U, 0xc035a266U, + 0x37744ebcU, 0xa6fc82caU, 0xb0e090d0U, 0x1533a7d8U, + 0x4af10498U, 0xf741ecdaU, 0x0e7fcd50U, 0x2f1791f6U, + 0x8d764dd6U, 0x4d43efb0U, 0x54ccaa4dU, 0xdfe49604U, + 0xe39ed1b5U, 0x1b4c6a88U, 0xb8c12c1fU, 0x7f466551U, + 0x049d5eeaU, 0x5d018c35U, 0x73fa8774U, 0x2efb0b41U, + 0x5ab3671dU, 0x5292dbd2U, 0x33e91056U, 0x136dd647U, + 0x8c9ad761U, 0x7a37a10cU, 0x8e59f814U, 0x89eb133cU, + 0xeecea927U, 0x35b761c9U, 0xede11ce5U, 0x3c7a47b1U, + 0x599cd2dfU, 0x3f55f273U, 0x791814ceU, 0xbf73c737U, + 0xea53f7cdU, 0x5b5ffdaaU, 0x14df3d6fU, 0x867844dbU, + 0x81caaff3U, 0x3eb968c4U, 0x2c382434U, 0x5fc2a340U, + 0x72161dc3U, 0x0cbce225U, 0x8b283c49U, 0x41ff0d95U, + 0x7139a801U, 0xde080cb3U, 0x9cd8b4e4U, 0x906456c1U, + 0x617bcb84U, 0x70d532b6U, 0x74486c5cU, 0x42d0b857U, +}; +static const u32 Td2[256] = { + 0xa75051f4U, 0x65537e41U, 0xa4c31a17U, 0x5e963a27U, + 0x6bcb3babU, 0x45f11f9dU, 0x58abacfaU, 0x03934be3U, + 0xfa552030U, 0x6df6ad76U, 0x769188ccU, 0x4c25f502U, + 0xd7fc4fe5U, 0xcbd7c52aU, 0x44802635U, 0xa38fb562U, + 0x5a49deb1U, 0x1b6725baU, 0x0e9845eaU, 0xc0e15dfeU, + 0x7502c32fU, 0xf012814cU, 0x97a38d46U, 0xf9c66bd3U, + 0x5fe7038fU, 0x9c951592U, 0x7aebbf6dU, 0x59da9552U, + 0x832dd4beU, 0x21d35874U, 0x692949e0U, 0xc8448ec9U, + 0x896a75c2U, 0x7978f48eU, 0x3e6b9958U, 0x71dd27b9U, + 0x4fb6bee1U, 0xad17f088U, 0xac66c920U, 0x3ab47dceU, + 0x4a1863dfU, 0x3182e51aU, 0x33609751U, 0x7f456253U, + 0x77e0b164U, 0xae84bb6bU, 0xa01cfe81U, 0x2b94f908U, + 0x68587048U, 0xfd198f45U, 0x6c8794deU, 0xf8b7527bU, + 0xd323ab73U, 0x02e2724bU, 0x8f57e31fU, 0xab2a6655U, + 0x2807b2ebU, 0xc2032fb5U, 0x7b9a86c5U, 0x08a5d337U, + 0x87f23028U, 0xa5b223bfU, 0x6aba0203U, 0x825ced16U, + 0x1c2b8acfU, 0xb492a779U, 0xf2f0f307U, 0xe2a14e69U, + 0xf4cd65daU, 0xbed50605U, 0x621fd134U, 0xfe8ac4a6U, + 0x539d342eU, 0x55a0a2f3U, 0xe132058aU, 0xeb75a4f6U, + 0xec390b83U, 0xefaa4060U, 0x9f065e71U, 0x1051bd6eU, + + 0x8af93e21U, 0x063d96ddU, 0x05aedd3eU, 0xbd464de6U, + 0x8db59154U, 0x5d0571c4U, 0xd46f0406U, 0x15ff6050U, + 0xfb241998U, 0xe997d6bdU, 0x43cc8940U, 0x9e7767d9U, + 0x42bdb0e8U, 0x8b880789U, 0x5b38e719U, 0xeedb79c8U, + 0x0a47a17cU, 0x0fe97c42U, 0x1ec9f884U, 0x00000000U, + 0x86830980U, 0xed48322bU, 0x70ac1e11U, 0x724e6c5aU, + 0xfffbfd0eU, 0x38560f85U, 0xd51e3daeU, 0x3927362dU, + 0xd9640a0fU, 0xa621685cU, 0x54d19b5bU, 0x2e3a2436U, + 0x67b10c0aU, 0xe70f9357U, 0x96d2b4eeU, 0x919e1b9bU, + 0xc54f80c0U, 0x20a261dcU, 0x4b695a77U, 0x1a161c12U, + 0xba0ae293U, 0x2ae5c0a0U, 0xe0433c22U, 0x171d121bU, + 0x0d0b0e09U, 0xc7adf28bU, 0xa8b92db6U, 0xa9c8141eU, + 0x198557f1U, 0x074caf75U, 0xddbbee99U, 0x60fda37fU, + 0x269ff701U, 0xf5bc5c72U, 0x3bc54466U, 0x7e345bfbU, + 0x29768b43U, 0xc6dccb23U, 0xfc68b6edU, 0xf163b8e4U, + 0xdccad731U, 0x85104263U, 0x22401397U, 0x112084c6U, + 0x247d854aU, 0x3df8d2bbU, 0x3211aef9U, 0xa16dc729U, + 0x2f4b1d9eU, 0x30f3dcb2U, 0x52ec0d86U, 0xe3d077c1U, + 0x166c2bb3U, 0xb999a970U, 0x48fa1194U, 0x642247e9U, + 0x8cc4a8fcU, 0x3f1aa0f0U, 0x2cd8567dU, 0x90ef2233U, + 0x4ec78749U, 0xd1c1d938U, 0xa2fe8ccaU, 0x0b3698d4U, + 0x81cfa6f5U, 0xde28a57aU, 0x8e26dab7U, 0xbfa43fadU, + 0x9de42c3aU, 0x920d5078U, 0xcc9b6a5fU, 0x4662547eU, + 0x13c2f68dU, 0xb8e890d8U, 0xf75e2e39U, 0xaff582c3U, + 0x80be9f5dU, 0x937c69d0U, 0x2da96fd5U, 0x12b3cf25U, + 0x993bc8acU, 0x7da71018U, 0x636ee89cU, 0xbb7bdb3bU, + 0x7809cd26U, 0x18f46e59U, 0xb701ec9aU, 0x9aa8834fU, + 0x6e65e695U, 0xe67eaaffU, 0xcf0821bcU, 0xe8e6ef15U, + 0x9bd9bae7U, 0x36ce4a6fU, 0x09d4ea9fU, 0x7cd629b0U, + 0xb2af31a4U, 0x23312a3fU, 0x9430c6a5U, 0x66c035a2U, + 0xbc37744eU, 0xcaa6fc82U, 0xd0b0e090U, 0xd81533a7U, + 0x984af104U, 0xdaf741ecU, 0x500e7fcdU, 0xf62f1791U, + 0xd68d764dU, 0xb04d43efU, 0x4d54ccaaU, 0x04dfe496U, + 0xb5e39ed1U, 0x881b4c6aU, 0x1fb8c12cU, 0x517f4665U, + 0xea049d5eU, 0x355d018cU, 0x7473fa87U, 0x412efb0bU, + 0x1d5ab367U, 0xd25292dbU, 0x5633e910U, 0x47136dd6U, + 0x618c9ad7U, 0x0c7a37a1U, 0x148e59f8U, 0x3c89eb13U, + 0x27eecea9U, 0xc935b761U, 0xe5ede11cU, 0xb13c7a47U, + 0xdf599cd2U, 0x733f55f2U, 0xce791814U, 0x37bf73c7U, + 0xcdea53f7U, 0xaa5b5ffdU, 0x6f14df3dU, 0xdb867844U, + 0xf381caafU, 0xc43eb968U, 0x342c3824U, 0x405fc2a3U, + 0xc372161dU, 0x250cbce2U, 0x498b283cU, 0x9541ff0dU, + 0x017139a8U, 0xb3de080cU, 0xe49cd8b4U, 0xc1906456U, + 0x84617bcbU, 0xb670d532U, 0x5c74486cU, 0x5742d0b8U, +}; +static const u32 Td3[256] = { + 0xf4a75051U, 0x4165537eU, 0x17a4c31aU, 0x275e963aU, + 0xab6bcb3bU, 0x9d45f11fU, 0xfa58abacU, 0xe303934bU, + 0x30fa5520U, 0x766df6adU, 0xcc769188U, 0x024c25f5U, + 0xe5d7fc4fU, 0x2acbd7c5U, 0x35448026U, 0x62a38fb5U, + 0xb15a49deU, 0xba1b6725U, 0xea0e9845U, 0xfec0e15dU, + 0x2f7502c3U, 0x4cf01281U, 0x4697a38dU, 0xd3f9c66bU, + 0x8f5fe703U, 0x929c9515U, 0x6d7aebbfU, 0x5259da95U, + 0xbe832dd4U, 0x7421d358U, 0xe0692949U, 0xc9c8448eU, + 0xc2896a75U, 0x8e7978f4U, 0x583e6b99U, 0xb971dd27U, + 0xe14fb6beU, 0x88ad17f0U, 0x20ac66c9U, 0xce3ab47dU, + 0xdf4a1863U, 0x1a3182e5U, 0x51336097U, 0x537f4562U, + 0x6477e0b1U, 0x6bae84bbU, 0x81a01cfeU, 0x082b94f9U, + 0x48685870U, 0x45fd198fU, 0xde6c8794U, 0x7bf8b752U, + 0x73d323abU, 0x4b02e272U, 0x1f8f57e3U, 0x55ab2a66U, + 0xeb2807b2U, 0xb5c2032fU, 0xc57b9a86U, 0x3708a5d3U, + 0x2887f230U, 0xbfa5b223U, 0x036aba02U, 0x16825cedU, + 0xcf1c2b8aU, 0x79b492a7U, 0x07f2f0f3U, 0x69e2a14eU, + 0xdaf4cd65U, 0x05bed506U, 0x34621fd1U, 0xa6fe8ac4U, + 0x2e539d34U, 0xf355a0a2U, 0x8ae13205U, 0xf6eb75a4U, + 0x83ec390bU, 0x60efaa40U, 0x719f065eU, 0x6e1051bdU, + 0x218af93eU, 0xdd063d96U, 0x3e05aeddU, 0xe6bd464dU, + 0x548db591U, 0xc45d0571U, 0x06d46f04U, 0x5015ff60U, + 0x98fb2419U, 0xbde997d6U, 0x4043cc89U, 0xd99e7767U, + 0xe842bdb0U, 0x898b8807U, 0x195b38e7U, 0xc8eedb79U, + 0x7c0a47a1U, 0x420fe97cU, 0x841ec9f8U, 0x00000000U, + 0x80868309U, 0x2bed4832U, 0x1170ac1eU, 0x5a724e6cU, + 0x0efffbfdU, 0x8538560fU, 0xaed51e3dU, 0x2d392736U, + 0x0fd9640aU, 0x5ca62168U, 0x5b54d19bU, 0x362e3a24U, + 0x0a67b10cU, 0x57e70f93U, 0xee96d2b4U, 0x9b919e1bU, + 0xc0c54f80U, 0xdc20a261U, 0x774b695aU, 0x121a161cU, + 0x93ba0ae2U, 0xa02ae5c0U, 0x22e0433cU, 0x1b171d12U, + 0x090d0b0eU, 0x8bc7adf2U, 0xb6a8b92dU, 0x1ea9c814U, + 0xf1198557U, 0x75074cafU, 0x99ddbbeeU, 0x7f60fda3U, + 0x01269ff7U, 0x72f5bc5cU, 0x663bc544U, 0xfb7e345bU, + 0x4329768bU, 0x23c6dccbU, 0xedfc68b6U, 0xe4f163b8U, + 0x31dccad7U, 0x63851042U, 0x97224013U, 0xc6112084U, + 0x4a247d85U, 0xbb3df8d2U, 0xf93211aeU, 0x29a16dc7U, + 0x9e2f4b1dU, 0xb230f3dcU, 0x8652ec0dU, 0xc1e3d077U, + 0xb3166c2bU, 0x70b999a9U, 0x9448fa11U, 0xe9642247U, + 0xfc8cc4a8U, 0xf03f1aa0U, 0x7d2cd856U, 0x3390ef22U, + 0x494ec787U, 0x38d1c1d9U, 0xcaa2fe8cU, 0xd40b3698U, + 0xf581cfa6U, 0x7ade28a5U, 0xb78e26daU, 0xadbfa43fU, + 0x3a9de42cU, 0x78920d50U, 0x5fcc9b6aU, 0x7e466254U, + 0x8d13c2f6U, 0xd8b8e890U, 0x39f75e2eU, 0xc3aff582U, + 0x5d80be9fU, 0xd0937c69U, 0xd52da96fU, 0x2512b3cfU, + 0xac993bc8U, 0x187da710U, 0x9c636ee8U, 0x3bbb7bdbU, + 0x267809cdU, 0x5918f46eU, 0x9ab701ecU, 0x4f9aa883U, + 0x956e65e6U, 0xffe67eaaU, 0xbccf0821U, 0x15e8e6efU, + 0xe79bd9baU, 0x6f36ce4aU, 0x9f09d4eaU, 0xb07cd629U, + 0xa4b2af31U, 0x3f23312aU, 0xa59430c6U, 0xa266c035U, + 0x4ebc3774U, 0x82caa6fcU, 0x90d0b0e0U, 0xa7d81533U, + 0x04984af1U, 0xecdaf741U, 0xcd500e7fU, 0x91f62f17U, + 0x4dd68d76U, 0xefb04d43U, 0xaa4d54ccU, 0x9604dfe4U, + 0xd1b5e39eU, 0x6a881b4cU, 0x2c1fb8c1U, 0x65517f46U, + 0x5eea049dU, 0x8c355d01U, 0x877473faU, 0x0b412efbU, + 0x671d5ab3U, 0xdbd25292U, 0x105633e9U, 0xd647136dU, + 0xd7618c9aU, 0xa10c7a37U, 0xf8148e59U, 0x133c89ebU, + 0xa927eeceU, 0x61c935b7U, 0x1ce5ede1U, 0x47b13c7aU, + 0xd2df599cU, 0xf2733f55U, 0x14ce7918U, 0xc737bf73U, + 0xf7cdea53U, 0xfdaa5b5fU, 0x3d6f14dfU, 0x44db8678U, + 0xaff381caU, 0x68c43eb9U, 0x24342c38U, 0xa3405fc2U, + 0x1dc37216U, 0xe2250cbcU, 0x3c498b28U, 0x0d9541ffU, + 0xa8017139U, 0x0cb3de08U, 0xb4e49cd8U, 0x56c19064U, + 0xcb84617bU, 0x32b670d5U, 0x6c5c7448U, 0xb85742d0U, +}; +static const u32 Td4[256] = { + 0x52525252U, 0x09090909U, 0x6a6a6a6aU, 0xd5d5d5d5U, + 0x30303030U, 0x36363636U, 0xa5a5a5a5U, 0x38383838U, + 0xbfbfbfbfU, 0x40404040U, 0xa3a3a3a3U, 0x9e9e9e9eU, + 0x81818181U, 0xf3f3f3f3U, 0xd7d7d7d7U, 0xfbfbfbfbU, + 0x7c7c7c7cU, 0xe3e3e3e3U, 0x39393939U, 0x82828282U, + 0x9b9b9b9bU, 0x2f2f2f2fU, 0xffffffffU, 0x87878787U, + 0x34343434U, 0x8e8e8e8eU, 0x43434343U, 0x44444444U, + 0xc4c4c4c4U, 0xdedededeU, 0xe9e9e9e9U, 0xcbcbcbcbU, + 0x54545454U, 0x7b7b7b7bU, 0x94949494U, 0x32323232U, + 0xa6a6a6a6U, 0xc2c2c2c2U, 0x23232323U, 0x3d3d3d3dU, + 0xeeeeeeeeU, 0x4c4c4c4cU, 0x95959595U, 0x0b0b0b0bU, + 0x42424242U, 0xfafafafaU, 0xc3c3c3c3U, 0x4e4e4e4eU, + 0x08080808U, 0x2e2e2e2eU, 0xa1a1a1a1U, 0x66666666U, + 0x28282828U, 0xd9d9d9d9U, 0x24242424U, 0xb2b2b2b2U, + 0x76767676U, 0x5b5b5b5bU, 0xa2a2a2a2U, 0x49494949U, + 0x6d6d6d6dU, 0x8b8b8b8bU, 0xd1d1d1d1U, 0x25252525U, + 0x72727272U, 0xf8f8f8f8U, 0xf6f6f6f6U, 0x64646464U, + 0x86868686U, 0x68686868U, 0x98989898U, 0x16161616U, + 0xd4d4d4d4U, 0xa4a4a4a4U, 0x5c5c5c5cU, 0xccccccccU, + 0x5d5d5d5dU, 0x65656565U, 0xb6b6b6b6U, 0x92929292U, + 0x6c6c6c6cU, 0x70707070U, 0x48484848U, 0x50505050U, + 0xfdfdfdfdU, 0xededededU, 0xb9b9b9b9U, 0xdadadadaU, + 0x5e5e5e5eU, 0x15151515U, 0x46464646U, 0x57575757U, + 0xa7a7a7a7U, 0x8d8d8d8dU, 0x9d9d9d9dU, 0x84848484U, + 0x90909090U, 0xd8d8d8d8U, 0xababababU, 0x00000000U, + 0x8c8c8c8cU, 0xbcbcbcbcU, 0xd3d3d3d3U, 0x0a0a0a0aU, + 0xf7f7f7f7U, 0xe4e4e4e4U, 0x58585858U, 0x05050505U, + 0xb8b8b8b8U, 0xb3b3b3b3U, 0x45454545U, 0x06060606U, + 0xd0d0d0d0U, 0x2c2c2c2cU, 0x1e1e1e1eU, 0x8f8f8f8fU, + 0xcacacacaU, 0x3f3f3f3fU, 0x0f0f0f0fU, 0x02020202U, + 0xc1c1c1c1U, 0xafafafafU, 0xbdbdbdbdU, 0x03030303U, + 0x01010101U, 0x13131313U, 0x8a8a8a8aU, 0x6b6b6b6bU, + 0x3a3a3a3aU, 0x91919191U, 0x11111111U, 0x41414141U, + 0x4f4f4f4fU, 0x67676767U, 0xdcdcdcdcU, 0xeaeaeaeaU, + 0x97979797U, 0xf2f2f2f2U, 0xcfcfcfcfU, 0xcecececeU, + 0xf0f0f0f0U, 0xb4b4b4b4U, 0xe6e6e6e6U, 0x73737373U, + 0x96969696U, 0xacacacacU, 0x74747474U, 0x22222222U, + 0xe7e7e7e7U, 0xadadadadU, 0x35353535U, 0x85858585U, + 0xe2e2e2e2U, 0xf9f9f9f9U, 0x37373737U, 0xe8e8e8e8U, + 0x1c1c1c1cU, 0x75757575U, 0xdfdfdfdfU, 0x6e6e6e6eU, + 0x47474747U, 0xf1f1f1f1U, 0x1a1a1a1aU, 0x71717171U, + 0x1d1d1d1dU, 0x29292929U, 0xc5c5c5c5U, 0x89898989U, + 0x6f6f6f6fU, 0xb7b7b7b7U, 0x62626262U, 0x0e0e0e0eU, + 0xaaaaaaaaU, 0x18181818U, 0xbebebebeU, 0x1b1b1b1bU, + 0xfcfcfcfcU, 0x56565656U, 0x3e3e3e3eU, 0x4b4b4b4bU, + 0xc6c6c6c6U, 0xd2d2d2d2U, 0x79797979U, 0x20202020U, + 0x9a9a9a9aU, 0xdbdbdbdbU, 0xc0c0c0c0U, 0xfefefefeU, + 0x78787878U, 0xcdcdcdcdU, 0x5a5a5a5aU, 0xf4f4f4f4U, + 0x1f1f1f1fU, 0xddddddddU, 0xa8a8a8a8U, 0x33333333U, + 0x88888888U, 0x07070707U, 0xc7c7c7c7U, 0x31313131U, + 0xb1b1b1b1U, 0x12121212U, 0x10101010U, 0x59595959U, + 0x27272727U, 0x80808080U, 0xececececU, 0x5f5f5f5fU, + 0x60606060U, 0x51515151U, 0x7f7f7f7fU, 0xa9a9a9a9U, + 0x19191919U, 0xb5b5b5b5U, 0x4a4a4a4aU, 0x0d0d0d0dU, + 0x2d2d2d2dU, 0xe5e5e5e5U, 0x7a7a7a7aU, 0x9f9f9f9fU, + 0x93939393U, 0xc9c9c9c9U, 0x9c9c9c9cU, 0xefefefefU, + 0xa0a0a0a0U, 0xe0e0e0e0U, 0x3b3b3b3bU, 0x4d4d4d4dU, + 0xaeaeaeaeU, 0x2a2a2a2aU, 0xf5f5f5f5U, 0xb0b0b0b0U, + 0xc8c8c8c8U, 0xebebebebU, 0xbbbbbbbbU, 0x3c3c3c3cU, + 0x83838383U, 0x53535353U, 0x99999999U, 0x61616161U, + 0x17171717U, 0x2b2b2b2bU, 0x04040404U, 0x7e7e7e7eU, + 0xbabababaU, 0x77777777U, 0xd6d6d6d6U, 0x26262626U, + 0xe1e1e1e1U, 0x69696969U, 0x14141414U, 0x63636363U, + 0x55555555U, 0x21212121U, 0x0c0c0c0cU, 0x7d7d7d7dU, +}; +static const u32 rcon[] = { + 0x01000000, 0x02000000, 0x04000000, 0x08000000, + 0x10000000, 0x20000000, 0x40000000, 0x80000000, + 0x1B000000, 0x36000000, /* for 128-bit blocks, Rijndael never uses more than 10 rcon values */ +}; + +#define SWAP(x) (_lrotl(x, 8) & 0x00ff00ff | _lrotr(x, 8) & 0xff00ff00) + +#ifdef _MSC_VER +#define GETU32(p) SWAP(*((u32 *)(p))) +#define PUTU32(ct, st) { *((u32 *)(ct)) = SWAP((st)); } +#else +#define GETU32(pt) (((u32)(pt)[0] << 24) ^ ((u32)(pt)[1] << 16) ^ ((u32)(pt)[2] << 8) ^ ((u32)(pt)[3])) +#define PUTU32(ct, st) { (ct)[0] = (u8)((st) >> 24); (ct)[1] = (u8)((st) >> 16); (ct)[2] = (u8)((st) >> 8); (ct)[3] = (u8)(st); } +#endif + +/** + * Expand the cipher key into the encryption key schedule. + * + * @return the number of rounds for the given cipher key size. + */ +int rijndaelKeySetupEnc(u32 rk[/*4*(Nr + 1)*/], const u8 cipherKey[], int keyBits) { + int i = 0; + u32 temp; + + rk[0] = GETU32(cipherKey ); + rk[1] = GETU32(cipherKey + 4); + rk[2] = GETU32(cipherKey + 8); + rk[3] = GETU32(cipherKey + 12); + if (keyBits == 128) { + for (;;) { + temp = rk[3]; + rk[4] = rk[0] ^ + (Te4[(temp >> 16) & 0xff] & 0xff000000) ^ + (Te4[(temp >> 8) & 0xff] & 0x00ff0000) ^ + (Te4[(temp ) & 0xff] & 0x0000ff00) ^ + (Te4[(temp >> 24) ] & 0x000000ff) ^ + rcon[i]; + rk[5] = rk[1] ^ rk[4]; + rk[6] = rk[2] ^ rk[5]; + rk[7] = rk[3] ^ rk[6]; + if (++i == 10) { + return 10; + } + rk += 4; + } + } + rk[4] = GETU32(cipherKey + 16); + rk[5] = GETU32(cipherKey + 20); + if (keyBits == 192) { + for (;;) { + temp = rk[ 5]; + rk[ 6] = rk[ 0] ^ + (Te4[(temp >> 16) & 0xff] & 0xff000000) ^ + (Te4[(temp >> 8) & 0xff] & 0x00ff0000) ^ + (Te4[(temp ) & 0xff] & 0x0000ff00) ^ + (Te4[(temp >> 24) ] & 0x000000ff) ^ + rcon[i]; + rk[ 7] = rk[ 1] ^ rk[ 6]; + rk[ 8] = rk[ 2] ^ rk[ 7]; + rk[ 9] = rk[ 3] ^ rk[ 8]; + if (++i == 8) { + return 12; + } + rk[10] = rk[ 4] ^ rk[ 9]; + rk[11] = rk[ 5] ^ rk[10]; + rk += 6; + } + } + rk[6] = GETU32(cipherKey + 24); + rk[7] = GETU32(cipherKey + 28); + if (keyBits == 256) { + for (;;) { + temp = rk[ 7]; + rk[ 8] = rk[ 0] ^ + (Te4[(temp >> 16) & 0xff] & 0xff000000) ^ + (Te4[(temp >> 8) & 0xff] & 0x00ff0000) ^ + (Te4[(temp ) & 0xff] & 0x0000ff00) ^ + (Te4[(temp >> 24) ] & 0x000000ff) ^ + rcon[i]; + rk[ 9] = rk[ 1] ^ rk[ 8]; + rk[10] = rk[ 2] ^ rk[ 9]; + rk[11] = rk[ 3] ^ rk[10]; + if (++i == 7) { + return 14; + } + temp = rk[11]; + rk[12] = rk[ 4] ^ + (Te4[(temp >> 24) ] & 0xff000000) ^ + (Te4[(temp >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(temp >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(temp ) & 0xff] & 0x000000ff); + rk[13] = rk[ 5] ^ rk[12]; + rk[14] = rk[ 6] ^ rk[13]; + rk[15] = rk[ 7] ^ rk[14]; + + rk += 8; + } + } + return 0; +} + +/** + * Expand the cipher key into the decryption key schedule. + * + * @return the number of rounds for the given cipher key size. + */ +int rijndaelKeySetupDec(u32 rk[/*4*(Nr + 1)*/], const u8 cipherKey[], int keyBits) { + int Nr, i, j; + u32 temp; + + /* expand the cipher key: */ + Nr = rijndaelKeySetupEnc(rk, cipherKey, keyBits); + /* invert the order of the round keys: */ + for (i = 0, j = 4*Nr; i < j; i += 4, j -= 4) { + temp = rk[i ]; rk[i ] = rk[j ]; rk[j ] = temp; + temp = rk[i + 1]; rk[i + 1] = rk[j + 1]; rk[j + 1] = temp; + temp = rk[i + 2]; rk[i + 2] = rk[j + 2]; rk[j + 2] = temp; + temp = rk[i + 3]; rk[i + 3] = rk[j + 3]; rk[j + 3] = temp; + } + /* apply the inverse MixColumn transform to all round keys but the first and the last: */ + for (i = 1; i < Nr; i++) { + rk += 4; + rk[0] = + Td0[Te4[(rk[0] >> 24) ] & 0xff] ^ + Td1[Te4[(rk[0] >> 16) & 0xff] & 0xff] ^ + Td2[Te4[(rk[0] >> 8) & 0xff] & 0xff] ^ + Td3[Te4[(rk[0] ) & 0xff] & 0xff]; + rk[1] = + Td0[Te4[(rk[1] >> 24) ] & 0xff] ^ + Td1[Te4[(rk[1] >> 16) & 0xff] & 0xff] ^ + Td2[Te4[(rk[1] >> 8) & 0xff] & 0xff] ^ + Td3[Te4[(rk[1] ) & 0xff] & 0xff]; + rk[2] = + Td0[Te4[(rk[2] >> 24) ] & 0xff] ^ + Td1[Te4[(rk[2] >> 16) & 0xff] & 0xff] ^ + Td2[Te4[(rk[2] >> 8) & 0xff] & 0xff] ^ + Td3[Te4[(rk[2] ) & 0xff] & 0xff]; + rk[3] = + Td0[Te4[(rk[3] >> 24) ] & 0xff] ^ + Td1[Te4[(rk[3] >> 16) & 0xff] & 0xff] ^ + Td2[Te4[(rk[3] >> 8) & 0xff] & 0xff] ^ + Td3[Te4[(rk[3] ) & 0xff] & 0xff]; + } + return Nr; +} + +void rijndaelEncrypt(const u32 rk[/*4*(Nr + 1)*/], int Nr, const u8 pt[16], u8 ct[16]) { + u32 s0, s1, s2, s3, t0, t1, t2, t3; +#ifndef FULL_UNROLL + int r; +#endif /* ?FULL_UNROLL */ + + /* + * map byte array block to cipher state + * and add initial round key: + */ + s0 = GETU32(pt ) ^ rk[0]; + s1 = GETU32(pt + 4) ^ rk[1]; + s2 = GETU32(pt + 8) ^ rk[2]; + s3 = GETU32(pt + 12) ^ rk[3]; +#ifdef FULL_UNROLL + /* round 1: */ + t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >> 8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[ 4]; + t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >> 8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[ 5]; + t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >> 8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[ 6]; + t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >> 8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[ 7]; + /* round 2: */ + s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >> 8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[ 8]; + s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >> 8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[ 9]; + s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >> 8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[10]; + s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >> 8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[11]; + /* round 3: */ + t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >> 8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[12]; + t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >> 8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[13]; + t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >> 8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[14]; + t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >> 8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[15]; + /* round 4: */ + s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >> 8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[16]; + s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >> 8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[17]; + s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >> 8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[18]; + s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >> 8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[19]; + /* round 5: */ + t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >> 8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[20]; + t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >> 8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[21]; + t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >> 8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[22]; + t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >> 8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[23]; + /* round 6: */ + s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >> 8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[24]; + s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >> 8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[25]; + s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >> 8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[26]; + s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >> 8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[27]; + /* round 7: */ + t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >> 8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[28]; + t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >> 8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[29]; + t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >> 8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[30]; + t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >> 8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[31]; + /* round 8: */ + s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >> 8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[32]; + s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >> 8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[33]; + s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >> 8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[34]; + s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >> 8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[35]; + /* round 9: */ + t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >> 8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[36]; + t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >> 8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[37]; + t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >> 8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[38]; + t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >> 8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[39]; + if (Nr > 10) { + /* round 10: */ + s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >> 8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[40]; + s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >> 8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[41]; + s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >> 8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[42]; + s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >> 8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[43]; + /* round 11: */ + t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >> 8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[44]; + t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >> 8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[45]; + t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >> 8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[46]; + t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >> 8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[47]; + if (Nr > 12) { + /* round 12: */ + s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >> 8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[48]; + s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >> 8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[49]; + s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >> 8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[50]; + s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >> 8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[51]; + /* round 13: */ + t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >> 8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[52]; + t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >> 8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[53]; + t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >> 8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[54]; + t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >> 8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[55]; + } + } + rk += Nr << 2; +#else /* !FULL_UNROLL */ + /* + * Nr - 1 full rounds: + */ + r = Nr >> 1; + for (;;) { + t0 = + Te0[(s0 >> 24) ] ^ + Te1[(s1 >> 16) & 0xff] ^ + Te2[(s2 >> 8) & 0xff] ^ + Te3[(s3 ) & 0xff] ^ + rk[4]; + t1 = + Te0[(s1 >> 24) ] ^ + Te1[(s2 >> 16) & 0xff] ^ + Te2[(s3 >> 8) & 0xff] ^ + Te3[(s0 ) & 0xff] ^ + rk[5]; + t2 = + Te0[(s2 >> 24) ] ^ + Te1[(s3 >> 16) & 0xff] ^ + Te2[(s0 >> 8) & 0xff] ^ + Te3[(s1 ) & 0xff] ^ + rk[6]; + t3 = + Te0[(s3 >> 24) ] ^ + Te1[(s0 >> 16) & 0xff] ^ + Te2[(s1 >> 8) & 0xff] ^ + Te3[(s2 ) & 0xff] ^ + rk[7]; + + rk += 8; + if (--r == 0) { + break; + } + + s0 = + Te0[(t0 >> 24) ] ^ + Te1[(t1 >> 16) & 0xff] ^ + Te2[(t2 >> 8) & 0xff] ^ + Te3[(t3 ) & 0xff] ^ + rk[0]; + s1 = + Te0[(t1 >> 24) ] ^ + Te1[(t2 >> 16) & 0xff] ^ + Te2[(t3 >> 8) & 0xff] ^ + Te3[(t0 ) & 0xff] ^ + rk[1]; + s2 = + Te0[(t2 >> 24) ] ^ + Te1[(t3 >> 16) & 0xff] ^ + Te2[(t0 >> 8) & 0xff] ^ + Te3[(t1 ) & 0xff] ^ + rk[2]; + s3 = + Te0[(t3 >> 24) ] ^ + Te1[(t0 >> 16) & 0xff] ^ + Te2[(t1 >> 8) & 0xff] ^ + Te3[(t2 ) & 0xff] ^ + rk[3]; + } +#endif /* ?FULL_UNROLL */ + /* + * apply last round and + * map cipher state to byte array block: + */ + s0 = + (Te4[(t0 >> 24) ] & 0xff000000) ^ + (Te4[(t1 >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(t2 >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(t3 ) & 0xff] & 0x000000ff) ^ + rk[0]; + PUTU32(ct , s0); + s1 = + (Te4[(t1 >> 24) ] & 0xff000000) ^ + (Te4[(t2 >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(t3 >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(t0 ) & 0xff] & 0x000000ff) ^ + rk[1]; + PUTU32(ct + 4, s1); + s2 = + (Te4[(t2 >> 24) ] & 0xff000000) ^ + (Te4[(t3 >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(t0 >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(t1 ) & 0xff] & 0x000000ff) ^ + rk[2]; + PUTU32(ct + 8, s2); + s3 = + (Te4[(t3 >> 24) ] & 0xff000000) ^ + (Te4[(t0 >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(t1 >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(t2 ) & 0xff] & 0x000000ff) ^ + rk[3]; + PUTU32(ct + 12, s3); +} + +void rijndaelDecrypt(const u32 rk[/*4*(Nr + 1)*/], int Nr, const u8 ct[16], u8 pt[16]) { + u32 s0, s1, s2, s3, t0, t1, t2, t3; +#ifndef FULL_UNROLL + int r; +#endif /* ?FULL_UNROLL */ + + /* + * map byte array block to cipher state + * and add initial round key: + */ + s0 = GETU32(ct ) ^ rk[0]; + s1 = GETU32(ct + 4) ^ rk[1]; + s2 = GETU32(ct + 8) ^ rk[2]; + s3 = GETU32(ct + 12) ^ rk[3]; +#ifdef FULL_UNROLL + /* round 1: */ + t0 = Td0[s0 >> 24] ^ Td1[(s3 >> 16) & 0xff] ^ Td2[(s2 >> 8) & 0xff] ^ Td3[s1 & 0xff] ^ rk[ 4]; + t1 = Td0[s1 >> 24] ^ Td1[(s0 >> 16) & 0xff] ^ Td2[(s3 >> 8) & 0xff] ^ Td3[s2 & 0xff] ^ rk[ 5]; + t2 = Td0[s2 >> 24] ^ Td1[(s1 >> 16) & 0xff] ^ Td2[(s0 >> 8) & 0xff] ^ Td3[s3 & 0xff] ^ rk[ 6]; + t3 = Td0[s3 >> 24] ^ Td1[(s2 >> 16) & 0xff] ^ Td2[(s1 >> 8) & 0xff] ^ Td3[s0 & 0xff] ^ rk[ 7]; + /* round 2: */ + s0 = Td0[t0 >> 24] ^ Td1[(t3 >> 16) & 0xff] ^ Td2[(t2 >> 8) & 0xff] ^ Td3[t1 & 0xff] ^ rk[ 8]; + s1 = Td0[t1 >> 24] ^ Td1[(t0 >> 16) & 0xff] ^ Td2[(t3 >> 8) & 0xff] ^ Td3[t2 & 0xff] ^ rk[ 9]; + s2 = Td0[t2 >> 24] ^ Td1[(t1 >> 16) & 0xff] ^ Td2[(t0 >> 8) & 0xff] ^ Td3[t3 & 0xff] ^ rk[10]; + s3 = Td0[t3 >> 24] ^ Td1[(t2 >> 16) & 0xff] ^ Td2[(t1 >> 8) & 0xff] ^ Td3[t0 & 0xff] ^ rk[11]; + /* round 3: */ + t0 = Td0[s0 >> 24] ^ Td1[(s3 >> 16) & 0xff] ^ Td2[(s2 >> 8) & 0xff] ^ Td3[s1 & 0xff] ^ rk[12]; + t1 = Td0[s1 >> 24] ^ Td1[(s0 >> 16) & 0xff] ^ Td2[(s3 >> 8) & 0xff] ^ Td3[s2 & 0xff] ^ rk[13]; + t2 = Td0[s2 >> 24] ^ Td1[(s1 >> 16) & 0xff] ^ Td2[(s0 >> 8) & 0xff] ^ Td3[s3 & 0xff] ^ rk[14]; + t3 = Td0[s3 >> 24] ^ Td1[(s2 >> 16) & 0xff] ^ Td2[(s1 >> 8) & 0xff] ^ Td3[s0 & 0xff] ^ rk[15]; + /* round 4: */ + s0 = Td0[t0 >> 24] ^ Td1[(t3 >> 16) & 0xff] ^ Td2[(t2 >> 8) & 0xff] ^ Td3[t1 & 0xff] ^ rk[16]; + s1 = Td0[t1 >> 24] ^ Td1[(t0 >> 16) & 0xff] ^ Td2[(t3 >> 8) & 0xff] ^ Td3[t2 & 0xff] ^ rk[17]; + s2 = Td0[t2 >> 24] ^ Td1[(t1 >> 16) & 0xff] ^ Td2[(t0 >> 8) & 0xff] ^ Td3[t3 & 0xff] ^ rk[18]; + s3 = Td0[t3 >> 24] ^ Td1[(t2 >> 16) & 0xff] ^ Td2[(t1 >> 8) & 0xff] ^ Td3[t0 & 0xff] ^ rk[19]; + /* round 5: */ + t0 = Td0[s0 >> 24] ^ Td1[(s3 >> 16) & 0xff] ^ Td2[(s2 >> 8) & 0xff] ^ Td3[s1 & 0xff] ^ rk[20]; + t1 = Td0[s1 >> 24] ^ Td1[(s0 >> 16) & 0xff] ^ Td2[(s3 >> 8) & 0xff] ^ Td3[s2 & 0xff] ^ rk[21]; + t2 = Td0[s2 >> 24] ^ Td1[(s1 >> 16) & 0xff] ^ Td2[(s0 >> 8) & 0xff] ^ Td3[s3 & 0xff] ^ rk[22]; + t3 = Td0[s3 >> 24] ^ Td1[(s2 >> 16) & 0xff] ^ Td2[(s1 >> 8) & 0xff] ^ Td3[s0 & 0xff] ^ rk[23]; + /* round 6: */ + s0 = Td0[t0 >> 24] ^ Td1[(t3 >> 16) & 0xff] ^ Td2[(t2 >> 8) & 0xff] ^ Td3[t1 & 0xff] ^ rk[24]; + s1 = Td0[t1 >> 24] ^ Td1[(t0 >> 16) & 0xff] ^ Td2[(t3 >> 8) & 0xff] ^ Td3[t2 & 0xff] ^ rk[25]; + s2 = Td0[t2 >> 24] ^ Td1[(t1 >> 16) & 0xff] ^ Td2[(t0 >> 8) & 0xff] ^ Td3[t3 & 0xff] ^ rk[26]; + s3 = Td0[t3 >> 24] ^ Td1[(t2 >> 16) & 0xff] ^ Td2[(t1 >> 8) & 0xff] ^ Td3[t0 & 0xff] ^ rk[27]; + /* round 7: */ + t0 = Td0[s0 >> 24] ^ Td1[(s3 >> 16) & 0xff] ^ Td2[(s2 >> 8) & 0xff] ^ Td3[s1 & 0xff] ^ rk[28]; + t1 = Td0[s1 >> 24] ^ Td1[(s0 >> 16) & 0xff] ^ Td2[(s3 >> 8) & 0xff] ^ Td3[s2 & 0xff] ^ rk[29]; + t2 = Td0[s2 >> 24] ^ Td1[(s1 >> 16) & 0xff] ^ Td2[(s0 >> 8) & 0xff] ^ Td3[s3 & 0xff] ^ rk[30]; + t3 = Td0[s3 >> 24] ^ Td1[(s2 >> 16) & 0xff] ^ Td2[(s1 >> 8) & 0xff] ^ Td3[s0 & 0xff] ^ rk[31]; + /* round 8: */ + s0 = Td0[t0 >> 24] ^ Td1[(t3 >> 16) & 0xff] ^ Td2[(t2 >> 8) & 0xff] ^ Td3[t1 & 0xff] ^ rk[32]; + s1 = Td0[t1 >> 24] ^ Td1[(t0 >> 16) & 0xff] ^ Td2[(t3 >> 8) & 0xff] ^ Td3[t2 & 0xff] ^ rk[33]; + s2 = Td0[t2 >> 24] ^ Td1[(t1 >> 16) & 0xff] ^ Td2[(t0 >> 8) & 0xff] ^ Td3[t3 & 0xff] ^ rk[34]; + s3 = Td0[t3 >> 24] ^ Td1[(t2 >> 16) & 0xff] ^ Td2[(t1 >> 8) & 0xff] ^ Td3[t0 & 0xff] ^ rk[35]; + /* round 9: */ + t0 = Td0[s0 >> 24] ^ Td1[(s3 >> 16) & 0xff] ^ Td2[(s2 >> 8) & 0xff] ^ Td3[s1 & 0xff] ^ rk[36]; + t1 = Td0[s1 >> 24] ^ Td1[(s0 >> 16) & 0xff] ^ Td2[(s3 >> 8) & 0xff] ^ Td3[s2 & 0xff] ^ rk[37]; + t2 = Td0[s2 >> 24] ^ Td1[(s1 >> 16) & 0xff] ^ Td2[(s0 >> 8) & 0xff] ^ Td3[s3 & 0xff] ^ rk[38]; + t3 = Td0[s3 >> 24] ^ Td1[(s2 >> 16) & 0xff] ^ Td2[(s1 >> 8) & 0xff] ^ Td3[s0 & 0xff] ^ rk[39]; + if (Nr > 10) { + /* round 10: */ + s0 = Td0[t0 >> 24] ^ Td1[(t3 >> 16) & 0xff] ^ Td2[(t2 >> 8) & 0xff] ^ Td3[t1 & 0xff] ^ rk[40]; + s1 = Td0[t1 >> 24] ^ Td1[(t0 >> 16) & 0xff] ^ Td2[(t3 >> 8) & 0xff] ^ Td3[t2 & 0xff] ^ rk[41]; + s2 = Td0[t2 >> 24] ^ Td1[(t1 >> 16) & 0xff] ^ Td2[(t0 >> 8) & 0xff] ^ Td3[t3 & 0xff] ^ rk[42]; + s3 = Td0[t3 >> 24] ^ Td1[(t2 >> 16) & 0xff] ^ Td2[(t1 >> 8) & 0xff] ^ Td3[t0 & 0xff] ^ rk[43]; + /* round 11: */ + t0 = Td0[s0 >> 24] ^ Td1[(s3 >> 16) & 0xff] ^ Td2[(s2 >> 8) & 0xff] ^ Td3[s1 & 0xff] ^ rk[44]; + t1 = Td0[s1 >> 24] ^ Td1[(s0 >> 16) & 0xff] ^ Td2[(s3 >> 8) & 0xff] ^ Td3[s2 & 0xff] ^ rk[45]; + t2 = Td0[s2 >> 24] ^ Td1[(s1 >> 16) & 0xff] ^ Td2[(s0 >> 8) & 0xff] ^ Td3[s3 & 0xff] ^ rk[46]; + t3 = Td0[s3 >> 24] ^ Td1[(s2 >> 16) & 0xff] ^ Td2[(s1 >> 8) & 0xff] ^ Td3[s0 & 0xff] ^ rk[47]; + if (Nr > 12) { + /* round 12: */ + s0 = Td0[t0 >> 24] ^ Td1[(t3 >> 16) & 0xff] ^ Td2[(t2 >> 8) & 0xff] ^ Td3[t1 & 0xff] ^ rk[48]; + s1 = Td0[t1 >> 24] ^ Td1[(t0 >> 16) & 0xff] ^ Td2[(t3 >> 8) & 0xff] ^ Td3[t2 & 0xff] ^ rk[49]; + s2 = Td0[t2 >> 24] ^ Td1[(t1 >> 16) & 0xff] ^ Td2[(t0 >> 8) & 0xff] ^ Td3[t3 & 0xff] ^ rk[50]; + s3 = Td0[t3 >> 24] ^ Td1[(t2 >> 16) & 0xff] ^ Td2[(t1 >> 8) & 0xff] ^ Td3[t0 & 0xff] ^ rk[51]; + /* round 13: */ + t0 = Td0[s0 >> 24] ^ Td1[(s3 >> 16) & 0xff] ^ Td2[(s2 >> 8) & 0xff] ^ Td3[s1 & 0xff] ^ rk[52]; + t1 = Td0[s1 >> 24] ^ Td1[(s0 >> 16) & 0xff] ^ Td2[(s3 >> 8) & 0xff] ^ Td3[s2 & 0xff] ^ rk[53]; + t2 = Td0[s2 >> 24] ^ Td1[(s1 >> 16) & 0xff] ^ Td2[(s0 >> 8) & 0xff] ^ Td3[s3 & 0xff] ^ rk[54]; + t3 = Td0[s3 >> 24] ^ Td1[(s2 >> 16) & 0xff] ^ Td2[(s1 >> 8) & 0xff] ^ Td3[s0 & 0xff] ^ rk[55]; + } + } + rk += Nr << 2; +#else /* !FULL_UNROLL */ + /* + * Nr - 1 full rounds: + */ + r = Nr >> 1; + for (;;) { + t0 = + Td0[(s0 >> 24) ] ^ + Td1[(s3 >> 16) & 0xff] ^ + Td2[(s2 >> 8) & 0xff] ^ + Td3[(s1 ) & 0xff] ^ + rk[4]; + t1 = + Td0[(s1 >> 24) ] ^ + Td1[(s0 >> 16) & 0xff] ^ + Td2[(s3 >> 8) & 0xff] ^ + Td3[(s2 ) & 0xff] ^ + rk[5]; + t2 = + Td0[(s2 >> 24) ] ^ + Td1[(s1 >> 16) & 0xff] ^ + Td2[(s0 >> 8) & 0xff] ^ + Td3[(s3 ) & 0xff] ^ + rk[6]; + t3 = + Td0[(s3 >> 24) ] ^ + Td1[(s2 >> 16) & 0xff] ^ + Td2[(s1 >> 8) & 0xff] ^ + Td3[(s0 ) & 0xff] ^ + rk[7]; + + rk += 8; + if (--r == 0) { + break; + } + + s0 = + Td0[(t0 >> 24) ] ^ + Td1[(t3 >> 16) & 0xff] ^ + Td2[(t2 >> 8) & 0xff] ^ + Td3[(t1 ) & 0xff] ^ + rk[0]; + s1 = + Td0[(t1 >> 24) ] ^ + Td1[(t0 >> 16) & 0xff] ^ + Td2[(t3 >> 8) & 0xff] ^ + Td3[(t2 ) & 0xff] ^ + rk[1]; + s2 = + Td0[(t2 >> 24) ] ^ + Td1[(t1 >> 16) & 0xff] ^ + Td2[(t0 >> 8) & 0xff] ^ + Td3[(t3 ) & 0xff] ^ + rk[2]; + s3 = + Td0[(t3 >> 24) ] ^ + Td1[(t2 >> 16) & 0xff] ^ + Td2[(t1 >> 8) & 0xff] ^ + Td3[(t0 ) & 0xff] ^ + rk[3]; + } +#endif /* ?FULL_UNROLL */ + /* + * apply last round and + * map cipher state to byte array block: + */ + s0 = + (Td4[(t0 >> 24) ] & 0xff000000) ^ + (Td4[(t3 >> 16) & 0xff] & 0x00ff0000) ^ + (Td4[(t2 >> 8) & 0xff] & 0x0000ff00) ^ + (Td4[(t1 ) & 0xff] & 0x000000ff) ^ + rk[0]; + PUTU32(pt , s0); + s1 = + (Td4[(t1 >> 24) ] & 0xff000000) ^ + (Td4[(t0 >> 16) & 0xff] & 0x00ff0000) ^ + (Td4[(t3 >> 8) & 0xff] & 0x0000ff00) ^ + (Td4[(t2 ) & 0xff] & 0x000000ff) ^ + rk[1]; + PUTU32(pt + 4, s1); + s2 = + (Td4[(t2 >> 24) ] & 0xff000000) ^ + (Td4[(t1 >> 16) & 0xff] & 0x00ff0000) ^ + (Td4[(t0 >> 8) & 0xff] & 0x0000ff00) ^ + (Td4[(t3 ) & 0xff] & 0x000000ff) ^ + rk[2]; + PUTU32(pt + 8, s2); + s3 = + (Td4[(t3 >> 24) ] & 0xff000000) ^ + (Td4[(t2 >> 16) & 0xff] & 0x00ff0000) ^ + (Td4[(t1 >> 8) & 0xff] & 0x0000ff00) ^ + (Td4[(t0 ) & 0xff] & 0x000000ff) ^ + rk[3]; + PUTU32(pt + 12, s3); +} + +#ifdef INTERMEDIATE_VALUE_KAT + +void rijndaelEncryptRound(const u32 rk[/*4*(Nr + 1)*/], int Nr, u8 block[16], int rounds) { + int r; + u32 s0, s1, s2, s3, t0, t1, t2, t3; + + /* + * map byte array block to cipher state + * and add initial round key: + */ + s0 = GETU32(block ) ^ rk[0]; + s1 = GETU32(block + 4) ^ rk[1]; + s2 = GETU32(block + 8) ^ rk[2]; + s3 = GETU32(block + 12) ^ rk[3]; + rk += 4; + + /* + * Nr - 1 full rounds: + */ + for (r = (rounds < Nr ? rounds : Nr - 1); r > 0; r--) { + t0 = + Te0[(s0 >> 24) ] ^ + Te1[(s1 >> 16) & 0xff] ^ + Te2[(s2 >> 8) & 0xff] ^ + Te3[(s3 ) & 0xff] ^ + rk[0]; + t1 = + Te0[(s1 >> 24) ] ^ + Te1[(s2 >> 16) & 0xff] ^ + Te2[(s3 >> 8) & 0xff] ^ + Te3[(s0 ) & 0xff] ^ + rk[1]; + t2 = + Te0[(s2 >> 24) ] ^ + Te1[(s3 >> 16) & 0xff] ^ + Te2[(s0 >> 8) & 0xff] ^ + Te3[(s1 ) & 0xff] ^ + rk[2]; + t3 = + Te0[(s3 >> 24) ] ^ + Te1[(s0 >> 16) & 0xff] ^ + Te2[(s1 >> 8) & 0xff] ^ + Te3[(s2 ) & 0xff] ^ + rk[3]; + + s0 = t0; + s1 = t1; + s2 = t2; + s3 = t3; + rk += 4; + + } + + /* + * apply last round and + * map cipher state to byte array block: + */ + if (rounds == Nr) { + t0 = + (Te4[(s0 >> 24) ] & 0xff000000) ^ + (Te4[(s1 >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(s2 >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(s3 ) & 0xff] & 0x000000ff) ^ + rk[0]; + t1 = + (Te4[(s1 >> 24) ] & 0xff000000) ^ + (Te4[(s2 >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(s3 >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(s0 ) & 0xff] & 0x000000ff) ^ + rk[1]; + t2 = + (Te4[(s2 >> 24) ] & 0xff000000) ^ + (Te4[(s3 >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(s0 >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(s1 ) & 0xff] & 0x000000ff) ^ + rk[2]; + t3 = + (Te4[(s3 >> 24) ] & 0xff000000) ^ + (Te4[(s0 >> 16) & 0xff] & 0x00ff0000) ^ + (Te4[(s1 >> 8) & 0xff] & 0x0000ff00) ^ + (Te4[(s2 ) & 0xff] & 0x000000ff) ^ + rk[3]; + + s0 = t0; + s1 = t1; + s2 = t2; + s3 = t3; + } + + PUTU32(block , s0); + PUTU32(block + 4, s1); + PUTU32(block + 8, s2); + PUTU32(block + 12, s3); +} + +void rijndaelDecryptRound(const u32 rk[/*4*(Nr + 1)*/], int Nr, u8 block[16], int rounds) { + int r; + u32 s0, s1, s2, s3, t0, t1, t2, t3; + + /* + * map byte array block to cipher state + * and add initial round key: + */ + s0 = GETU32(block ) ^ rk[0]; + s1 = GETU32(block + 4) ^ rk[1]; + s2 = GETU32(block + 8) ^ rk[2]; + s3 = GETU32(block + 12) ^ rk[3]; + rk += 4; + + /* + * Nr - 1 full rounds: + */ + for (r = (rounds < Nr ? rounds : Nr) - 1; r > 0; r--) { + t0 = + Td0[(s0 >> 24) ] ^ + Td1[(s3 >> 16) & 0xff] ^ + Td2[(s2 >> 8) & 0xff] ^ + Td3[(s1 ) & 0xff] ^ + rk[0]; + t1 = + Td0[(s1 >> 24) ] ^ + Td1[(s0 >> 16) & 0xff] ^ + Td2[(s3 >> 8) & 0xff] ^ + Td3[(s2 ) & 0xff] ^ + rk[1]; + t2 = + Td0[(s2 >> 24) ] ^ + Td1[(s1 >> 16) & 0xff] ^ + Td2[(s0 >> 8) & 0xff] ^ + Td3[(s3 ) & 0xff] ^ + rk[2]; + t3 = + Td0[(s3 >> 24) ] ^ + Td1[(s2 >> 16) & 0xff] ^ + Td2[(s1 >> 8) & 0xff] ^ + Td3[(s0 ) & 0xff] ^ + rk[3]; + + s0 = t0; + s1 = t1; + s2 = t2; + s3 = t3; + rk += 4; + + } + + /* + * complete the last round and + * map cipher state to byte array block: + */ + t0 = + (Td4[(s0 >> 24) ] & 0xff000000) ^ + (Td4[(s3 >> 16) & 0xff] & 0x00ff0000) ^ + (Td4[(s2 >> 8) & 0xff] & 0x0000ff00) ^ + (Td4[(s1 ) & 0xff] & 0x000000ff); + t1 = + (Td4[(s1 >> 24) ] & 0xff000000) ^ + (Td4[(s0 >> 16) & 0xff] & 0x00ff0000) ^ + (Td4[(s3 >> 8) & 0xff] & 0x0000ff00) ^ + (Td4[(s2 ) & 0xff] & 0x000000ff); + t2 = + (Td4[(s2 >> 24) ] & 0xff000000) ^ + (Td4[(s1 >> 16) & 0xff] & 0x00ff0000) ^ + (Td4[(s0 >> 8) & 0xff] & 0x0000ff00) ^ + (Td4[(s3 ) & 0xff] & 0x000000ff); + t3 = + (Td4[(s3 >> 24) ] & 0xff000000) ^ + (Td4[(s2 >> 16) & 0xff] & 0x00ff0000) ^ + (Td4[(s1 >> 8) & 0xff] & 0x0000ff00) ^ + (Td4[(s0 ) & 0xff] & 0x000000ff); + + if (rounds == Nr) { + t0 ^= rk[0]; + t1 ^= rk[1]; + t2 ^= rk[2]; + t3 ^= rk[3]; + } + + PUTU32(block , t0); + PUTU32(block + 4, t1); + PUTU32(block + 8, t2); + PUTU32(block + 12, t3); +} + +#endif /* INTERMEDIATE_VALUE_KAT */ diff --git a/lib/rijndael-alg-fst.h b/lib/rijndael-alg-fst.h new file mode 100644 index 0000000..2c3396e --- /dev/null +++ b/lib/rijndael-alg-fst.h @@ -0,0 +1,47 @@ +/** + * rijndael-alg-fst.h + * + * @version 3.0 (December 2000) + * + * Optimised ANSI C code for the Rijndael cipher (now AES) + * + * @author Vincent Rijmen + * @author Antoon Bosselaers + * @author Paulo Barreto + * + * This code is hereby placed in the public domain. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHORS ''AS IS'' AND ANY EXPRESS + * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED + * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR + * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, + * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE + * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, + * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ +#ifndef __RIJNDAEL_ALG_FST_H +#define __RIJNDAEL_ALG_FST_H + +#define MAXKC (256/32) +#define MAXKB (256/8) +#define MAXNR 14 + +typedef unsigned char u8; +typedef unsigned short u16; +typedef unsigned int u32; + +int rijndaelKeySetupEnc(u32 rk[/*4*(Nr + 1)*/], const u8 cipherKey[], int keyBits); +int rijndaelKeySetupDec(u32 rk[/*4*(Nr + 1)*/], const u8 cipherKey[], int keyBits); +void rijndaelEncrypt(const u32 rk[/*4*(Nr + 1)*/], int Nr, const u8 pt[16], u8 ct[16]); +void rijndaelDecrypt(const u32 rk[/*4*(Nr + 1)*/], int Nr, const u8 ct[16], u8 pt[16]); + +#ifdef INTERMEDIATE_VALUE_KAT +void rijndaelEncryptRound(const u32 rk[/*4*(Nr + 1)*/], int Nr, u8 block[16], int rounds); +void rijndaelDecryptRound(const u32 rk[/*4*(Nr + 1)*/], int Nr, u8 block[16], int rounds); +#endif /* INTERMEDIATE_VALUE_KAT */ + +#endif /* __RIJNDAEL_ALG_FST_H */ diff --git a/src/Makefile b/src/Makefile new file mode 100644 index 0000000..f602975 --- /dev/null +++ b/src/Makefile @@ -0,0 +1,66 @@ +CC=gcc + +CFLAGS=-Wall -pedantic -Wextra -std=gnu99 -I../lib -I./ -Ioutput_modules -O2 -g +LDFLAGS=-g -pthread +LDLIBS= -lpcap -lgmp -lm +TARGETS=zmap +VPATH=../lib:output_modules:probe_modules +PREFIX=/usr/local + +INSTALL=install +INSTALLDATA=install -m 644 +mandir=/usr/share/man/man1/ +bindir=$(PREFIX)/sbin + +# Hardening and warnings for building with gcc +#M aybe add -Wstrict-prototypes -Wmissing-prototypes -Wmissing-declarations +#-Wold-style-definition -Wswitch-enum +GCCWARNINGS = -Wall -fno-strict-aliasing -W -Wfloat-equal -Wundef \ +-Wpointer-arith \ +-Wwrite-strings -Wredundant-decls -Wchar-subscripts -Wcomment \ +-Wformat=2 -Wwrite-strings -Wredundant-decls -Wno-unused-but-set-variable \ +-Wnested-externs -Wbad-function-cast -Winit-self \ +-Wmissing-field-initializers \ +-Waddress -Wmissing-noreturn -Wnormalized=id \ +-Woverride-init -Wstrict-overflow=1 -Wextra -Warray-bounds \ +-Wstack-protector -Wformat -Wformat-security -Wpointer-sign -Wno-format-nonliteral -Wno-format-y2k +GCCHARDENING=-D_FORTIFY_SOURCE=2 -fstack-protector-all -fwrapv -fPIC --param ssp-buffer-size=1 +LDHARDENING=-z relro -z now + +EXTRACFLAGS=-g -O2 $(EXTRA_CFLAGS) $(GCCHARDENING) $(GCCWARNINGS) -Werror +EXTRALDFLAGS= $(LDHARDENING) + +CFLAGS+=-Wall $(INCLUDE) $(EXTRACFLAGS) +LDFLAGS+=$(EXTRALDFLAGS) + +modules=module_tcp_synscan.o module_icmp_echo.o module_udp.o #ADD YOUR MODULE HERE + +objects=constraint.o blacklist.o cyclic.o logger.o send.o recv.o state.o monitor.o zopt.o zmap.o random.o output_modules.o module_simple_file.o module_extended_file.o packet.o probe_modules.o ${modules} validate.o rijndael-alg-fst.o get_gateway.o aesrand.o + +ifeq ($(REDIS), true) + LDLIBS+=-lhiredis + objects+=module_redis.o redis.o module_ssldb.o + CFLAGS+=-DREDIS +endif + +all: $(TARGETS) + +$(TARGETS): + $(CC) $(CFLAGS) $(DFLAGS) $(LDFLAGS) $^ -o $@ $(LDLIBS) + +zmap: $(objects) + +zopt.c zopt.h: zopt.ggo + gengetopt -C --no-help --no-version -i $^ -F $* + +install: zmap + $(INSTALL) zmap $(bindir)/zmap + test -d /etc/zmap || (mkdir /etc/zmap && $(INSTALLDATA) ../conf/* /etc/zmap/) + $(INSTALLDATA) ./zmap.1 $(mandir) + echo "\n\n\n\n**************\nSuccess! ZMap is installed. Try running (as root):\nzmap -p 80 -N 10 -B 1M -o -\n**************" + +clean: + -rm -f $(objects) $(TARGETS) + +.PHONY: install clean + diff --git a/src/aesrand.c b/src/aesrand.c new file mode 100644 index 0000000..c8e8c28 --- /dev/null +++ b/src/aesrand.c @@ -0,0 +1,57 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include + +#include "../lib/rijndael-alg-fst.h" +#include "../lib/random.h" +#include "../lib/logger.h" + +#define AES_ROUNDS 10 +#define AES_BLOCK_WORDS 4 +#define AES_KEY_BYTES 16 +#define OUTPUT_BYTES 16 + +static uint32_t aes_input[AES_BLOCK_WORDS]; +static uint32_t aes_sched[(AES_ROUNDS+1)*4]; +static uint8_t aes_output[OUTPUT_BYTES]; +static int init = 0; + +void aesrand_init(uint32_t seed) +{ + memset(&aes_input, 0, sizeof(aes_input)); + uint8_t key[AES_KEY_BYTES]; + if (seed) { + memset(key, 0, AES_KEY_BYTES*sizeof(uint8_t)); + memcpy(key, &seed, sizeof(uint32_t)); + } else { + if (!random_bytes(key, AES_KEY_BYTES)) { + log_fatal("aesrand", "couldn't get random bytes"); + } + } + if (rijndaelKeySetupEnc(aes_sched, key, AES_KEY_BYTES*8) != AES_ROUNDS) { + log_fatal("aesrand", "could not initialize AES key"); + } + memset(aes_output, 0, OUTPUT_BYTES*sizeof(uint8_t)); + init = 1; +} + +uint64_t aesrand_getword(void) +{ + assert(init); + memcpy(aes_input, aes_output, sizeof(aes_input)); + rijndaelEncrypt(aes_sched, AES_ROUNDS, + (uint8_t *)aes_input, aes_output); + uint64_t retval; + memcpy(&retval, aes_output, sizeof(retval)); + return retval; +} + diff --git a/src/aesrand.h b/src/aesrand.h new file mode 100644 index 0000000..2d3dedb --- /dev/null +++ b/src/aesrand.h @@ -0,0 +1,18 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include + +#ifndef _AESRAND_H +#define _AESRAND_H + +void aesrand_init(uint32_t seed); + +uint64_t aesrand_getword(void); + +#endif diff --git a/src/cyclic.c b/src/cyclic.c new file mode 100644 index 0000000..a42c7f4 --- /dev/null +++ b/src/cyclic.c @@ -0,0 +1,184 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +/* + * cyclic provides an inexpensive approach to iterating over the IPv4 address + * space in a random(-ish) manner such that we connect to every host once in + * a scan execution without having to keep track of the IPs that have been + * scanned or need to be scanned and such that each scan has a different + * ordering. We accomplish this by utilizing a cyclic multiplicative group + * of integers modulo a prime and generating a new primitive root (generator) + * for each scan. + * + * We know that 3 is a generator of (Z mod 2^32 + 15 - {0}, *) + * and that we have coverage over the entire address space because 2**32 + 15 + * is prime and ||(Z mod PRIME - {0}, *)|| == PRIME - 1. Therefore, we + * just need to find a new generator (primitive root) of the cyclic group for + * each scan that we perform. + * + * Because generators map to generators over an isomorphism, we can efficiently + * find random primitive roots of our mult. group by finding random generators + * of the group (Zp-1, +) which is isomorphic to (Zp*, *). Specifically the + * generators of (Zp-1, +) are { s | (s, p-1) == 1 } which implies that + * the generators of (Zp*, *) are { d^s | (s, p-1) == 1 }. where d is a known + * generator of the multiplicative group. We efficiently find + * generators of the additive group by precalculating the psub1_f of + * p - 1 and randomly checking random numbers against the psub1_f until + * we find one that is coprime and map it into Zp*. Because + * totient(totient(p)) ~= 10^9, this should take relatively few + * iterations to find a new generator. + */ + +#include "cyclic.h" + +#include +#include +#include +#include +#include +#include +#include + +#include +#include +#include + +#include + +#include "../lib/logger.h" +#include "../lib/blacklist.h" + +#include "state.h" +#include "aesrand.h" + +#define LSRC "cyclic" +#define PRIME 4294967311 // 2^32 + 15 +#define KNOWN_PRIMROOT 3 + +// distinct prime factors of 2^32 + 15 +static const uint64_t psub1_f[] = { 2, 3, 5, 131, 364289 }; + +// selected primitive root that we'll use as the generator +static uint64_t primroot = 0; +static uint64_t current = 0; + +#define COPRIME 1 +#define NOT_COPRIME 0 + +// check whether two integers are coprime +static int check_coprime(uint64_t check) +{ + for (unsigned i=0; i < sizeof(psub1_f)/sizeof(psub1_f[0]); i++) { + if (psub1_f[i] > check && !(psub1_f[i] % check)) { + return NOT_COPRIME; + } else if (psub1_f[i] < check && !(check % psub1_f[i])) { + return NOT_COPRIME; + } else if (psub1_f[i] == check) { + return NOT_COPRIME; + } + } + return COPRIME; +} + +// find gen of cyclic group Z modulo PRIME +static uint64_t find_primroot(void) +{ + // what luck, rand() returns a uint32_t! + uint32_t candidate = (uint32_t) aesrand_getword() & 0xFFFF; + while(check_coprime(candidate) != COPRIME) { + ++candidate; + } + // pre-modded result is gigantic so use GMP + mpz_t base, power, prime, primroot; + mpz_init_set_d(base, (double) KNOWN_PRIMROOT); + mpz_init_set_d(power, (double) candidate); + mpz_init_set_d(prime, (double) PRIME); + mpz_init(primroot); + mpz_powm(primroot, base, power, prime); + uint64_t retv = (uint64_t) mpz_get_ui(primroot); + mpz_clear(base); + mpz_clear(power); + mpz_clear(prime); + mpz_clear(primroot); + return retv; +} + +int cyclic_init(uint32_t primroot_, uint32_t current_) +{ + assert(!(!primroot_ && current_)); + + if (zconf.use_seed) { + aesrand_init(zconf.seed+1); + } else { + aesrand_init(0); + } + if (!primroot_) { + do { + primroot = find_primroot(); + } while (primroot >= (1LL << 32)); + log_debug(LSRC, "primitive root: %lld", primroot); + current = (uint32_t) aesrand_getword() & 0xFFFF; + log_debug(LSRC, "starting point: %lld", current); + } else { + primroot = primroot_; + log_debug(LSRC, "primitive root %lld specified by caller", + primroot); + if (!current_) { + current = (uint32_t) aesrand_getword() & 0xFFFF; + log_debug(LSRC, "no cyclic starting point, " + "selected random startpoint: %lld", + current); + } else { + current = current_; + log_debug(LSRC, "starting point %lld specified by caller", + current); + } + } + zconf.generator = primroot; + if (blacklist_init_from_files(zconf.whitelist_filename, + zconf.blacklist_filename)) { + return -1; + } + + // make sure current is an allowed ip + cyclic_get_next_ip(); + + return 0; +} + +uint32_t cyclic_get_curr_ip(void) +{ + return (uint32_t) current; +} + +uint32_t cyclic_get_primroot(void) +{ + return (uint32_t) primroot; +} + +static inline uint32_t cyclic_get_next_elem(void) +{ + do { + current *= primroot; + current %= PRIME; + } while (current >= (1LL << 32)); + return (uint32_t) current; +} + +uint32_t cyclic_get_next_ip(void) +{ + while (1) { + uint32_t candidate = cyclic_get_next_elem(); + if (!blacklist_is_allowed(candidate)) { + zsend.blacklisted++; + } else { + return candidate; + } + } +} + diff --git a/src/cyclic.h b/src/cyclic.h new file mode 100644 index 0000000..f7bb7a2 --- /dev/null +++ b/src/cyclic.h @@ -0,0 +1,25 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include + +#ifndef _CYCLIC_H +#define _CYCLIC_H + +int cyclic_init(uint32_t, uint32_t); + +// get next IP address to scan +uint32_t cyclic_get_next_ip(void); + +// what IP address was returned last +uint32_t cyclic_get_curr_ip(void); + +// what primitive root was generated for this current scan +uint32_t cyclic_get_primroot(void); + +#endif diff --git a/src/get_gateway.c b/src/get_gateway.c new file mode 100644 index 0000000..26f6404 --- /dev/null +++ b/src/get_gateway.c @@ -0,0 +1,237 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "../lib/logger.h" + +int read_nl_sock(int sock, char *buf, int buf_len) +{ + int msg_len = 0; + char *pbuf = buf; + do { + int len = recv(sock, pbuf, buf_len - msg_len, 0); + if (len <= 0) { + log_debug("get-gw", "recv failed: %s", strerror(errno)); + return -1; + } + struct nlmsghdr *nlhdr = (struct nlmsghdr *)pbuf; + if (NLMSG_OK(nlhdr, ((unsigned int)len)) == 0 || + nlhdr->nlmsg_type == NLMSG_ERROR) { + log_debug("get-gw", "recv failed: %s", strerror(errno)); + return -1; + } + if (nlhdr->nlmsg_type == NLMSG_DONE) { + break; + } else { + msg_len += len; + pbuf += len; + } + if ((nlhdr->nlmsg_flags & NLM_F_MULTI) == 0) { + break; + } + } while (1); + return msg_len; +} + +int send_nl_req(uint16_t msg_type, uint32_t seq, + void *payload, uint32_t payload_len) +{ + int sock = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE); + if (sock < 0) { + log_error("get-gw", "unable to get socket: %s", strerror(errno)); + return -1; + } + if (NLMSG_SPACE(payload_len) < payload_len) { + // Integer overflow + return -1; + } + struct nlmsghdr *nlmsg; + nlmsg = malloc(NLMSG_SPACE(payload_len)); + if (!nlmsg) { + return -1; + } + + memset(nlmsg, 0, sizeof(nlmsg)); + memcpy(NLMSG_DATA(nlmsg), payload, payload_len); + nlmsg->nlmsg_type = msg_type; + nlmsg->nlmsg_len = NLMSG_LENGTH(payload_len); + nlmsg->nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST; + nlmsg->nlmsg_seq = seq; + nlmsg->nlmsg_pid = getpid(); + + if (send(sock, nlmsg, nlmsg->nlmsg_len, 0) < 0) { + log_error("get-gw", "failure sending: %s", strerror(errno)); + return -1; + } + free(nlmsg); + return sock; +} + +int get_hw_addr(struct in_addr *gw_ip, char *iface, unsigned char *hw_mac) +{ + char buf[8192]; + struct ndmsg req; + struct nlmsghdr *nlhdr; + + if (!gw_ip || !hw_mac) { + return -1; + } + // Send RTM_GETNEIGH request + req.ndm_family = AF_INET; + req.ndm_ifindex = if_nametoindex(iface); + req.ndm_state = NUD_REACHABLE; + req.ndm_type = NDA_LLADDR; + + int sock = send_nl_req(RTM_GETNEIGH, 1, &req, sizeof(req)); + + // Read responses + unsigned nl_len = read_nl_sock(sock, buf, sizeof(buf)); + if (nl_len <= 0) { + return -1; + } + // Parse responses + nlhdr = (struct nlmsghdr *)buf; + while (NLMSG_OK(nlhdr, nl_len)) { + struct rtattr *rt_attr; + struct rtmsg *rt_msg; + int rt_len; + unsigned char mac[6]; + struct in_addr dst_ip; + int correct_ip = 0; + + rt_msg = (struct rtmsg *) NLMSG_DATA(nlhdr); + + if ((rt_msg->rtm_family != AF_INET)) { + return -1; + } + + rt_attr = (struct rtattr *) RTM_RTA(rt_msg); + rt_len = RTM_PAYLOAD(nlhdr); + while (RTA_OK(rt_attr, rt_len)) { + switch (rt_attr->rta_type) { + case NDA_LLADDR: + assert(RTA_PAYLOAD(rt_attr) == IFHWADDRLEN); + memcpy(mac, RTA_DATA(rt_attr), IFHWADDRLEN); + break; + case NDA_DST: + assert(RTA_PAYLOAD(rt_attr) == sizeof(dst_ip)); + memcpy(&dst_ip, RTA_DATA(rt_attr), sizeof(dst_ip)); + if (memcmp(&dst_ip, gw_ip, sizeof(dst_ip)) == 0) { + correct_ip = 1; + } + break; + } + rt_attr = RTA_NEXT(rt_attr, rt_len); + } + if (correct_ip) { + memcpy(hw_mac, mac, IFHWADDRLEN); + return 0; + } + nlhdr = NLMSG_NEXT(nlhdr, nl_len); + } + return -1; +} + +// gw and iface[IF_NAMESIZE] MUST be allocated +int get_default_gw(struct in_addr *gw, char *iface) +{ + struct rtmsg req; + unsigned int nl_len; + char buf[8192]; + struct nlmsghdr *nlhdr; + + if (!gw || !iface) { + return -1; + } + + // Send RTM_GETROUTE request + memset(&req, 0, sizeof(req)); + int sock = send_nl_req(RTM_GETROUTE, 0, &req, sizeof(req)); + + // Read responses + nl_len = read_nl_sock(sock, buf, sizeof(buf)); + if (nl_len <= 0) { + return -1; + } + + // Parse responses + nlhdr = (struct nlmsghdr *)buf; + while (NLMSG_OK(nlhdr, nl_len)) { + struct rtattr *rt_attr; + struct rtmsg *rt_msg; + int rt_len; + int has_gw = 0; + + rt_msg = (struct rtmsg *) NLMSG_DATA(nlhdr); + + if ((rt_msg->rtm_family != AF_INET) || (rt_msg->rtm_table != RT_TABLE_MAIN)) { + return -1; + } + + rt_attr = (struct rtattr *) RTM_RTA(rt_msg); + rt_len = RTM_PAYLOAD(nlhdr); + while (RTA_OK(rt_attr, rt_len)) { + switch (rt_attr->rta_type) { + case RTA_OIF: + if_indextoname(*(int *) RTA_DATA(rt_attr), iface); + break; + case RTA_GATEWAY: + gw->s_addr = *(unsigned int *) RTA_DATA(rt_attr); + has_gw = 1; + break; + } + rt_attr = RTA_NEXT(rt_attr, rt_len); + } + + if (has_gw) { + return 0; + } + nlhdr = NLMSG_NEXT(nlhdr, nl_len); + } + return -1; +} + +// Returns the first IP address for a given iface +int get_iface_ip(char *iface, struct in_addr *ip) +{ + int sock; + struct ifreq ifr; + sock = socket(AF_INET, SOCK_DGRAM, 0); + if (sock < 0) { + log_error("get-gw", "failure opening socket: %s", strerror(errno)); + return -1; + } + ifr.ifr_addr.sa_family = AF_INET; + strncpy(ifr.ifr_name, iface, IFNAMSIZ-1); + + if (ioctl(sock, SIOCGIFADDR, &ifr) < 0) { + log_error("get-gw", "ioctl failure: %s", strerror(errno)); + close(sock); + return -1; + } + close(sock); + memcpy(ip, &((struct sockaddr_in*)&ifr.ifr_addr)->sin_addr, sizeof(*ip)); + return 0; +} + diff --git a/src/get_gateway.h b/src/get_gateway.h new file mode 100644 index 0000000..398e3e1 --- /dev/null +++ b/src/get_gateway.h @@ -0,0 +1,18 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#ifndef _GET_GATEWAY_H +#define _GET_GATEWAY_H + +#include + +int get_hw_addr(struct in_addr *gw_ip, char *iface, unsigned char *hw_mac); +int get_default_gw(struct in_addr *gw, char *iface); +int get_iface_ip(char *iface, struct in_addr *ip); + +#endif diff --git a/src/monitor.c b/src/monitor.c new file mode 100644 index 0000000..273329b --- /dev/null +++ b/src/monitor.c @@ -0,0 +1,231 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +// module responsible for printing on-screen updates during the scan process + +#include +#include +#include +#include + +#include "recv.h" +#include "monitor.h" +#include "state.h" + +#include "../lib/logger.h" + +#define UPDATE_INTERVAL 1 //seconds + +static double last_now = 0.0; +static uint32_t last_sent = 0; +static uint32_t last_rcvd = 0; +static uint32_t last_drop = 0; +static uint32_t last_failures = 0; + +static double min_d(double array[], int n) +{ + double value=INFINITY; + for (int i=0; i 0) { + snprintf(buf, len, "%d years", y); + } else if (d > 9) { + snprintf(buf, len, "%dd", d); + } else if (d > 0) { + snprintf(buf, len, "%dd%02dh", d, h); + } else if (h > 9) { + snprintf(buf, len, "%dh", h); + } else if (h > 0) { + snprintf(buf, len, "%dh%02dm", h, m); + } else if (m > 9) { + snprintf(buf, len, "%dm", m); + } else if (m > 0) { + snprintf(buf, len, "%dm%02ds", m, s); + } else { + snprintf(buf, len, "%ds", s); + } + } else { + if (d > 0) { + snprintf(buf, len, "%dd%d:%02d:%02d", d, h, m, s); + } else if (h > 0) { + snprintf(buf, len, "%d:%02d:%02d", h, m, s); + } else { + snprintf(buf, len, "%d:%02d", m, s); + } + } +} + +// pretty print quantities +static void number_string(uint32_t n, char *buf, size_t len) +{ + int figs = 0; + if (n < 1000) { + snprintf(buf, len, "%u ", n); + } else if (n < 1000000) { + if (n < 10000) { + figs = 2; + } else if (n < 100000) { + figs = 1; + } + snprintf(buf, len, "%0.*f K", figs, (float)n/1000.); + } else { + if (figs < 10000000) { + figs = 2; + } else if (figs < 100000000) { + figs = 1; + } + snprintf(buf, len, "%0.*f M", figs, (float)n/1000000.); + } +} + +// estimate time remaining time based on config and state +double compute_remaining_time(double age) +{ + if (!zsend.complete) { + double remaining[] = {INFINITY, INFINITY, INFINITY}; + if (zsend.targets) { + double done = (double)zsend.sent/zsend.targets; + remaining[0] = (1. - done)*(age/done) + zconf.cooldown_secs; + } + if (zconf.max_runtime) { + remaining[1] = (zconf.max_runtime - age)+zconf.cooldown_secs; + } + if (zconf.max_results) { + double done = (double)zrecv.success_unique/zconf.max_results; + remaining[2] = (1. - done)*(age/done); + } + return min_d(remaining, sizeof(remaining)/sizeof(double)); + } else { + return zconf.cooldown_secs - (now() - zsend.finish); + } +} + +static void monitor_update(void) +{ + if (last_now > 0.0) { + double age = now() - zsend.start; + double delta = now() - last_now; + double remaining_secs = compute_remaining_time(age); + double percent_complete = 100.*age/(age + remaining_secs); + + // ask pcap for fresh values + recv_update_pcap_stats(); + + // format times for display + char time_left[20]; + if (age < 5) { + time_left[0] = '\0'; + } else { + char buf[20]; + time_string((int)remaining_secs, 1, buf, sizeof(buf)); + snprintf(time_left, sizeof(time_left), " (%s left)", buf); + } + char time_past[20]; + time_string((int)age, 0, time_past, sizeof(time_past)); + + char send_rate[20], send_avg[20], + recv_rate[20], recv_avg[20], + pcap_drop[20], pcap_drop_avg[20]; + // recv stats + number_string((zrecv.success_unique - last_rcvd)/delta, + recv_rate, sizeof(recv_rate)); + number_string((zrecv.success_unique/age), recv_avg, sizeof(recv_avg)); + // dropped stats + number_string((zrecv.pcap_drop + zrecv.pcap_ifdrop - last_drop)/delta, + pcap_drop, sizeof(pcap_drop)); + number_string(((zrecv.pcap_drop + zrecv.pcap_ifdrop)/age), + pcap_drop_avg, sizeof(pcap_drop_avg)); + + // Warn if we drop > 5% of our average receive rate + uint32_t drop_rate = (uint32_t)((zrecv.pcap_drop + zrecv.pcap_ifdrop - last_drop) / delta); + if (drop_rate > (uint32_t)((zrecv.success_unique - last_rcvd) / delta) / 20) { + log_warn("monitor", "Dropped %d packets in the last second, (%d total dropped (pcap: %d + iface: %d))", + drop_rate, zrecv.pcap_drop + zrecv.pcap_ifdrop, zrecv.pcap_drop, zrecv.pcap_ifdrop); + } + + // Warn if we fail to send > 1% of our average send rate + uint32_t fail_rate = (uint32_t)((zsend.sendto_failures - last_failures) / delta); // failures/sec + if (fail_rate > ((zsend.sent / age) / 100)) { + log_warn("monitor", "Failed to send %d packets/sec (%d total failures)", + fail_rate, zsend.sendto_failures); + } + + if (!zsend.complete) { + // main display (during sending) + number_string((zsend.sent - last_sent)/delta, + send_rate, sizeof(send_rate)); + number_string((zsend.sent/age), send_avg, sizeof(send_avg)); + fprintf(stderr, + "%5s %0.0f%%%s; send: %u %sp/s (%sp/s avg); " + "recv: %u %sp/s (%sp/s avg); " + "drops: %sp/s (%sp/s avg); " + "hits: %0.2f%%\n", + time_past, + percent_complete, + time_left, + zsend.sent, + send_rate, + send_avg, + zrecv.success_unique, + recv_rate, + recv_avg, + pcap_drop, + pcap_drop_avg, + zrecv.success_unique*100./zsend.sent); + } else { + // alternate display (during cooldown) + number_string((zsend.sent/(zsend.finish - zsend.start)), send_avg, sizeof(send_avg)); + fprintf(stderr, + "%5s %0.0f%%%s; send: %u done (%sp/s avg); " + "recv: %u %sp/s (%sp/s avg); " + "drops: %sp/s (%sp/s avg); " + "hits: %0.2f%%\n", + time_past, + percent_complete, + time_left, + zsend.sent, + send_avg, + zrecv.success_unique, + recv_rate, + recv_avg, + pcap_drop, + pcap_drop_avg, + zrecv.success_unique*100./zsend.sent); + } + } + last_now = now(); + last_sent = zsend.sent; + last_rcvd = zrecv.success_unique; + last_drop = zrecv.pcap_drop + zrecv.pcap_ifdrop; + last_failures = zsend.sendto_failures; +} + +void monitor_run(void) +{ + while (!(zsend.complete && zrecv.complete)) { + monitor_update(); + sleep(UPDATE_INTERVAL); + } +} + diff --git a/src/monitor.h b/src/monitor.h new file mode 100644 index 0000000..c8bd9fb --- /dev/null +++ b/src/monitor.h @@ -0,0 +1,14 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#ifndef _MONITOR_H +#define _MONITOR_H + +void monitor_run(); + +#endif diff --git a/src/output_modules/module_extended_file.c b/src/output_modules/module_extended_file.c new file mode 100644 index 0000000..477bd67 --- /dev/null +++ b/src/output_modules/module_extended_file.c @@ -0,0 +1,109 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include +#include + +#include +#include +#include +#include + +#include "../../lib/logger.h" + +#include "output_modules.h" + +static FILE *file = NULL; +#define UNUSED __attribute__((unused)) + +int extendedfile_init(struct state_conf *conf) +{ + assert(conf); + if (conf->output_filename) { + if (!strcmp(conf->output_filename, "-")) { + file = stdout; + } else { + if (!(file = fopen(conf->output_filename, "w"))) { + perror("Couldn't open output file"); + exit(EXIT_FAILURE); + } + } + fprintf(file, "response, saddr, daddr, sport, " + "dport, seq, ack, in_cooldown, is_repeat, timestamp\n"); + } + return EXIT_SUCCESS; +} + +static void fprint_tv(FILE *f, struct timeval *tv) +{ + char time_string[40]; + struct tm *ptm = localtime(&tv->tv_sec); + strftime(time_string, sizeof (time_string), + "%Y-%m-%d %H:%M:%S", ptm); + long milliseconds = tv->tv_usec / 1000; + fprintf(f, "%s.%03ld\n", time_string, milliseconds); +} + + +int extendedfile_ip(ipaddr_n_t saddr, ipaddr_n_t daddr, + const char *response_type, int is_repeat, + int in_cooldown, const u_char *packet, size_t buflen) +{ + struct iphdr *ip_hdr = (struct iphdr *)&packet[sizeof(struct ethhdr)]; + if (buflen < (sizeof(struct ethhdr) + ip_hdr->ihl*4 + sizeof(struct tcphdr))) + return EXIT_FAILURE; + struct tcphdr *tcp = (struct tcphdr *)((char *)ip_hdr + ip_hdr->ihl * 4); + + if (file) { + struct in_addr addr; + addr.s_addr = saddr; + // inet_ntoa returns a <> char * + fprintf(file, "%s, %s, ", + response_type, + inet_ntoa(addr)); + addr.s_addr = daddr; + fprintf(file, "%s, %u, %u, %u, %u, %i, %i,", + inet_ntoa(addr), + ntohs(tcp->source), + ntohs(tcp->dest), + ntohl(tcp->seq), + ntohl(tcp->ack_seq), + in_cooldown, + is_repeat); + struct timeval t; + gettimeofday(&t, NULL); + fprint_tv(file, &t); + fflush(file); + } + return EXIT_SUCCESS; +} + +int extendedfile_close(UNUSED struct state_conf* c, + UNUSED struct state_send* s, UNUSED struct state_recv* r) +{ + if (file) { + fflush(file); + fclose(file); + } + return EXIT_SUCCESS; +} + +output_module_t module_extended_file = { + .name = "extended_file", + .init = &extendedfile_init, + .start = NULL, + .update = NULL, + .update_interval = 0, + .close = &extendedfile_close, + .success_ip = &extendedfile_ip, + .other_ip = &extendedfile_ip +}; + diff --git a/src/output_modules/module_extended_file.h b/src/output_modules/module_extended_file.h new file mode 100644 index 0000000..d8b6357 --- /dev/null +++ b/src/output_modules/module_extended_file.h @@ -0,0 +1,20 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include + +int extendedfile_init(struct state_conf *conf); + +int extendedfile_ip(ipaddr_n_t saddr, ipaddr_n_t daddr, + port_n_t sport, port_n_t dport, struct timeval* t, + const char *response_type, int is_repeat, + int in_cooldown, const u_char *packet); + +int extendedfile_close(struct state_conf* c, struct state_send* s, + struct state_recv* r); + diff --git a/src/output_modules/module_redis.c b/src/output_modules/module_redis.c new file mode 100644 index 0000000..ae75045 --- /dev/null +++ b/src/output_modules/module_redis.c @@ -0,0 +1,96 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include + +#include +#include +#include + +#include "../../lib/zdlibc/logger.h" +#include "../../lib/zdlibc/redis.h" + +#include "output_modules.h" + +#define UNUSED __attribute__((unused)) + +typedef struct scannable_t { + in_addr_t ip_address; + uint8_t source; +} scannable_t; + +#define QUEUE_NAME "zmap_results" +#define BUFFER_SIZE 500 +#define SOURCE_ZMAP 0 + +static scannable_t* buffer; +static int buffer_fill = 0; + +int redismodule_init(UNUSED struct state_conf *conf) +{ + buffer = calloc(BUFFER_SIZE, sizeof(scannable_t)); + assert(buffer); + buffer_fill = 0; + return redis_init(); +} + +int redismodule_flush(void) +{ + if (redis_lpush(QUEUE_NAME, buffer, + buffer_fill, sizeof(scannable_t))) { + return EXIT_FAILURE; + } + buffer_fill = 0; + return EXIT_SUCCESS; +} + +int redismodule_newip(ipaddr_n_t saddr, UNUSED ipaddr_n_t daddr, + UNUSED port_n_t sport, UNUSED port_n_t dport, + UNUSED const char *response_type, int is_repeat, + UNUSED int in_cooldown, UNUSED const u_char *packet) +{ + if (!is_repeat) { + buffer[buffer_fill].ip_address = saddr; + buffer[buffer_fill].source = SOURCE_ZMAP; + + if (++buffer_fill == BUFFER_SIZE) { + if (redismodule_flush()) { + return EXIT_FAILURE; + } + } + } + return EXIT_SUCCESS; +} + +int redismodule_close(UNUSED struct state_conf* c, + UNUSED struct state_send* s, + UNUSED struct state_recv* r) +{ + if (redismodule_flush()) { + return EXIT_FAILURE; + } + if (redis_close()) { + return EXIT_FAILURE; + } + return EXIT_SUCCESS; +} + +output_module_t module_redis = { + .name = "redis", + .init = &redismodule_init, + .start = NULL, + .update = NULL, + .update_interval = 0, + .close = &redismodule_close, + .success_ip = &redismodule_newip, + .other_ip = NULL +}; + diff --git a/src/output_modules/module_redis.h b/src/output_modules/module_redis.h new file mode 100644 index 0000000..9f1c234 --- /dev/null +++ b/src/output_modules/module_redis.h @@ -0,0 +1,19 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include + +int redismodule_init(struct state_conf *conf); + +int redismodule_newip(ipaddr_n_t saddr, ipaddr_n_t daddr, + port_n_t sport, port_n_t dport, struct timeval* t, + const char *response_type, int is_repeat, + int in_cooldown, const u_char *packet); + +int redismodule_close(struct state_conf* c, + struct state_send* s, struct state_recv* r); diff --git a/src/output_modules/module_simple_file.c b/src/output_modules/module_simple_file.c new file mode 100644 index 0000000..2061972 --- /dev/null +++ b/src/output_modules/module_simple_file.c @@ -0,0 +1,75 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + + +#include +#include +#include +#include +#include +#include +#include +#include + +#include "output_modules.h" + +#define UNUSED __attribute__((unused)) + +static FILE *file = NULL; + +int simplefile_init(struct state_conf *conf) +{ + assert(conf); + if (conf->output_filename) { + if (!strcmp(conf->output_filename, "-")) { + file = stdout; + } else { + if (!(file = fopen(conf->output_filename, "w"))) { + perror("Couldn't open output file"); + exit(EXIT_FAILURE); + } + } + } + return EXIT_SUCCESS; +} + +int simplefile_synack_newip(ipaddr_n_t saddr, UNUSED ipaddr_n_t daddr, + UNUSED const char *response_type, + int is_repeat, UNUSED int in_cooldown, UNUSED const u_char *packet, + UNUSED size_t buflen) +{ + if (file && !is_repeat) { + struct in_addr addr; + addr.s_addr = saddr; + fprintf(file, "%s\n", inet_ntoa(addr)); + } + fflush(file); + return EXIT_SUCCESS; +} + +int simplefile_close(UNUSED struct state_conf* c, + UNUSED struct state_send* s, + UNUSED struct state_recv* r) +{ + if (file) { + fflush(file); + fclose(file); + } + return EXIT_SUCCESS; +} + +output_module_t module_simple_file = { + .name = "simple_file", + .init = &simplefile_init, + .start = NULL, + .update = NULL, + .update_interval = 0, + .close = &simplefile_close, + .success_ip = &simplefile_synack_newip, + .other_ip = NULL, +}; diff --git a/src/output_modules/output_modules.c b/src/output_modules/output_modules.c new file mode 100644 index 0000000..438902c --- /dev/null +++ b/src/output_modules/output_modules.c @@ -0,0 +1,54 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + + +#include +#include + +#include "output_modules.h" + +extern output_module_t module_simple_file; +extern output_module_t module_extended_file; +// ADD YOUR MODULE HERE + +#ifdef REDIS +extern output_module_t module_redis; +extern output_module_t module_ssldbfeed; +#endif + + +output_module_t* output_modules[] = { + &module_simple_file, + &module_extended_file, +#ifdef REDIS + &module_redis, + &module_ssldbfeed, +#endif + // ADD YOUR MODULE HERE +}; + + + +output_module_t* get_output_module_by_name(const char* name) +{ + int num_modules = (int) (sizeof(output_modules)/sizeof(output_modules[0])); + for (int i=0; i < num_modules; i++) { + if (!strcmp(output_modules[i]->name, name)) { + return output_modules[i]; + } + } + return NULL; +} + +void print_output_modules(void) +{ + int num_modules = (int) (sizeof(output_modules)/sizeof(output_modules[0])); + for (int i=0; i < num_modules; i++) { + printf("%s\n", output_modules[i]->name); + } +} diff --git a/src/output_modules/output_modules.h b/src/output_modules/output_modules.h new file mode 100644 index 0000000..bfda3f4 --- /dev/null +++ b/src/output_modules/output_modules.h @@ -0,0 +1,44 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#ifndef HEADER_OUTPUT_MODULES_H +#define HEADER_OUTPUT_MODULES_H + +#include "../state.h" + +// called at scanner initialization +typedef int (*output_init_cb)(struct state_conf *); + +// called on packet receipt +typedef int (*output_packet_cb)(ipaddr_n_t src_ip, ipaddr_n_t dst_ip, + const char* response_type, + int is_repeat, int in_cooldown, + const u_char* packetbuf, size_t buflen); + +// called periodically during the scan +typedef int (*output_update_cb)(struct state_conf*, struct state_send*, struct state_recv*); + + +typedef struct output_module { + const char *name; + unsigned update_interval; + + output_init_cb init; + output_update_cb start; + output_update_cb update; + output_update_cb close; + output_packet_cb success_ip; + output_packet_cb other_ip; + +} output_module_t; + + +output_module_t* get_output_module_by_name(const char*); +void print_output_modules(void); + +#endif // HEADER_OUTPUT_MODULES_H diff --git a/src/probe_modules/module_icmp_echo.c b/src/probe_modules/module_icmp_echo.c new file mode 100644 index 0000000..4c06763 --- /dev/null +++ b/src/probe_modules/module_icmp_echo.c @@ -0,0 +1,219 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +// probe module for performing ICMP echo request (ping) scans + +#include +#include +#include +#include +#include +#include + +#include +#include +#include + +#include +#include +#include + +#include "probe_modules.h" +#include "packet.h" +#include "validate.h" + +probe_module_t module_icmp_echo; + +int icmp_echo_init_perthread(void* buf, macaddr_t *src, + macaddr_t *gw, __attribute__((unused)) port_h_t dst_port) +{ + memset(buf, 0, MAX_PACKET_SIZE); + + struct ethhdr *eth_header = (struct ethhdr *)buf; + make_eth_header(eth_header, src, gw); + + struct iphdr *ip_header = (struct iphdr*)(ð_header[1]); + uint16_t len = htons(sizeof(struct iphdr) + sizeof(struct icmp) - 8); + make_ip_header(ip_header, IPPROTO_ICMP, len); + + struct icmp *icmp_header = (struct icmp*)(&ip_header[1]); + make_icmp_header(icmp_header); + + return EXIT_SUCCESS; +} + +int icmp_echo_make_packet(void *buf, ipaddr_n_t src_ip, ipaddr_n_t dst_ip, + uint32_t *validation, __attribute__((unused))int probe_num) +{ + struct ethhdr *eth_header = (struct ethhdr *)buf; + struct iphdr *ip_header = (struct iphdr*)(ð_header[1]); + struct icmp *icmp_header = (struct icmp*)(&ip_header[1]); + uint16_t icmp_idnum = validation[2] & 0xFFFF; + + ip_header->saddr = src_ip; + ip_header->daddr = dst_ip; + + icmp_header->icmp_id = icmp_idnum; + icmp_header->icmp_cksum = 0; + icmp_header->icmp_cksum = icmp_checksum((unsigned short *) icmp_header); + + ip_header->check = 0; + ip_header->check = ip_checksum((unsigned short *) ip_header); + + return EXIT_SUCCESS; +} + +void icmp_echo_print_packet(FILE *fp, void* packet) +{ + struct ethhdr *ethh = (struct ethhdr *) packet; + struct iphdr *iph = (struct iphdr *) ðh[1]; + struct icmp *icmp_header = (struct icmp*)(&iph[1]); + + fprintf(fp, "icmp { type: %u | code: %u " + "| checksum: %u | id: %u | seq: %u }\n", + icmp_header->icmp_type, + icmp_header->icmp_code, + ntohs(icmp_header->icmp_cksum), + ntohs(icmp_header->icmp_id), + ntohs(icmp_header->icmp_seq)); + struct in_addr *s = (struct in_addr *) &(iph->saddr); + struct in_addr *d = (struct in_addr *) &(iph->daddr); + char srcip[20]; + char dstip[20]; + // inet_ntoa is a const char * so we if just call it in + // fprintf, you'll get back wrong results since we're + // calling it twice. + strncpy(srcip, inet_ntoa(*s), 19); + strncpy(dstip, inet_ntoa(*d), 19); + fprintf(fp, "ip { saddr: %s | daddr: %s | checksum: %u }\n", + srcip, + dstip, + ntohl(iph->check)); + fprintf(fp, "eth { shost: %02x:%02x:%02x:%02x:%02x:%02x | " + "dhost: %02x:%02x:%02x:%02x:%02x:%02x }\n", + (int) ((unsigned char *) ethh->h_source)[0], + (int) ((unsigned char *) ethh->h_source)[1], + (int) ((unsigned char *) ethh->h_source)[2], + (int) ((unsigned char *) ethh->h_source)[3], + (int) ((unsigned char *) ethh->h_source)[4], + (int) ((unsigned char *) ethh->h_source)[5], + (int) ((unsigned char *) ethh->h_dest)[0], + (int) ((unsigned char *) ethh->h_dest)[1], + (int) ((unsigned char *) ethh->h_dest)[2], + (int) ((unsigned char *) ethh->h_dest)[3], + (int) ((unsigned char *) ethh->h_dest)[4], + (int) ((unsigned char *) ethh->h_dest)[5]); + fprintf(fp, "------------------------------------------------------\n"); +} + +response_type_t* icmp_echo_classify_packet(const u_char *packet, uint32_t len) +{ + (void)len; + struct iphdr *ip_hdr = (struct iphdr *)&packet[sizeof(struct ethhdr)]; + struct icmp *icmp_hdr = (struct icmp*)((char *)ip_hdr + + sizeof(struct iphdr)); + switch (icmp_hdr->icmp_type) { + case ICMP_ECHOREPLY: + return &(module_icmp_echo.responses[0]); + case ICMP_UNREACH: + return &(module_icmp_echo.responses[1]); + case ICMP_SOURCEQUENCH: + return &(module_icmp_echo.responses[2]); + case ICMP_REDIRECT: + return &(module_icmp_echo.responses[3]); + case ICMP_TIMXCEED: + return &(module_icmp_echo.responses[4]); + default: + return &(module_icmp_echo.responses[5]); + } +} + +int icmp_validate_packet(const struct iphdr *ip_hdr, uint32_t len, uint32_t *src_ip, uint32_t *validation) +{ + if (ip_hdr->protocol != IPPROTO_ICMP) { + return 0; + } + + if ((4*ip_hdr->ihl + sizeof(struct icmphdr)) > len) { + // buffer not large enough to contain expected icmp header + return 0; + } + + struct icmphdr *icmp_h = (struct icmphdr*)((char *)ip_hdr + 4*ip_hdr->ihl); + uint16_t icmp_idnum = icmp_h->un.echo.id; + + // ICMP validation is tricky: for some packet types, we must look inside + // the payload + if (icmp_h->type == ICMP_TIME_EXCEEDED || icmp_h->type == ICMP_DEST_UNREACH) { + if ((4*ip_hdr->ihl + sizeof(struct icmphdr) + + sizeof(struct iphdr)) > len) { + return 0; + } + struct iphdr *ip_inner = (struct iphdr *)(icmp_h + 1); + if ((4*ip_hdr->ihl + sizeof(struct icmphdr) + + 4*ip_inner->ihl + sizeof(struct icmphdr)) > len) { + return 0; + } + struct icmphdr *icmp_inner = (struct icmphdr*)((char *)ip_inner + 4 *ip_hdr->ihl); + + // Regenerate validation and icmp id based off inner payload + icmp_idnum = icmp_inner->un.echo.id; + *src_ip = ip_inner->daddr; + validate_gen(ip_hdr->daddr, ip_inner->daddr, (uint8_t *)validation); + } + + // validate icmp id + if (icmp_idnum != (validation[2] & 0xFFFF)) { + return 0; + } + + return 1; +} + +static response_type_t responses[] = { + { + .name = "echoreply", + .is_success = 1 + }, + { + .name = "unreach", + .is_success = 0 + }, + { + .name = "sourcequench", + .is_success = 0 + }, + { + .name = "redirect", + .is_success = 0 + }, + { + .name = "timxceed", + .is_success = 0 + }, + { + .name = "other", + .is_success = 0 + } +}; + +probe_module_t module_icmp_echo = { + .name = "icmp_echoscan", + .packet_length = 62, + .pcap_filter = "icmp and icmp[0]!=8", + .pcap_snaplen = 96, + .port_args = 0, + .thread_initialize = &icmp_echo_init_perthread, + .make_packet = &icmp_echo_make_packet, + .print_packet = &icmp_echo_print_packet, + .classify_packet = &icmp_echo_classify_packet, + .validate_packet = &icmp_validate_packet, + .close = NULL, + .responses = responses +}; + diff --git a/src/probe_modules/module_tcp_synscan.c b/src/probe_modules/module_tcp_synscan.c new file mode 100644 index 0000000..44bce1b --- /dev/null +++ b/src/probe_modules/module_tcp_synscan.c @@ -0,0 +1,196 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +// probe module for performing TCP SYN scans + +#include +#include +#include +#include +#include + +#include +#include +#include + +#include +#include +#include + +#include "probe_modules.h" +#include "packet.h" + +probe_module_t module_tcp_synscan; +uint32_t num_ports = 1; + +int synscan_init_perthread(void* buf, macaddr_t *src, + macaddr_t *gw, port_h_t dst_port) +{ + memset(buf, 0, MAX_PACKET_SIZE); + struct ethhdr *eth_header = (struct ethhdr *)buf; + make_eth_header(eth_header, src, gw); + struct iphdr *ip_header = (struct iphdr*)(ð_header[1]); + uint16_t len = htons(sizeof(struct iphdr) + sizeof(struct tcphdr)); + make_ip_header(ip_header, IPPROTO_TCP, len); + struct tcphdr *tcp_header = (struct tcphdr*)(&ip_header[1]); + make_tcp_header(tcp_header, dst_port); + num_ports = zconf.source_port_last - zconf.source_port_first + 1; + return EXIT_SUCCESS; +} + +int synscan_make_packet(void *buf, ipaddr_n_t src_ip, ipaddr_n_t dst_ip, + uint32_t *validation, int probe_num) +{ + struct ethhdr *eth_header = (struct ethhdr *)buf; + struct iphdr *ip_header = (struct iphdr*)(ð_header[1]); + struct tcphdr *tcp_header = (struct tcphdr*)(&ip_header[1]); + uint16_t src_port = zconf.source_port_first + + ((validation[1] + probe_num) % num_ports); + uint32_t tcp_seq = validation[0]; + + + ip_header->saddr = src_ip; + ip_header->daddr = dst_ip; + + tcp_header->source = htons(src_port); + tcp_header->seq = tcp_seq; + tcp_header->check = 0; + tcp_header->check = tcp_checksum(sizeof(struct tcphdr), + ip_header->saddr, ip_header->daddr, tcp_header); + + ip_header->check = 0; + ip_header->check = ip_checksum((unsigned short *) ip_header); + + return EXIT_SUCCESS; +} + +void synscan_print_packet(FILE *fp, void* packet) +{ + struct ethhdr *ethh = (struct ethhdr *) packet; + struct iphdr *iph = (struct iphdr *) ðh[1]; + struct tcphdr *tcph = (struct tcphdr *) &iph[1]; + fprintf(fp, "tcp { source: %u | dest: %u | seq: %u | checksum: %u }\n", + ntohs(tcph->source), + ntohs(tcph->dest), + ntohl(tcph->seq), + ntohl(tcph->check)); + struct in_addr *s = (struct in_addr *) &(iph->saddr); + struct in_addr *d = (struct in_addr *) &(iph->daddr); + char srcip[20]; + char dstip[20]; + // inet_ntoa is a const char * so we if just call it in + // fprintf, you'll get back wrong results since we're + // calling it twice. + strncpy(srcip, inet_ntoa(*s), 19); + strncpy(dstip, inet_ntoa(*d), 19); + fprintf(fp, "ip { saddr: %s | daddr: %s | checksum: %u }\n", + srcip, + dstip, + ntohl(iph->check)); + fprintf(fp, "eth { shost: %02x:%02x:%02x:%02x:%02x:%02x | " + "dhost: %02x:%02x:%02x:%02x:%02x:%02x }\n", + (int) ((unsigned char *) ethh->h_source)[0], + (int) ((unsigned char *) ethh->h_source)[1], + (int) ((unsigned char *) ethh->h_source)[2], + (int) ((unsigned char *) ethh->h_source)[3], + (int) ((unsigned char *) ethh->h_source)[4], + (int) ((unsigned char *) ethh->h_source)[5], + (int) ((unsigned char *) ethh->h_dest)[0], + (int) ((unsigned char *) ethh->h_dest)[1], + (int) ((unsigned char *) ethh->h_dest)[2], + (int) ((unsigned char *) ethh->h_dest)[3], + (int) ((unsigned char *) ethh->h_dest)[4], + (int) ((unsigned char *) ethh->h_dest)[5]); + fprintf(fp, "------------------------------------------------------\n"); +} + +response_type_t* synscan_classify_packet(const u_char *packet, uint32_t len) +{ + (void)len; + struct iphdr *ip_hdr = (struct iphdr *)&packet[sizeof(struct ethhdr)]; + struct tcphdr *tcp = (struct tcphdr*)((char *)ip_hdr + + (sizeof(struct iphdr))); + if (tcp->rst) { // RST packet + return &(module_tcp_synscan.responses[1]); + } else { // SYNACK packet + return &(module_tcp_synscan.responses[0]); + } +} +// Returns 0 if dst_port is outside the expected valid range, non-zero otherwise +static inline int check_dst_port(uint16_t port, uint32_t *validation) +{ + if (port > zconf.source_port_last + || port < zconf.source_port_first) { + return EXIT_FAILURE; + } + int32_t to_validate = port - zconf.source_port_first; + int32_t min = validation[1] % num_ports; + int32_t max = (validation[1] + zconf.packet_streams - 1) % num_ports; + + return (((max - min) % num_ports) >= ((to_validate - min) % num_ports)); +} + +int synscan_validate_packet(const struct iphdr *ip_hdr, uint32_t len, + __attribute__((unused))uint32_t *src_ip, uint32_t *validation) +{ + if (ip_hdr->protocol != IPPROTO_TCP) { + return 0; + } + + if ((4*ip_hdr->ihl + sizeof(struct tcphdr)) > len) { + // buffer not large enough to contain expected tcp header + return 0; + } + struct tcphdr *tcp = (struct tcphdr*)((char *)ip_hdr + 4*ip_hdr->ihl); + uint16_t sport = tcp->source; + uint16_t dport = tcp->dest; + + // validate source port + if (ntohs(sport) != zconf.target_port) { + return 0; + } + + // validate destination port + if (!check_dst_port(ntohs(dport), validation)) { + return 0; + } + + // validate tcp acknowledgement number + if (htonl(tcp->ack_seq) != htonl(validation[0])+1) { + return 0; + } + + return 1; +} + +static response_type_t responses[] = { + { + .is_success = 1, + .name = "synack" + }, + { + .is_success = 0, + .name = "rst" + } +}; + +probe_module_t module_tcp_synscan = { + .name = "tcp_synscan", + .packet_length = 54, + .pcap_filter = "tcp && tcp[13] & 4 != 0 || tcp[13] == 18", + .pcap_snaplen = 96, + .port_args = 1, + .thread_initialize = &synscan_init_perthread, + .make_packet = &synscan_make_packet, + .print_packet = &synscan_print_packet, + .classify_packet = &synscan_classify_packet, + .validate_packet = &synscan_validate_packet, + .close = NULL, + .responses = responses, +}; + diff --git a/src/probe_modules/module_udp.c b/src/probe_modules/module_udp.c new file mode 100644 index 0000000..d7e2278 --- /dev/null +++ b/src/probe_modules/module_udp.c @@ -0,0 +1,227 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +/* send module for performing TCP SYN scans */ + +#include +#include +#include +#include +#include +#include + +#include +#include +#include + +#include +#include +#include + +#include "probe_modules.h" +#include "packet.h" + +const char *udp_send_msg = "GET / HTTP/1.1\r\n\r\n"; // Must be null-terminated +static int num_ports = 1; + +probe_module_t module_udp; + +int udp_init_perthread(void* buf, macaddr_t *src, + macaddr_t *gw, __attribute__((unused)) port_h_t dst_port) +{ + memset(buf, 0, MAX_PACKET_SIZE); + struct ethhdr *eth_header = (struct ethhdr *)buf; + make_eth_header(eth_header, src, gw); + struct iphdr *ip_header = (struct iphdr*)(ð_header[1]); + uint16_t len = htons(sizeof(struct iphdr) + sizeof(struct udphdr) + strlen(udp_send_msg)); + make_ip_header(ip_header, IPPROTO_UDP, len); + + struct udphdr *udp_header = (struct udphdr*)(&ip_header[1]); + len = sizeof(struct udphdr) + strlen(udp_send_msg); + make_udp_header(udp_header, zconf.target_port, len); + + char* payload = (char*)(&udp_header[1]); + + module_udp.packet_length = sizeof(struct ethhdr) + sizeof(struct iphdr) + + sizeof(struct udphdr) + strlen(udp_send_msg); + assert(module_udp.packet_length <= MAX_PACKET_SIZE); + + strcpy(payload, udp_send_msg); + + num_ports = zconf.source_port_last - zconf.source_port_first + 1; + + return EXIT_SUCCESS; +} + +int udp_make_packet(void *buf, ipaddr_n_t src_ip, ipaddr_n_t dst_ip, + uint32_t *validation, int probe_num) +{ + struct ethhdr *eth_header = (struct ethhdr *)buf; + struct iphdr *ip_header = (struct iphdr*)(ð_header[1]); + struct udphdr *udp_header = (struct udphdr*)(&ip_header[1]); + uint16_t src_port = zconf.source_port_first + + ((validation[1] + probe_num) % num_ports); + + ip_header->saddr = src_ip; + ip_header->daddr = dst_ip; + udp_header->source = src_port; + + ip_header->check = 0; + ip_header->check = ip_checksum((unsigned short *) ip_header); + + return EXIT_SUCCESS; +} + +void udp_print_packet(FILE *fp, void* packet) +{ + struct ethhdr *ethh = (struct ethhdr *) packet; + struct iphdr *iph = (struct iphdr *) ðh[1]; + struct udphdr *udph = (struct udphdr*)(&iph[1]); + fprintf(fp, "udp { source: %u | dest: %u | checksum: %u }\n", + ntohs(udph->source), + ntohs(udph->dest), + ntohl(udph->check)); + //ip_header = (struct iphdr*)(ð_header[1]) + struct in_addr *s = (struct in_addr *) &(iph->saddr); + struct in_addr *d = (struct in_addr *) &(iph->daddr); + char srcip[20]; + char dstip[20]; + // inet_ntoa is a const char * so we if just call it in + // fprintf, you'll get back wrong results since we're + // calling it twice. + strncpy(srcip, inet_ntoa(*s), 19); + strncpy(dstip, inet_ntoa(*d), 19); + fprintf(fp, "ip { saddr: %s | daddr: %s | checksum: %u }\n", + srcip, + dstip, + ntohl(iph->check)); + fprintf(fp, "eth { shost: %02x:%02x:%02x:%02x:%02x:%02x | " + "dhost: %02x:%02x:%02x:%02x:%02x:%02x }\n", + (int) ((unsigned char *) ethh->h_source)[0], + (int) ((unsigned char *) ethh->h_source)[1], + (int) ((unsigned char *) ethh->h_source)[2], + (int) ((unsigned char *) ethh->h_source)[3], + (int) ((unsigned char *) ethh->h_source)[4], + (int) ((unsigned char *) ethh->h_source)[5], + (int) ((unsigned char *) ethh->h_dest)[0], + (int) ((unsigned char *) ethh->h_dest)[1], + (int) ((unsigned char *) ethh->h_dest)[2], + (int) ((unsigned char *) ethh->h_dest)[3], + (int) ((unsigned char *) ethh->h_dest)[4], + (int) ((unsigned char *) ethh->h_dest)[5]); + fprintf(fp, "------------------------------------------------------\n"); +} + + +response_type_t* udp_classify_packet(const u_char *packet, uint32_t len) +{ + (void)len; + struct iphdr *ip_hdr = (struct iphdr *)&packet[sizeof(struct ethhdr)]; + if (ip_hdr->protocol == IPPROTO_UDP) { + return &(module_udp.responses[0]); + } else if (ip_hdr->protocol == IPPROTO_ICMP) { + return &(module_udp.responses[1]); + } else { + return &(module_udp.responses[2]); + } +} + +// Returns 0 if dst_port is outside the expected valid range, non-zero otherwise +static inline int check_dst_port(uint16_t port, uint32_t *validation) +{ + if (port > zconf.source_port_last + || port < zconf.source_port_first) { + return EXIT_FAILURE; + } + int32_t to_validate = port - zconf.source_port_first; + int32_t min = validation[1] % num_ports; + int32_t max = (validation[1] + zconf.packet_streams - 1) % num_ports; + + return (((max - min) % num_ports) >= ((to_validate - min) % num_ports)); +} + +int udp_validate_packet(const struct iphdr *ip_hdr, uint32_t len, + __attribute__((unused))uint32_t *src_ip, uint32_t *validation) +{ + uint16_t dport, sport; + if (ip_hdr->protocol == IPPROTO_UDP) { + if ((4*ip_hdr->ihl + sizeof(struct udphdr)) > len) { + // buffer not large enough to contain expected udp header + return 0; + } + struct udphdr *udp = (struct udphdr*)((char *)ip_hdr + 4*ip_hdr->ihl); + + sport = ntohs(udp->dest); + dport = ntohs(udp->source); + } else if (ip_hdr->protocol == IPPROTO_ICMP) { + // UDP can return ICMP Destination unreach + // IP( ICMP( IP( UDP ) ) ) for a destination unreach + uint32_t min_len = 4*ip_hdr->ihl + sizeof(struct icmphdr) + + sizeof(struct iphdr) + sizeof(struct udphdr); + if (len < min_len) { + // Not enough information for us to validate + return 0; + } + + struct icmphdr *icmp = (struct icmphdr*)((char *)ip_hdr + 4*ip_hdr->ihl); + if (icmp->type != ICMP_DEST_UNREACH) { + return 0; + } + + struct iphdr *ip_inner = (struct iphdr*)&icmp[1]; + // Now we know the actual inner ip length, we should recheck the buffer + if (len < 4*ip_inner->ihl - sizeof(struct iphdr) + min_len) { + return 0; + } + // This is the packet we sent + struct udphdr *udp = (struct udphdr *)((char*)ip_inner + 4*ip_inner->ihl); + + sport = ntohs(udp->source); + dport = ntohs(udp->dest); + } else { + return 0; + } + if (dport != zconf.target_port) { + return 0; + } + if (!check_dst_port(sport, validation)) { + return 0; + } + return 1; +} + +static response_type_t responses[] = { + { + .is_success = 1, + .name = "data" + }, + { + .is_success = 0, + .name = "port-unreach" + }, + { + .is_success = 0, + .name = "invalid" + } +}; + +probe_module_t module_udp = { + .name = "udp", + .packet_length = 96, + .pcap_filter = "udp || icmp", + .pcap_snaplen = 96, + .port_args = 1, + .thread_initialize = &udp_init_perthread, + .make_packet = &udp_make_packet, + .print_packet = &udp_print_packet, + .validate_packet = &udp_validate_packet, + .classify_packet = &udp_classify_packet, + .close = NULL, + .responses = responses +}; + diff --git a/src/probe_modules/packet.c b/src/probe_modules/packet.c new file mode 100644 index 0000000..7e2ee42 --- /dev/null +++ b/src/probe_modules/packet.c @@ -0,0 +1,89 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + + +#include "packet.h" + +#include +#include +#include + +#include +#include +#include +#include +#include + +#include +#include + +#include "state.h" + +void print_macaddr(struct ifreq* i) +{ + printf("Device %s -> Ethernet %02x:%02x:%02x:%02x:%02x:%02x\n", + i->ifr_name, + (int) ((unsigned char *) &i->ifr_hwaddr.sa_data)[0], + (int) ((unsigned char *) &i->ifr_hwaddr.sa_data)[1], + (int) ((unsigned char *) &i->ifr_hwaddr.sa_data)[2], + (int) ((unsigned char *) &i->ifr_hwaddr.sa_data)[3], + (int) ((unsigned char *) &i->ifr_hwaddr.sa_data)[4], + (int) ((unsigned char *) &i->ifr_hwaddr.sa_data)[5]); +} + +void make_eth_header(struct ethhdr *ethh, macaddr_t *src, macaddr_t *dst) +{ + memcpy(ethh->h_source, src, ETH_ALEN); + memcpy(ethh->h_dest, dst, ETH_ALEN); + ethh->h_proto = htons(ETH_P_IP); +} + +void make_ip_header(struct iphdr *iph, uint8_t protocol, uint16_t len) +{ + iph->ihl = 5; // Internet Header Length + iph->version = 4; // IPv4 + iph->tos = 0; // Type of Service + iph->tot_len = len; + iph->id = htons(54321); // identification number + iph->frag_off = 0; //fragmentation falg + iph->ttl = MAXTTL; // time to live (TTL) + iph->protocol = protocol; // upper layer protocol => TCP + // we set the checksum = 0 for now because that's + // what it needs to be when we run the IP checksum + iph->check = 0; +} + +void make_icmp_header(struct icmp *buf) +{ + buf->icmp_type = ICMP_ECHO; + buf->icmp_code = 0; + buf->icmp_seq = 0; +} + +void make_tcp_header(struct tcphdr *tcp_header, port_h_t dest_port) +{ + tcp_header->seq = random(); + tcp_header->ack_seq = 0; + tcp_header->res2 = 0; + tcp_header->doff = 5; // data offset + tcp_header->syn = 1; + tcp_header->window = htons(65535); // largest possible window + tcp_header->check = 0; + tcp_header->urg_ptr = 0; + tcp_header->dest = htons(dest_port); +} + +void make_udp_header(struct udphdr *udp_header, port_h_t dest_port, + uint16_t len) +{ + udp_header->dest = htons(dest_port); + udp_header->len = htons(len); + // checksum ignored in IPv4 if 0 + udp_header->check = 0; +} + diff --git a/src/probe_modules/packet.h b/src/probe_modules/packet.h new file mode 100644 index 0000000..20f50c2 --- /dev/null +++ b/src/probe_modules/packet.h @@ -0,0 +1,82 @@ +#include "state.h" + +#include +#include +#include +#include +#include + +#ifndef HEADER_ZMAP_PACKET_H +#define HEADER_ZMAP_PACKET_H + +#define MAX_PACKET_SIZE 4096 + +typedef unsigned short __attribute__((__may_alias__)) alias_unsigned_short; + +void make_eth_header(struct ethhdr *ethh, macaddr_t *src, macaddr_t *dst); + +void make_ip_header(struct iphdr *iph, uint8_t, uint16_t); +void make_tcp_header(struct tcphdr*, port_h_t); +void make_icmp_header(struct icmp *); +void make_udp_header(struct udphdr *udp_header, port_h_t dest_port, + uint16_t len); + +static inline unsigned short in_checksum(unsigned short *ip_pkt, int len) +{ + unsigned long sum = 0; + for (int nwords = len/2; nwords > 0; nwords--) { + sum += *ip_pkt++; + } + sum = (sum >> 16) + (sum & 0xffff); + sum += (sum >> 16); + return (unsigned short) (~sum); +} + +__attribute__((unused)) static inline unsigned short ip_checksum( + unsigned short *buf) +{ + return in_checksum(buf, (int) sizeof(struct iphdr)); +} + +__attribute__((unused)) static inline unsigned short icmp_checksum( + unsigned short *buf) +{ + return in_checksum(buf, (int) sizeof(struct icmp)); +} + +static __attribute__((unused)) uint16_t tcp_checksum(unsigned short len_tcp, + uint32_t saddr, uint32_t daddr, struct tcphdr *tcp_pkt) +{ + alias_unsigned_short *src_addr = (alias_unsigned_short *) &saddr; + alias_unsigned_short *dest_addr = (alias_unsigned_short *) &daddr; + + unsigned char prot_tcp = 6; + unsigned long sum = 0; + int nleft = len_tcp; + unsigned short *w; + + w = (unsigned short *) tcp_pkt; + // calculate the checksum for the tcp header and tcp data + while(nleft > 1) { + sum += *w++; + nleft -= 2; + } + // if nleft is 1 there ist still on byte left. + // We add a padding byte (0xFF) to build a 16bit word + if (nleft > 0) { + sum += *w & ntohs(0xFF00); + } + // add the pseudo header + sum += src_addr[0]; + sum += src_addr[1]; + sum += dest_addr[0]; + sum += dest_addr[1]; + sum += htons(len_tcp); + sum += htons(prot_tcp); + sum = (sum >> 16) + (sum & 0xFFFF); + sum += (sum >> 16); + // Take the one's complement of sum + return (unsigned short) (~sum); +} + +#endif diff --git a/src/probe_modules/probe_modules.c b/src/probe_modules/probe_modules.c new file mode 100644 index 0000000..f67e831 --- /dev/null +++ b/src/probe_modules/probe_modules.c @@ -0,0 +1,42 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include + +#include "probe_modules.h" + +extern probe_module_t module_tcp_synscan; +extern probe_module_t module_icmp_echo; +extern probe_module_t module_udp; +// ADD YOUR MODULE HERE + +probe_module_t* probe_modules[] = { + &module_tcp_synscan, + &module_icmp_echo, + &module_udp + // ADD YOUR MODULE HERE +}; + + +probe_module_t* get_probe_module_by_name(const char* name) +{ + for (int i=0; i < (int) (sizeof(probe_modules)/sizeof(probe_modules[0])); i++) { + if (!strcmp(probe_modules[i]->name, name)) { + return probe_modules[i]; + } + } + return NULL; +} + +void print_probe_modules(void) +{ + for (int i=0; i < (int) (sizeof(probe_modules)/sizeof(probe_modules[0])); i++) { + printf("%s\n", probe_modules[i]->name); + } +} diff --git a/src/probe_modules/probe_modules.h b/src/probe_modules/probe_modules.h new file mode 100644 index 0000000..18f8438 --- /dev/null +++ b/src/probe_modules/probe_modules.h @@ -0,0 +1,45 @@ +#include "../state.h" + +#ifndef HEADER_PROBE_MODULES_H +#define HEADER_PROBE_MODULES_H + +typedef struct probe_response_type { + const uint8_t is_success; + const char *name; +} response_type_t; + +typedef int (*probe_global_init_cb)(struct state_conf *); +typedef int (*probe_thread_init_cb)(void* packetbuf, macaddr_t* src_mac, macaddr_t* gw_mac, port_n_t src_port); +typedef int (*probe_make_packet_cb)(void* packetbuf, ipaddr_n_t src_ip, ipaddr_n_t dst_ip, + uint32_t *validation, int probe_num); +typedef void (*probe_print_packet_cb)(FILE *, void* packetbuf); +typedef int (*probe_close_cb)(struct state_conf*, struct state_send*, struct state_recv*); +typedef response_type_t* (*probe_classify_packet_cb)(const u_char* packetbuf, uint32_t len); +typedef int (*probe_validate_packet_cb)(const struct iphdr *ip_hdr, uint32_t len, uint32_t *src_ip, uint32_t *validation); + +typedef struct probe_module { + const char *name; + size_t packet_length; + const char *pcap_filter; + size_t pcap_snaplen; + + // Should ZMap complain if the user hasn't specified valid + // source and target port numbers? + uint8_t port_args; + + response_type_t *responses; + + probe_global_init_cb global_initialize; + probe_thread_init_cb thread_initialize; + probe_make_packet_cb make_packet; + probe_print_packet_cb print_packet; + probe_validate_packet_cb validate_packet; + probe_classify_packet_cb classify_packet; + probe_close_cb close; + +} probe_module_t; + +probe_module_t* get_probe_module_by_name(const char*); +void print_probe_modules(void); + +#endif // HEADER_PROBE_MODULES_H diff --git a/src/recv.c b/src/recv.c new file mode 100644 index 0000000..1afa9ee --- /dev/null +++ b/src/recv.c @@ -0,0 +1,195 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include +#include + +#include +#include + +#include +#include +#include +#include +#include + +#include +#include +#include +#include +#include +#include + +#include "../lib/logger.h" + +#include "state.h" +#include "validate.h" +#include "probe_modules/probe_modules.h" +#include "output_modules/output_modules.h" + +#define PCAP_PROMISC 1 +#define PCAP_TIMEOUT 1000 + +static uint32_t num_src_ports; +static pcap_t *pc = NULL; + +// bitmap of observed IP addresses +static uint64_t *ip_seen = NULL; +static const int IP_SEEN_SIZE = 0x4000000; // == 2^32/64 + +// check if we've received a response from this address previously +static inline int check_ip(uint32_t ip) +{ + return (ip_seen[ip >> 6] >> (ip & 0x3F)) & 1; +} + +// set that we've received a response from the address +static inline void set_ip(uint32_t ip) +{ + ip_seen[ip >> 6] |= (uint64_t)1 << (ip & 0x3F); +} + +void packet_cb(u_char __attribute__((__unused__)) *user, + const struct pcap_pkthdr *p, const u_char *bytes) +{ + if (!p) { + return; + } + if (zrecv.success_unique >= zconf.max_results) { + // Libpcap can process multiple packets per pcap_dispatch; + // we need to throw out results once we've + // gotten our --max-results worth. + return; + } + // length of entire packet captured by libpcap + uint32_t buflen = (uint32_t) p->caplen; + + if ((sizeof(struct iphdr) + sizeof(struct ethhdr)) > buflen) { + // buffer not large enough to contain ethernet + // and ip headers. further action would overrun buf + return; + } + struct iphdr *ip_hdr = (struct iphdr *)&bytes[sizeof(struct ethhdr)]; + + uint32_t src_ip = ip_hdr->saddr; + + uint32_t validation[VALIDATE_BYTES/sizeof(uint8_t)]; + // TODO: for TTL exceeded messages, ip_hdr->saddr is going to be different + // and we must calculate off potential payload message instead + validate_gen(ip_hdr->daddr, ip_hdr->saddr, (uint8_t *)validation); + + if (!zconf.probe_module->validate_packet(ip_hdr, buflen - sizeof(struct ethhdr), + &src_ip, validation)) { + return; + } + + int is_repeat = check_ip(src_ip); + response_type_t *r = zconf.probe_module->classify_packet(bytes, buflen); + + if (r->is_success) { + zrecv.success_total++; + if (!is_repeat) { + zrecv.success_unique++; + set_ip(src_ip); + } + if (zsend.complete) { + zrecv.cooldown_total++; + if (!is_repeat) { + zrecv.cooldown_unique++; + } + } + if (zconf.output_module && zconf.output_module->success_ip) { + zconf.output_module->success_ip( + ip_hdr->saddr, ip_hdr->daddr, + r->name, is_repeat, zsend.complete, bytes, buflen); + } + } else { + zrecv.failure_total++; + if (zconf.output_module && zconf.output_module->other_ip) { + zconf.output_module->other_ip( + ip_hdr->saddr, ip_hdr->daddr, + r->name, is_repeat, zsend.complete, bytes, buflen); + } + } + + if (zconf.output_module && zconf.output_module->update + && !(zrecv.success_unique % zconf.output_module->update_interval)) { + zconf.output_module->update(&zconf, &zsend, &zrecv); + } +} + +int recv_update_pcap_stats(void) +{ + if (!pc) { + return EXIT_FAILURE; + } + struct pcap_stat pcst; + if (pcap_stats(pc, &pcst)) { + log_error("recv", "unable to retrieve pcap statistics: %s", + pcap_geterr(pc)); + return EXIT_FAILURE; + } else { + zrecv.pcap_recv = pcst.ps_recv; + zrecv.pcap_drop = pcst.ps_drop; + zrecv.pcap_ifdrop = pcst.ps_ifdrop; + } + return EXIT_SUCCESS; +} + +int recv_run(pthread_mutex_t *recv_ready_mutex) +{ + log_debug("recv", "thread started"); + num_src_ports = zconf.source_port_last - zconf.source_port_first + 1; + ip_seen = calloc(IP_SEEN_SIZE, sizeof(uint64_t)); + if (!ip_seen) { + log_fatal("recv", "couldn't allocate address bitmap"); + } + log_debug("recv", "using dev %s", zconf.iface); + char errbuf[PCAP_ERRBUF_SIZE]; + pc = pcap_open_live(zconf.iface, zconf.probe_module->pcap_snaplen, + PCAP_PROMISC, PCAP_TIMEOUT, errbuf); + if (pc == NULL) { + log_fatal("recv", "couldn't open device %s: %s", + zconf.iface, errbuf); + } + struct bpf_program bpf; + if (pcap_compile(pc, &bpf, zconf.probe_module->pcap_filter, 1, 0) < 0) { + log_fatal("recv", "couldn't compile filter"); + } + if (pcap_setfilter(pc, &bpf) < 0) { + log_fatal("recv", "couldn't install filter"); + } + log_debug("recv", "receiver ready"); + pthread_mutex_lock(recv_ready_mutex); + zconf.recv_ready = 1; + pthread_mutex_unlock(recv_ready_mutex); + zrecv.start = now(); + if (zconf.max_results == 0) { + zconf.max_results = -1; + } + do { + if (pcap_dispatch(pc, 0, packet_cb, NULL) == -1) { + log_fatal("recv", "pcap_dispatch error"); + } + if (zconf.max_results && zrecv.success_unique >= zconf.max_results) { + zsend.complete = 1; + break; + } + } while (!(zsend.complete && (now()-zsend.finish > zconf.cooldown_secs))); + zrecv.finish = now(); + // get final pcap statistics before closing + recv_update_pcap_stats(); + pcap_close(pc); + zrecv.complete = 1; + log_debug("recv", "thread finished"); + return 0; +} + diff --git a/src/recv.h b/src/recv.h new file mode 100644 index 0000000..e50c25e --- /dev/null +++ b/src/recv.h @@ -0,0 +1,16 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#ifndef _RECV_H +#define _RECV_H + +int recv_update_pcap_stats(void); + +int recv_run(); + +#endif //_RECV_H diff --git a/src/send.c b/src/send.c new file mode 100644 index 0000000..e30c044 --- /dev/null +++ b/src/send.c @@ -0,0 +1,295 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include "send.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include +#include +#include +#include +#include + +#include "../lib/logger.h" +#include "../lib/random.h" +#include "../lib/blacklist.h" + +#include "cyclic.h" +#include "state.h" +#include "probe_modules/packet.h" +#include "probe_modules/probe_modules.h" +#include "validate.h" + + +// lock to manage access to share send state (e.g. counters and cyclic) +pthread_mutex_t send_mutex = PTHREAD_MUTEX_INITIALIZER; +// lock to provide thread safety to the user provided send callback +pthread_mutex_t syncb_mutex = PTHREAD_MUTEX_INITIALIZER; + +// globals to handle sending from multiple ip addresses (shared across threads) +static uint16_t num_src_ports; +static uint32_t num_addrs; +static in_addr_t srcip_first; +static in_addr_t srcip_last; +// offset send addresses according to a random chosen per scan execution +// in order to help prevent cross-scan interference +static uint32_t srcip_offset; + +// global sender initialize (not thread specific) +int send_init(void) +{ + // generate a new primitive root and starting position + cyclic_init(0, 0); + zsend.first_scanned = cyclic_get_curr_ip(); + + // compute number of targets + uint64_t allowed = blacklist_count_allowed(); + if (allowed == (1LL << 32)) { + zsend.targets = 0xFFFFFFFF; + } else { + zsend.targets = allowed; + } + if (zsend.targets > zconf.max_targets) { + zsend.targets = zconf.max_targets; + } + + // process the dotted-notation addresses passed to ZMAP and determine + // the source addresses from which we'll send packets; + srcip_first = inet_addr(zconf.source_ip_first); + if (srcip_first == INADDR_NONE) { + log_fatal("send", "invalid begin source ip address: `%s'", + zconf.source_ip_first); + } + srcip_last = inet_addr(zconf.source_ip_last); + if (srcip_last == INADDR_NONE) { + log_fatal("send", "invalid end source ip address: `%s'", + zconf.source_ip_last); + } + if (srcip_first == srcip_last) { + srcip_offset = 0; + num_addrs = 1; + } else { + srcip_offset = rand() % (srcip_last - srcip_first); + num_addrs = ntohl(srcip_last) - ntohl(srcip_first) + 1; + } + + // process the source port range that ZMap is allowed to use + num_src_ports = zconf.source_port_last - zconf.source_port_first + 1; + log_debug("send", "will send from %i address%s on %u source ports", + num_addrs, ((num_addrs==1)?"":"es"), num_src_ports); + + // global initialization for send module + assert(zconf.probe_module); + if (zconf.probe_module->global_initialize) { + zconf.probe_module->global_initialize(&zconf); + } + + // concert specified bandwidth to packet rate + if (zconf.bandwidth > 0) { + int pkt_len = zconf.probe_module->packet_length; + pkt_len *= 8; + pkt_len += 8*24; // 7 byte MAC preamble, 1 byte Start frame, + // 4 byte CRC, 12 byte inter-frame gap + if (pkt_len < 84*8) { + pkt_len = 84*8; + } + if (zconf.bandwidth / pkt_len > 0xFFFFFFFF) { + zconf.rate = 0; + } else { + zconf.rate = zconf.bandwidth / pkt_len; + if (zconf.rate == 0) { + log_warn("send", "bandwidth %lu bit/s is slower than 1 pkt/s, " + "setting rate to 1 pkt/s", zconf.bandwidth); + zconf.rate = 1; + } + } + log_debug("send", "using bandwidth %lu bits/s, rate set to %d pkt/s", + zconf.bandwidth, zconf.rate); + } + + if (zconf.dryrun) { + log_info("send", "dryrun mode -- won't actually send packets"); + } + + // initialize random validation key + validate_init(); + + zsend.start = now(); + return EXIT_SUCCESS; +} + +static int get_socket(void) +{ + int sock = socket(AF_PACKET, SOCK_RAW, htons(ETH_P_ALL)); + if (sock <= 0) { + log_fatal("send", "couldn't create socket. " + "Are you root? Error: %s\n", strerror(errno)); + } + return sock; +} + +static inline ipaddr_n_t get_src_ip(ipaddr_n_t dst, int local_offset) +{ + if (srcip_first == srcip_last) { + return srcip_first; + } + return htonl(((ntohl(dst) + srcip_offset + local_offset) + % num_addrs)) + srcip_first; +} + +// one sender thread +int send_run(void) +{ + log_debug("send", "thread started"); + pthread_mutex_lock(&send_mutex); + int sock = get_socket(); + struct sockaddr_ll sockaddr; + // get source interface index + struct ifreq if_idx; + memset(&if_idx, 0, sizeof(struct ifreq)); + if (strlen(zconf.iface) >= IFNAMSIZ) { + log_error("send", "device interface name (%s) too long\n", + zconf.iface); + return -1; + } + strncpy(if_idx.ifr_name, zconf.iface, IFNAMSIZ-1); + if (ioctl(sock, SIOCGIFINDEX, &if_idx) < 0) { + perror("SIOCGIFINDEX"); + return -1; + } + int ifindex = if_idx.ifr_ifindex; + // get source interface mac + struct ifreq if_mac; + memset(&if_mac, 0, sizeof(struct ifreq)); + strncpy(if_mac.ifr_name, zconf.iface, IFNAMSIZ-1); + if (ioctl(sock, SIOCGIFHWADDR, &if_mac) < 0) { + perror("SIOCGIFHWADDR"); + return -1; + } + // find source IP address associated with the dev from which we're sending. + // while we won't use this address for sending packets, we need the address + // to set certain socket options and it's easiest to just use the primary + // address the OS believes is associated. + struct ifreq if_ip; + memset(&if_ip, 0, sizeof(struct ifreq)); + strncpy(if_ip.ifr_name, zconf.iface, IFNAMSIZ-1); + if (ioctl(sock, SIOCGIFADDR, &if_ip) < 0) { + perror("SIOCGIFADDR"); + return -1; + } + // destination address for the socket + memset((void*) &sockaddr, 0, sizeof(struct sockaddr_ll)); + sockaddr.sll_ifindex = ifindex; + sockaddr.sll_halen = ETH_ALEN; + memcpy(sockaddr.sll_addr, zconf.gw_mac, ETH_ALEN); + + char buf[MAX_PACKET_SIZE]; + memset(buf, 0, MAX_PACKET_SIZE); + zconf.probe_module->thread_initialize(buf, + (unsigned char *)if_mac.ifr_hwaddr.sa_data, + zconf.gw_mac, zconf.target_port); + pthread_mutex_unlock(&send_mutex); + + // adaptive timing to hit target rate + uint32_t count = 0; + uint32_t last_count = count; + double last_time = now(); + uint32_t delay = 0; + int interval = 0; + volatile int vi; + if (zconf.rate > 0) { + // estimate initial rate + delay = 10000; + for (vi = delay; vi--; ) + ; + delay *= 1 / (now() - last_time) / (zconf.rate / zconf.senders); + interval = (zconf.rate / zconf.senders) / 20; + last_time = now(); + } + while (1) { + // adaptive timing delay + if (delay > 0) { + count++; + for (vi = delay; vi--; ) + ; + if (!interval || (count % interval == 0)) { + double t = now(); + delay *= (double)(count - last_count) + / (t - last_time) / (zconf.rate / zconf.senders); + if (delay < 1) + delay = 1; + last_count = count; + last_time = t; + } + } + // generate next ip from cyclic group and update global state + // (everything locked happens here) + pthread_mutex_lock(&send_mutex); + if (zsend.complete) { + pthread_mutex_unlock(&send_mutex); + break; + } + if (zsend.sent >= zconf.max_targets) { + zsend.complete = 1; + zsend.finish = now(); + pthread_mutex_unlock(&send_mutex); + break; + } + if (zconf.max_runtime && zconf.max_runtime <= now() - zsend.start) { + zsend.complete = 1; + zsend.finish = now(); + pthread_mutex_unlock(&send_mutex); + break; + } + uint32_t curr = cyclic_get_next_ip(); + if (curr == zsend.first_scanned) { + zsend.complete = 1; + zsend.finish = now(); + } + zsend.sent++; + pthread_mutex_unlock(&send_mutex); + for (int i=0; i < zconf.packet_streams; i++) { + uint32_t src_ip = get_src_ip(curr, i); + + uint32_t validation[VALIDATE_BYTES/sizeof(uint32_t)]; + validate_gen(src_ip, curr, (uint8_t *)validation); + zconf.probe_module->make_packet(buf, src_ip, curr, validation, i); + + if (zconf.dryrun) { + zconf.probe_module->print_packet(stdout, buf); + } else { + int l = zconf.probe_module->packet_length; + int rc = sendto(sock, buf, + l, 0, + (struct sockaddr *)&sockaddr, + sizeof(struct sockaddr_ll)); + if (rc < 0) { + struct in_addr addr; + addr.s_addr = curr; + log_debug("send", "sendto failed for %s. %s", + inet_ntoa(addr), strerror(errno)); + pthread_mutex_lock(&send_mutex); + zsend.sendto_failures++; + pthread_mutex_unlock(&send_mutex); + } + } + } + } + log_debug("send", "thread finished"); + return EXIT_SUCCESS; +} + diff --git a/src/send.h b/src/send.h new file mode 100644 index 0000000..5116e45 --- /dev/null +++ b/src/send.h @@ -0,0 +1,15 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#ifndef _SEND_H +#define _SEND_H + +int send_init(void); +int send_run(void); + +#endif //_SEND_H diff --git a/src/state.c b/src/state.c new file mode 100644 index 0000000..a1caa8a --- /dev/null +++ b/src/state.c @@ -0,0 +1,70 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + + +#include "state.h" +#include "../lib/logger.h" + +// global configuration and defaults +struct state_conf zconf = { + .log_level = LOG_INFO, + .source_port_first = 32768, // (these are the default + .source_port_last = 61000, // ephemeral range on Linux) + .output_filename = NULL, + .blacklist_filename = NULL, + .whitelist_filename = NULL, + .target_port = 0, + .max_targets = 0xFFFFFFFF, + .max_runtime = 0, + .max_results = 0, + .iface = NULL, + .rate = 0, + .bandwidth = 0, + .cooldown_secs = 0, + .senders = 1, + .packet_streams = 1, + .use_seed = 0, + .seed = 0, + .output_module = NULL, + .output_args = NULL, + .probe_module = NULL, + .probe_args = NULL, + .gw_mac = {0}, + .gw_mac_set = 0, + .source_ip_first = NULL, + .source_ip_last = NULL, + .dryrun = 0, + .quiet = 0, + .summary = 0, + .recv_ready = 0, +}; + +// global sender stats and defaults +struct state_send zsend = { + .start = 0.0, + .finish = 0.0, + .sent = 0, + .blacklisted = 0, + .complete = 0, + .sendto_failures = 0, + .targets = 0, +}; + +// global receiver stats and defaults +struct state_recv zrecv = { + .success_unique = 0, + .success_total = 0, + .cooldown_unique = 0, + .cooldown_total = 0, + .failure_total = 0, + .complete = 0, + .pcap_recv = 0, + .pcap_drop = 0, + .pcap_ifdrop = 0, +}; + diff --git a/src/state.h b/src/state.h new file mode 100644 index 0000000..f1c967e --- /dev/null +++ b/src/state.h @@ -0,0 +1,124 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include +#include +#include +#include + +#ifndef _STATE_H +#define _STATE_H + +#define MAX_PACKET_SIZE 4096 + +typedef uint32_t ipaddr_n_t; // IPv4 address network order +typedef uint32_t ipaddr_h_t; // IPv4 address host order +typedef uint16_t port_n_t; // port network order +typedef uint16_t port_h_t; // port host order +typedef unsigned char macaddr_t; + +struct probe_module; +struct output_module; + +// global configuration +struct state_conf { + int log_level; + port_h_t target_port; + port_h_t source_port_first; + port_h_t source_port_last; + // maximum number of packets that the scanner will send before terminating + uint32_t max_targets; + // maximum number of seconds that scanner will run before terminating + uint32_t max_runtime; + // maximum number of results before terminating + uint32_t max_results; + // name of network interface that + // will be utilized for sending/receiving + char *iface; + // rate in packets per second + // that the sender will maintain + int rate; + // rate in bits per second + uint64_t bandwidth; + // how many seconds after the termination of the sender will the receiver + // continue to process responses + int cooldown_secs; + // number of sending threads + int senders; + // should use CLI provided randomization seed instead of generating + // a random seed. + int use_seed; + uint32_t seed; + // generator of the cyclic multiplicative group that is utilized for + // address generation + uint32_t generator; + int packet_streams; + struct probe_module *probe_module; + struct output_module *output_module; + char *probe_args; + char *output_args; + macaddr_t gw_mac[IFHWADDRLEN]; + int gw_mac_set; + char *source_ip_first; + char *source_ip_last; + char *output_filename; + char *blacklist_filename; + char *whitelist_filename; + int dryrun; + int summary; + int quiet; + int recv_ready; +}; +extern struct state_conf zconf; + +// global sender stats +struct state_send { + double start; + double finish; + uint32_t sent; + uint32_t blacklisted; + int complete; + uint32_t first_scanned; + uint32_t targets; + uint32_t sendto_failures; +}; +extern struct state_send zsend; + +// global receiver stats +struct state_recv { + // valid responses classified as "success" + uint32_t success_total; + // unique IPs that sent valid responses classified as "success" + uint32_t success_unique; + // valid responses classified as "success" received during cooldown + uint32_t cooldown_total; + // unique IPs that first sent valid "success"es during cooldown + uint32_t cooldown_unique; + // valid responses NOT classified as "success" + uint32_t failure_total; + + int complete; // has the scanner finished sending? + double start; // timestamp of when recv started + double finish; // timestamp of when recv terminated + + // number of packets captured by pcap filter + uint32_t pcap_recv; + // number of packets dropped because there was no room in + // the operating system's buffer when they arrived, because + // packets weren't being read fast enough + uint32_t pcap_drop; + // number of packets dropped by the network interface or its driver. + uint32_t pcap_ifdrop; + +}; +extern struct state_recv zrecv; + +#endif // _STATE_H + diff --git a/src/validate.c b/src/validate.c new file mode 100644 index 0000000..2ffcc20 --- /dev/null +++ b/src/validate.c @@ -0,0 +1,47 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#include +#include +#include "../lib/rijndael-alg-fst.h" +#include "../lib/random.h" +#include "../lib/logger.h" +#include "validate.h" + +#define AES_ROUNDS 10 +#define AES_BLOCK_WORDS 4 +#define AES_KEY_BYTES 16 + +static int inited = 0; +static uint32_t aes_input[AES_BLOCK_WORDS]; +static uint32_t aes_sched[(AES_ROUNDS+1)*4]; + +void validate_init() +{ + for (int i=0; i < AES_BLOCK_WORDS; i++) { + aes_input[i] = 0; + } + uint8_t key[AES_KEY_BYTES]; + if (!random_bytes(key, AES_KEY_BYTES)) { + log_fatal("validate", "couldn't get random bytes"); + } + if (rijndaelKeySetupEnc(aes_sched, key, AES_KEY_BYTES*8) != AES_ROUNDS) { + log_fatal("validate", "couldn't initialize AES key"); + } + inited = 1; +} + +void validate_gen(const uint32_t src, const uint32_t dst, + uint8_t output[VALIDATE_BYTES]) +{ + assert(inited); + aes_input[0] = src; + aes_input[1] = dst; + rijndaelEncrypt(aes_sched, AES_ROUNDS, (uint8_t *)aes_input, output); +} + diff --git a/src/validate.h b/src/validate.h new file mode 100644 index 0000000..e1b0230 --- /dev/null +++ b/src/validate.h @@ -0,0 +1,17 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#ifndef _VALIDATE_H +#define _VALIDATE_H + +#define VALIDATE_BYTES 16 + +void validate_init(); +void validate_gen(const uint32_t src, const uint32_t dst, uint8_t output[VALIDATE_BYTES]); + +#endif//_VALIDATE_H diff --git a/src/zmap.1 b/src/zmap.1 new file mode 100644 index 0000000..63aa850 --- /dev/null +++ b/src/zmap.1 @@ -0,0 +1,155 @@ +.TH zmap 1 "06 Aug 2013" "1.0" "zmap man page" +.SH NAME +zmap \- A fast Internet-wide scanner +.SH SYNOPSIS +.B zmap +[ +.I "OPTIONS \&..." +] +.SH DESCRIPTION +.I ZMap +is a network tool for scanning the entire Internet (or large samples). +.SH OPTIONS +.SS "Basic options" +.TP +.B \-p, --target-port=port +TCP port number to scan (for SYN scans) +.TP +.B \-o, --output-file=name +When using an output module that uses a file (such as the default), +write results to this file. Use - for stdout. +.TP +.B \-b, --blacklist-file=path +File of subnets to exclude, in CIDR notation (e.g. 192.168.0.0/16), +one-per line. It is recommended you use this to exclude RFC 1918 +addresses, multicast, IANA reserved space, and other IANA +special-purpose addresses. An example blacklist file is provided in +.B conf/blacklist.example +for this purpose. +.TP +.B -w, --whitelist-file=path +File of subnets to constrain scan to, in CIDR +notation, e.g. 192.168.0.0/16 + +.SS "Scan options" +.TP +.B \-n, --max-targets=n +Cap number of targets to probe (as a number or +a percentage of the address space) +.TP +.B \-N, --max-results=n +Cap number of results to return +.TP +.B \-t, --max-runtime=secs +Cap length of time for sending packets +.TP +.B \-r, --rate=pps +Set send rate in packets/sec +.TP +.B \-B, --bandwidth=bps +Set send rate in bits/second (supports suffixes G, M and K). This +overrides the +.B --rate +flag. +.TP +.B \-c, --cooldown-time=secs +How long to continue receiving after sending +last probe (default=8) +.TP +.B \-e, --seed=n +Seed used to select address permutation. Specify the same seed in order to scan the same sample repeatedly. +.TP +.B \-T, --sender-threads=n +Threads used to send packets (default=1) +.TP +.B \-P, --probes=n +Number of probes to send to each IP +(default=1) +.TP +.B \-d, --dryrun +Print out each packet to stdout instead of sending it. +(May be useful for debugging.) + +.SS "Network options" +.TP +.B \-s, --source-port=port|range +Source port(s) for scan packets +.TP +.B \-S, --source-ip=ip|range +Source address(es) for scan packets +.TP +.B \-G, --gateway-mac=addr +Specify gateway MAC address. All packets will be sent to this +Ethernet address. +.TP +.B \-i, --interface=name +Specify network interface to use. + +.SS "Advanced options" +.TP +.B \-M, --probe-module=name +Select probe module (default=tcp_synscan) +.TP +.B \-O, --output-module=name +Select output module (default=simple_file) +.TP +.B --probe-args=args +Arguments to pass to probe module +.TP +.B --output-args=args +Arguments to pass to output module +.TP +.B --list-output-modules +List available output modules +.TP +.B --list-probe-modules +List available probe modules + +.SS "Additional options" +.TP +.B \-C, --config=filename +Read a configuration file, which can specify +any of these options (default=zmap.conf) +.TP +.B \-q, --quiet +Do not print status updates +.TP +.B \-g, --summary +Print configuration and summary at end of scan +.TP +.B \-v, --verbosity=n +Level of log detail (0-5) (default=3) +.TP +.B \-h, --help +Print help and exit +.TP +.B \-V, --version +Print version and exit + +.SH EXAMPLES +Scan the whole Internet for hosts with port 443 open (results discarded): +.PP +.\" -p: example of +.B zmap \-p 443 +.PP +Find 5 HTTP servers (port 80), scanning at 10 Mb/s, print the results to stdout: +.PP +.\" -N: example of +.\" -B: example of +.B zmap -N 5 -B 10M -p 80 -o - + +.SH WARNING +By default, ZMap attempts to scan at the line speed of your Ethernet +interface and can easily use 1 Gbit/second of bandwidth. If your +network is not able to support sending packets this quickly, your +local network may become congested, causing connectivity problems for +you and those around you. Use the -B (--bandwidth) option to set +ZMap's maximum bandwidth to an appropriate limit for your network and +upstream connection. +. +.SH AUTHOR +Zakir Durumeric, +Eric Wustrow, +J. Alex Halderman +.B (https://www.zmap.io) + diff --git a/src/zmap.c b/src/zmap.c new file mode 100644 index 0000000..9ebab9e --- /dev/null +++ b/src/zmap.c @@ -0,0 +1,483 @@ +/* + * ZMap Copyright 2013 Regents of the University of Michigan + * + * Licensed under the Apache License, Version 2.0 (the "License"); you may not + * use this file except in compliance with the License. You may obtain a copy + * of the License at http://www.apache.org/licenses/LICENSE-2.0 + */ + +#define _GNU_SOURCE + +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include +#include + +#include "../lib/logger.h" +#include "../lib/random.h" + +#include "zopt.h" +#include "send.h" +#include "recv.h" +#include "state.h" +#include "monitor.h" +#include "output_modules/output_modules.h" +#include "probe_modules/probe_modules.h" +#include "get_gateway.h" + +pthread_mutex_t cpu_affinity_mutex = PTHREAD_MUTEX_INITIALIZER; +pthread_mutex_t recv_ready_mutex = PTHREAD_MUTEX_INITIALIZER; + +static void set_cpu(void) +{ + pthread_mutex_lock(&cpu_affinity_mutex); + static int core=0; + int num_cores = sysconf(_SC_NPROCESSORS_ONLN); + cpu_set_t cpuset; + CPU_ZERO(&cpuset); + CPU_SET(core, &cpuset); + if (pthread_setaffinity_np(pthread_self(), + sizeof(cpu_set_t), &cpuset) != 0) { + log_error("zmap", "can't set thread CPU affinity"); + } + log_trace("zmap", "set thread %u affinity to core %d", + pthread_self(), core); + core = (core + 1) % num_cores; + pthread_mutex_unlock(&cpu_affinity_mutex); +} + +static void* start_send(__attribute__((unused)) void *arg) +{ + set_cpu(); + send_run(); + return NULL; +} + +static void* start_recv(__attribute__((unused)) void *arg) +{ + set_cpu(); + recv_run(&recv_ready_mutex); + return NULL; +} + +static void *start_mon(__attribute__((unused)) void *arg) +{ + set_cpu(); + monitor_run(); + return NULL; +} + +#define SI(w,x,y) printf("%s\t%s\t%i\n", w, x, y); +#define SD(w,x,y) printf("%s\t%s\t%f\n", w, x, y); +#define SU(w,x,y) printf("%s\t%s\t%u\n", w, x, y); +#define SLU(w,x,y) printf("%s\t%s\t%lu\n", w, x, y); +#define SS(w,x,y) printf("%s\t%s\t%s\n", w, x, y); +#define STRTIME_LEN 1024 + +static void summary(void) +{ + char send_start_time[STRTIME_LEN+1]; + assert(dstrftime(send_start_time, STRTIME_LEN, "%c", zsend.start)); + char send_end_time[STRTIME_LEN+1]; + assert(dstrftime(send_end_time, STRTIME_LEN, "%c", zsend.finish)); + char recv_start_time[STRTIME_LEN+1]; + assert(dstrftime(recv_start_time, STRTIME_LEN, "%c", zrecv.start)); + char recv_end_time[STRTIME_LEN+1]; + assert(dstrftime(recv_end_time, STRTIME_LEN, "%c", zrecv.finish)); + double hitrate = ((double) 100 * zrecv.success_unique)/((double)zsend.sent); + + SU("cnf", "target-port", zconf.target_port); + SU("cnf", "source-port-range-begin", zconf.source_port_first); + SU("cnf", "source-port-range-end", zconf.source_port_last); + SS("cnf", "source-addr-range-begin", zconf.source_ip_first); + SS("cnf", "source-addr-range-end", zconf.source_ip_last); + SU("cnf", "maximum-targets", zconf.max_targets); + SU("cnf", "maximum-runtime", zconf.max_runtime); + SU("cnf", "maximum-results", zconf.max_results); + SU("cnf", "permutation-seed", zconf.seed); + SI("cnf", "cooldown-period", zconf.cooldown_secs); + SS("cnf", "send-interface", zconf.iface); + SI("cnf", "rate", zconf.rate); + SLU("cnf", "bandwidth", zconf.bandwidth); + SU("env", "nprocessors", (unsigned) sysconf(_SC_NPROCESSORS_ONLN)); + SS("exc", "send-start-time", send_start_time); + SS("exc", "send-end-time", send_end_time); + SS("exc", "recv-start-time", recv_start_time); + SS("exc", "recv-end-time", recv_end_time); + SU("exc", "sent", zsend.sent); + SU("exc", "blacklisted", zsend.blacklisted); + SU("exc", "first-scanned", zsend.first_scanned); + SD("exc", "hit-rate", hitrate); + SU("exc", "success-total", zrecv.success_total); + SU("exc", "success-unique", zrecv.success_unique); + SU("exc", "success-cooldown-total", zrecv.cooldown_total); + SU("exc", "success-cooldown-unique", zrecv.cooldown_unique); + SU("exc", "failure-total", zrecv.failure_total); + SU("exc", "sendto-failures", zsend.sendto_failures); + SU("adv", "permutation-gen", zconf.generator); + SS("exc", "scan-type", zconf.probe_module->name); +} + +static void start_zmap(void) +{ + log_init(stderr, zconf.log_level); + log_info("zmap", "started"); + + // finish setting up configuration + if (zconf.iface == NULL) { + char errbuf[PCAP_ERRBUF_SIZE]; + char *iface = pcap_lookupdev(errbuf); + if (iface == NULL) { + log_fatal("zmap", "could not detect default network interface " + "(e.g. eth0). Try running as root or setting" + " interface using -i flag."); + } + log_debug("zmap", "no interface provided. will use %s", iface); + zconf.iface = iface; + } + if (zconf.source_ip_first == NULL) { + struct in_addr default_ip; + zconf.source_ip_first = malloc(INET_ADDRSTRLEN); + zconf.source_ip_last = zconf.source_ip_first; + if (get_iface_ip(zconf.iface, &default_ip) < 0) { + log_fatal("zmap", "could not detect default IP address for for %s." + " Try specifying a source address (-S).", zconf.iface); + } + inet_ntop(AF_INET, &default_ip, zconf.source_ip_first, INET_ADDRSTRLEN); + log_debug("zmap", "no source IP address given. will use %s", + zconf.source_ip_first); + } + if (!zconf.gw_mac_set) { + struct in_addr gw_ip; + char iface[IF_NAMESIZE]; + if (get_default_gw(&gw_ip, iface) < 0) { + log_fatal("zmap", "could not detect default gateway address for %i." + " Try setting default gateway mac address (-G)."); + } + log_debug("zmap", "found gateway IP %s on %s", inet_ntoa(gw_ip), iface); + if (get_hw_addr(&gw_ip, iface, zconf.gw_mac) < 0) { + log_fatal("zmap", "could not detect GW MAC address for %s on %s." + " Try setting default gateway mac address (-G).", + inet_ntoa(gw_ip), zconf.iface); + } + zconf.gw_mac_set = 1; + log_debug("zmap", "using default gateway MAC %02x:%02x:%02x:%02x:%02x:%02x", + zconf.gw_mac[0], zconf.gw_mac[1], zconf.gw_mac[2], + zconf.gw_mac[3], zconf.gw_mac[4], zconf.gw_mac[5]); + } + + // initialization + if (zconf.output_module && zconf.output_module->init) { + zconf.output_module->init(&zconf); + } + if (send_init()) { + exit(EXIT_FAILURE); + } + if (zconf.output_module && zconf.output_module->start) { + zconf.output_module->start(&zconf, &zsend, &zrecv); + } + + // start threads + pthread_t *tsend, trecv, tmon; + int r = pthread_create(&trecv, NULL, start_recv, NULL); + if (r != 0) { + log_fatal("zmap", "unable to create recv thread"); + exit(EXIT_FAILURE); + } + for (;;) { + pthread_mutex_lock(&recv_ready_mutex); + if (zconf.recv_ready) { + break; + } + pthread_mutex_unlock(&recv_ready_mutex); + } + tsend = malloc(zconf.senders * sizeof(pthread_t)); + assert(tsend); + log_debug("zmap", "using %d sender threads", zconf.senders); + for (int i=0; i < zconf.senders; i++) { + r = pthread_create(&tsend[i], NULL, start_send, NULL); + if (r != 0) { + log_fatal("zmap", "unable to create send thread"); + exit(EXIT_FAILURE); + } + } + if (!zconf.quiet) { + r = pthread_create(&tmon, NULL, start_mon, NULL); + if (r != 0) { + log_fatal("zmap", "unable to create monitor thread"); + exit(EXIT_FAILURE); + } + } + + // wait for completion + for (int i=0; i < zconf.senders; i++) { + pthread_join(tsend[i], NULL); + if (r != 0) { + log_fatal("zmap", "unable to join send thread"); + exit(EXIT_FAILURE); + } + } + log_debug("zmap", "senders finished"); + pthread_join(trecv, NULL); + if (r != 0) { + log_fatal("zmap", "unable to join recv thread"); + exit(EXIT_FAILURE); + } + if (!zconf.quiet) { + pthread_join(tmon, NULL); + if (r != 0) { + log_fatal("zmap", "unable to join monitor thread"); + exit(EXIT_FAILURE); + } + } + + // finished + if (zconf.summary) { + summary(); + } + if (zconf.output_module && zconf.output_module->close) { + zconf.output_module->close(&zconf, &zsend, &zrecv); + } + if (zconf.probe_module && zconf.probe_module->close) { + zconf.probe_module->close(&zconf, &zsend, &zrecv); + } + log_info("zmap", "completed"); +} + +static void enforce_range(const char *name, int v, int min, int max) +{ + if (v < min || v > max) { + fprintf(stderr, "%s: argument `%s' must be between %d and %d\n", + CMDLINE_PARSER_PACKAGE, name, min, max); + exit(EXIT_FAILURE); + } +} + +static int file_exists(char *name) +{ + FILE *file = fopen(name, "r"); + if (!file) + return 0; + fclose(file); + return 1; +} + +#define MAC_LEN IFHWADDRLEN +int parse_mac(macaddr_t *out, char *in) +{ + if (strlen(in) < MAC_LEN*3-1) + return 0; + char octet[3]; + octet[2] = '\0'; + for (int i=0; i < MAC_LEN; i++) { + if (i < MAC_LEN-1 && in[i*3+2] != ':') { + return 0; + } + strncpy(octet, &in[i*3], 2); + char *err = NULL; + long b = strtol(octet, &err, 16); + if (err && *err != '\0') { + return 0; + } + out[i] = b & 0xFF; + } + return 1; +} + +#define SET_IF_GIVEN(DST,ARG) \ + { if (args.ARG##_given) { (DST) = args.ARG##_arg; }; } +#define SET_BOOL(DST,ARG) \ + { if (args.ARG##_given) { (DST) = 1; }; } + +int main(int argc, char *argv[]) +{ + struct gengetopt_args_info args; + struct cmdline_parser_params *params; + params = cmdline_parser_params_create(); + params->initialize = 1; + params->override = 0; + params->check_required = 0; + if (cmdline_parser_ext(argc, argv, &args, params) != 0) { + exit(EXIT_SUCCESS); + } + if (args.help_given) { + cmdline_parser_print_help(); + exit(EXIT_SUCCESS); + } + if (args.version_given) { + cmdline_parser_print_version(); + exit(EXIT_SUCCESS); + } + if (args.list_output_modules_given) { + print_output_modules(); + exit(EXIT_SUCCESS); + } + if (args.list_probe_modules_given) { + print_probe_modules(); + exit(EXIT_SUCCESS); + } + if (args.config_given || file_exists(args.config_arg)) { + params->initialize = 0; + params->override = 0; + if (cmdline_parser_config_file(args.config_arg, &args, params) + != 0) { + exit(EXIT_FAILURE); + } + } + if (cmdline_parser_required(&args, CMDLINE_PARSER_PACKAGE) != 0) { + exit(EXIT_FAILURE); + } + + SET_IF_GIVEN(zconf.output_filename, output_file); + SET_IF_GIVEN(zconf.blacklist_filename, blacklist_file); + SET_IF_GIVEN(zconf.whitelist_filename, whitelist_file); + SET_IF_GIVEN(zconf.probe_args, probe_args); + SET_IF_GIVEN(zconf.output_args, output_args); + SET_IF_GIVEN(zconf.iface, interface); + SET_IF_GIVEN(zconf.max_runtime, max_runtime); + SET_IF_GIVEN(zconf.max_results, max_results); + SET_IF_GIVEN(zconf.rate, rate); + SET_IF_GIVEN(zconf.packet_streams, probes); + SET_BOOL(zconf.dryrun, dryrun); + SET_BOOL(zconf.quiet, quiet); + SET_BOOL(zconf.summary, summary); + zconf.cooldown_secs = args.cooldown_time_arg; + zconf.senders = args.sender_threads_arg; + zconf.log_level = args.verbosity_arg; + + zconf.output_module = get_output_module_by_name(args.output_module_arg); + if (!zconf.output_module) { + fprintf(stderr, "%s: specified output module (%s) does not exist\n", + args.output_module_arg, CMDLINE_PARSER_PACKAGE); + exit(EXIT_FAILURE); + } + zconf.probe_module = get_probe_module_by_name(args.probe_module_arg); + if (!zconf.probe_module) { + fprintf(stderr, "%s: specified probe module (%s) does not exist\n", + CMDLINE_PARSER_PACKAGE, args.probe_module_arg); + exit(EXIT_FAILURE); + } + if (zconf.probe_module->port_args) { + if (args.source_port_given) { + char *dash = strchr(args.source_port_arg, '-'); + if (dash) { // range + *dash = '\0'; + zconf.source_port_first = atoi(args.source_port_arg); + enforce_range("starting source-port", zconf.source_port_first, 0, 0xFFFF); + zconf.source_port_last = atoi(dash+1); + enforce_range("ending source-port", zconf.source_port_last, 0, 0xFFFF); + if (zconf.source_port_first > zconf.source_port_last) { + fprintf(stderr, "%s: invalid source port range: " + "last port is less than first port\n", + CMDLINE_PARSER_PACKAGE); + exit(EXIT_FAILURE); + } + } else { // single port + int port = atoi(args.source_port_arg); + enforce_range("source-port", port, 0, 0xFFFF); + zconf.source_port_first = port; + zconf.source_port_last = port; + } + } + if (!args.target_port_given) { + fprintf(stderr, "%s: target port is required for this type of probe\n", + CMDLINE_PARSER_PACKAGE); + exit(EXIT_FAILURE); + } + enforce_range("target-port", args.target_port_arg, 0, 0xFFFF); + zconf.target_port = args.target_port_arg; + } + if (args.source_ip_given) { + char *dash = strchr(args.source_ip_arg, '-'); + if (dash) { // range + *dash = '\0'; + zconf.source_ip_first = args.source_ip_arg; + zconf.source_ip_last = dash+1; + } else { // single address + zconf.source_ip_first = args.source_ip_arg; + zconf.source_ip_last = args.source_ip_arg; + } + } + if (args.gateway_mac_given) { + if (!parse_mac(zconf.gw_mac, args.gateway_mac_arg)) { + fprintf(stderr, "%s: invalid MAC address `%s'\n", + CMDLINE_PARSER_PACKAGE, args.gateway_mac_arg); + exit(EXIT_FAILURE); + } + zconf.gw_mac_set = 1; + } + if (args.seed_given) { + zconf.seed = args.seed_arg; + zconf.use_seed = 1; + } + if (args.bandwidth_given) { + // Supported: G,g=*1000000000; M,m=*1000000 K,k=*1000 bits per second + zconf.bandwidth = atoi(args.bandwidth_arg); + char *suffix = args.bandwidth_arg; + while (*suffix >= '0' && *suffix <= '9') { + suffix++; + } + if (*suffix) { + switch (*suffix) { + case 'G': case 'g': + zconf.bandwidth *= 1000000000; + break; + case 'M': case 'm': + zconf.bandwidth *= 1000000; + break; + case 'K': case 'k': + zconf.bandwidth *= 1000; + break; + default: + fprintf(stderr, "%s: unknown bandwidth suffix '%s' " + "(supported suffixes are G, M and K)\n", + CMDLINE_PARSER_PACKAGE, suffix); + exit(EXIT_FAILURE); + } + } + } + if (args.max_targets_given) { + errno = 0; + char *end; + double v = strtod(args.max_targets_arg, &end); + if (end == args.max_targets_arg || errno != 0) { + fprintf(stderr, "%s: can't convert max-targets to a number\n", + CMDLINE_PARSER_PACKAGE); + exit(EXIT_FAILURE); + } + if (end[0] == '%' && end[1] == '\0') { + // treat as percentage + v = v * (1L << 32) / 100.; + } else if (end[0] != '\0') { + fprintf(stderr, "%s: extra characters after max-targets\n", + CMDLINE_PARSER_PACKAGE); + exit(EXIT_FAILURE); + } + if (v <= 0) { + zconf.max_targets = 0; + } + else if (v >= (1L << 32)) { + zconf.max_targets = 0xFFFFFFFF; + } else { + zconf.max_targets = v; + } + } + + start_zmap(); + + cmdline_parser_free(&args); + free(params); + return EXIT_SUCCESS; +} + diff --git a/src/zopt.c b/src/zopt.c new file mode 100644 index 0000000..0441c76 --- /dev/null +++ b/src/zopt.c @@ -0,0 +1,1339 @@ +/* + File autogenerated by gengetopt version 2.22.5 + generated with the following command: + gengetopt -C --no-help --no-version -i zopt.ggo -F zopt + + The developers of gengetopt consider the fixed text that goes in all + gengetopt output files to be in the public domain: + we make no copyright claims on it. +*/ + +/* If we use autoconf. */ +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include +#include +#include + +#ifndef FIX_UNUSED +#define FIX_UNUSED(X) (void) (X) /* avoid warnings for unused params */ +#endif + +#include + +#include "zopt.h" + +const char *gengetopt_args_info_purpose = "A fast Internet-wide scanner."; + +const char *gengetopt_args_info_usage = "Usage: zmap [OPTIONS]..."; + +const char *gengetopt_args_info_description = ""; + +const char *gengetopt_args_info_help[] = { + "Basic arguments:", + " -p, --target-port=port TCP port number to scan (for SYN scans)", + " -o, --output-file=name Output file", + " -b, --blacklist-file=path File of subnets to exclude, in CIDR notation, \n e.g. 192.168.0.0/16", + " -w, --whitelist-file=path File of subnets to constrain scan to, in CIDR \n notation, e.g. 192.168.0.0/16", + "\nScan options:", + " -n, --max-targets=n Cap number of targets to probe (as a number or \n a percentage of the address space)", + " -N, --max-results=n Cap number of results to return", + " -t, --max-runtime=ses Cap length of time for sending packets", + " -r, --rate=pps Set send rate in packets/sec", + " -B, --bandwidth=bps Set send rate in bits/second (supports suffixes \n G, M and K)", + " -c, --cooldown-time=secs How long to continue receiving after sending \n last probe (default=`8')", + " -e, --seed=n Seed used to select address permutation", + " -T, --sender-threads=n Threads used to send packets (default=`1')", + " -P, --probes=n Number of probes to send to each IP \n (default=`1')", + " -d, --dryrun Don't actually send packets", + "\nNetwork options:", + " -s, --source-port=port|range Source port(s) for scan packets", + " -S, --source-ip=ip|range Source address(es) for scan packets", + " -G, --gateway-mac=addr Specify gateway MAC address", + " -i, --interface=name Specify network interface to use", + "\nAdvanced options:", + " -M, --probe-module=name Select probe module (default=`tcp_synscan')", + " -O, --output-module=name Select output module (default=`simple_file')", + " --probe-args=args Arguments to pass to probe module", + " --output-args=args Arguments to pass to output module", + " --list-output-modules List available output modules", + " --list-probe-modules List available probe modules", + "\nAdditional options:", + " -C, --config=filename Read a configuration file, which can specify \n any of these options \n (default=`/etc/zmap/zmap.conf')", + " -q, --quiet Do not print status updates", + " -g, --summary Print configuration and summary at end of scan", + " -v, --verbosity=n Level of log detail (0-5) (default=`3')", + " -h, --help Print help and exit", + " -V, --version Print version and exit", + "\nExamples:\n zmap -p 443 (scans the whole Internet for hosts with port 443 open)\n zmap -N 5 -B 10M -p 80 -o - (find 5 HTTP servers, scanning at 10 Mb/s)", + 0 +}; + +typedef enum {ARG_NO + , ARG_STRING + , ARG_INT +} cmdline_parser_arg_type; + +static +void clear_given (struct gengetopt_args_info *args_info); +static +void clear_args (struct gengetopt_args_info *args_info); + +static int +cmdline_parser_internal (int argc, char **argv, struct gengetopt_args_info *args_info, + struct cmdline_parser_params *params, const char *additional_error); + +struct line_list +{ + char * string_arg; + struct line_list * next; +}; + +static struct line_list *cmd_line_list = 0; +static struct line_list *cmd_line_list_tmp = 0; + +static void +free_cmd_list(void) +{ + /* free the list of a previous call */ + if (cmd_line_list) + { + while (cmd_line_list) { + cmd_line_list_tmp = cmd_line_list; + cmd_line_list = cmd_line_list->next; + free (cmd_line_list_tmp->string_arg); + free (cmd_line_list_tmp); + } + } +} + + +static char * +gengetopt_strdup (const char *s); + +static +void clear_given (struct gengetopt_args_info *args_info) +{ + args_info->target_port_given = 0 ; + args_info->output_file_given = 0 ; + args_info->blacklist_file_given = 0 ; + args_info->whitelist_file_given = 0 ; + args_info->max_targets_given = 0 ; + args_info->max_results_given = 0 ; + args_info->max_runtime_given = 0 ; + args_info->rate_given = 0 ; + args_info->bandwidth_given = 0 ; + args_info->cooldown_time_given = 0 ; + args_info->seed_given = 0 ; + args_info->sender_threads_given = 0 ; + args_info->probes_given = 0 ; + args_info->dryrun_given = 0 ; + args_info->source_port_given = 0 ; + args_info->source_ip_given = 0 ; + args_info->gateway_mac_given = 0 ; + args_info->interface_given = 0 ; + args_info->probe_module_given = 0 ; + args_info->output_module_given = 0 ; + args_info->probe_args_given = 0 ; + args_info->output_args_given = 0 ; + args_info->list_output_modules_given = 0 ; + args_info->list_probe_modules_given = 0 ; + args_info->config_given = 0 ; + args_info->quiet_given = 0 ; + args_info->summary_given = 0 ; + args_info->verbosity_given = 0 ; + args_info->help_given = 0 ; + args_info->version_given = 0 ; +} + +static +void clear_args (struct gengetopt_args_info *args_info) +{ + FIX_UNUSED (args_info); + args_info->target_port_orig = NULL; + args_info->output_file_arg = NULL; + args_info->output_file_orig = NULL; + args_info->blacklist_file_arg = NULL; + args_info->blacklist_file_orig = NULL; + args_info->whitelist_file_arg = NULL; + args_info->whitelist_file_orig = NULL; + args_info->max_targets_arg = NULL; + args_info->max_targets_orig = NULL; + args_info->max_results_orig = NULL; + args_info->max_runtime_orig = NULL; + args_info->rate_orig = NULL; + args_info->bandwidth_arg = NULL; + args_info->bandwidth_orig = NULL; + args_info->cooldown_time_arg = 8; + args_info->cooldown_time_orig = NULL; + args_info->seed_orig = NULL; + args_info->sender_threads_arg = 1; + args_info->sender_threads_orig = NULL; + args_info->probes_arg = 1; + args_info->probes_orig = NULL; + args_info->source_port_arg = NULL; + args_info->source_port_orig = NULL; + args_info->source_ip_arg = NULL; + args_info->source_ip_orig = NULL; + args_info->gateway_mac_arg = NULL; + args_info->gateway_mac_orig = NULL; + args_info->interface_arg = NULL; + args_info->interface_orig = NULL; + args_info->probe_module_arg = gengetopt_strdup ("tcp_synscan"); + args_info->probe_module_orig = NULL; + args_info->output_module_arg = gengetopt_strdup ("simple_file"); + args_info->output_module_orig = NULL; + args_info->probe_args_arg = NULL; + args_info->probe_args_orig = NULL; + args_info->output_args_arg = NULL; + args_info->output_args_orig = NULL; + args_info->config_arg = gengetopt_strdup ("/etc/zmap/zmap.conf"); + args_info->config_orig = NULL; + args_info->verbosity_arg = 3; + args_info->verbosity_orig = NULL; + +} + +static +void init_args_info(struct gengetopt_args_info *args_info) +{ + + + args_info->target_port_help = gengetopt_args_info_help[1] ; + args_info->output_file_help = gengetopt_args_info_help[2] ; + args_info->blacklist_file_help = gengetopt_args_info_help[3] ; + args_info->whitelist_file_help = gengetopt_args_info_help[4] ; + args_info->max_targets_help = gengetopt_args_info_help[6] ; + args_info->max_results_help = gengetopt_args_info_help[7] ; + args_info->max_runtime_help = gengetopt_args_info_help[8] ; + args_info->rate_help = gengetopt_args_info_help[9] ; + args_info->bandwidth_help = gengetopt_args_info_help[10] ; + args_info->cooldown_time_help = gengetopt_args_info_help[11] ; + args_info->seed_help = gengetopt_args_info_help[12] ; + args_info->sender_threads_help = gengetopt_args_info_help[13] ; + args_info->probes_help = gengetopt_args_info_help[14] ; + args_info->dryrun_help = gengetopt_args_info_help[15] ; + args_info->source_port_help = gengetopt_args_info_help[17] ; + args_info->source_ip_help = gengetopt_args_info_help[18] ; + args_info->gateway_mac_help = gengetopt_args_info_help[19] ; + args_info->interface_help = gengetopt_args_info_help[20] ; + args_info->probe_module_help = gengetopt_args_info_help[22] ; + args_info->output_module_help = gengetopt_args_info_help[23] ; + args_info->probe_args_help = gengetopt_args_info_help[24] ; + args_info->output_args_help = gengetopt_args_info_help[25] ; + args_info->list_output_modules_help = gengetopt_args_info_help[26] ; + args_info->list_probe_modules_help = gengetopt_args_info_help[27] ; + args_info->config_help = gengetopt_args_info_help[29] ; + args_info->quiet_help = gengetopt_args_info_help[30] ; + args_info->summary_help = gengetopt_args_info_help[31] ; + args_info->verbosity_help = gengetopt_args_info_help[32] ; + args_info->help_help = gengetopt_args_info_help[33] ; + args_info->version_help = gengetopt_args_info_help[34] ; + +} + +void +cmdline_parser_print_version (void) +{ + printf ("%s %s\n", + (strlen(CMDLINE_PARSER_PACKAGE_NAME) ? CMDLINE_PARSER_PACKAGE_NAME : CMDLINE_PARSER_PACKAGE), + CMDLINE_PARSER_VERSION); +} + +static void print_help_common(void) { + cmdline_parser_print_version (); + + if (strlen(gengetopt_args_info_purpose) > 0) + printf("\n%s\n", gengetopt_args_info_purpose); + + if (strlen(gengetopt_args_info_usage) > 0) + printf("\n%s\n", gengetopt_args_info_usage); + + printf("\n"); + + if (strlen(gengetopt_args_info_description) > 0) + printf("%s\n\n", gengetopt_args_info_description); +} + +void +cmdline_parser_print_help (void) +{ + int i = 0; + print_help_common(); + while (gengetopt_args_info_help[i]) + printf("%s\n", gengetopt_args_info_help[i++]); +} + +void +cmdline_parser_init (struct gengetopt_args_info *args_info) +{ + clear_given (args_info); + clear_args (args_info); + init_args_info (args_info); +} + +void +cmdline_parser_params_init(struct cmdline_parser_params *params) +{ + if (params) + { + params->override = 0; + params->initialize = 1; + params->check_required = 1; + params->check_ambiguity = 0; + params->print_errors = 1; + } +} + +struct cmdline_parser_params * +cmdline_parser_params_create(void) +{ + struct cmdline_parser_params *params = + (struct cmdline_parser_params *)malloc(sizeof(struct cmdline_parser_params)); + cmdline_parser_params_init(params); + return params; +} + +static void +free_string_field (char **s) +{ + if (*s) + { + free (*s); + *s = 0; + } +} + + +static void +cmdline_parser_release (struct gengetopt_args_info *args_info) +{ + + free_string_field (&(args_info->target_port_orig)); + free_string_field (&(args_info->output_file_arg)); + free_string_field (&(args_info->output_file_orig)); + free_string_field (&(args_info->blacklist_file_arg)); + free_string_field (&(args_info->blacklist_file_orig)); + free_string_field (&(args_info->whitelist_file_arg)); + free_string_field (&(args_info->whitelist_file_orig)); + free_string_field (&(args_info->max_targets_arg)); + free_string_field (&(args_info->max_targets_orig)); + free_string_field (&(args_info->max_results_orig)); + free_string_field (&(args_info->max_runtime_orig)); + free_string_field (&(args_info->rate_orig)); + free_string_field (&(args_info->bandwidth_arg)); + free_string_field (&(args_info->bandwidth_orig)); + free_string_field (&(args_info->cooldown_time_orig)); + free_string_field (&(args_info->seed_orig)); + free_string_field (&(args_info->sender_threads_orig)); + free_string_field (&(args_info->probes_orig)); + free_string_field (&(args_info->source_port_arg)); + free_string_field (&(args_info->source_port_orig)); + free_string_field (&(args_info->source_ip_arg)); + free_string_field (&(args_info->source_ip_orig)); + free_string_field (&(args_info->gateway_mac_arg)); + free_string_field (&(args_info->gateway_mac_orig)); + free_string_field (&(args_info->interface_arg)); + free_string_field (&(args_info->interface_orig)); + free_string_field (&(args_info->probe_module_arg)); + free_string_field (&(args_info->probe_module_orig)); + free_string_field (&(args_info->output_module_arg)); + free_string_field (&(args_info->output_module_orig)); + free_string_field (&(args_info->probe_args_arg)); + free_string_field (&(args_info->probe_args_orig)); + free_string_field (&(args_info->output_args_arg)); + free_string_field (&(args_info->output_args_orig)); + free_string_field (&(args_info->config_arg)); + free_string_field (&(args_info->config_orig)); + free_string_field (&(args_info->verbosity_orig)); + + + + clear_given (args_info); +} + + +static void +write_into_file(FILE *outfile, const char *opt, const char *arg, const char *values[]) +{ + FIX_UNUSED (values); + if (arg) { + fprintf(outfile, "%s=\"%s\"\n", opt, arg); + } else { + fprintf(outfile, "%s\n", opt); + } +} + + +int +cmdline_parser_dump(FILE *outfile, struct gengetopt_args_info *args_info) +{ + int i = 0; + + if (!outfile) + { + fprintf (stderr, "%s: cannot dump options to stream\n", CMDLINE_PARSER_PACKAGE); + return EXIT_FAILURE; + } + + if (args_info->target_port_given) + write_into_file(outfile, "target-port", args_info->target_port_orig, 0); + if (args_info->output_file_given) + write_into_file(outfile, "output-file", args_info->output_file_orig, 0); + if (args_info->blacklist_file_given) + write_into_file(outfile, "blacklist-file", args_info->blacklist_file_orig, 0); + if (args_info->whitelist_file_given) + write_into_file(outfile, "whitelist-file", args_info->whitelist_file_orig, 0); + if (args_info->max_targets_given) + write_into_file(outfile, "max-targets", args_info->max_targets_orig, 0); + if (args_info->max_results_given) + write_into_file(outfile, "max-results", args_info->max_results_orig, 0); + if (args_info->max_runtime_given) + write_into_file(outfile, "max-runtime", args_info->max_runtime_orig, 0); + if (args_info->rate_given) + write_into_file(outfile, "rate", args_info->rate_orig, 0); + if (args_info->bandwidth_given) + write_into_file(outfile, "bandwidth", args_info->bandwidth_orig, 0); + if (args_info->cooldown_time_given) + write_into_file(outfile, "cooldown-time", args_info->cooldown_time_orig, 0); + if (args_info->seed_given) + write_into_file(outfile, "seed", args_info->seed_orig, 0); + if (args_info->sender_threads_given) + write_into_file(outfile, "sender-threads", args_info->sender_threads_orig, 0); + if (args_info->probes_given) + write_into_file(outfile, "probes", args_info->probes_orig, 0); + if (args_info->dryrun_given) + write_into_file(outfile, "dryrun", 0, 0 ); + if (args_info->source_port_given) + write_into_file(outfile, "source-port", args_info->source_port_orig, 0); + if (args_info->source_ip_given) + write_into_file(outfile, "source-ip", args_info->source_ip_orig, 0); + if (args_info->gateway_mac_given) + write_into_file(outfile, "gateway-mac", args_info->gateway_mac_orig, 0); + if (args_info->interface_given) + write_into_file(outfile, "interface", args_info->interface_orig, 0); + if (args_info->probe_module_given) + write_into_file(outfile, "probe-module", args_info->probe_module_orig, 0); + if (args_info->output_module_given) + write_into_file(outfile, "output-module", args_info->output_module_orig, 0); + if (args_info->probe_args_given) + write_into_file(outfile, "probe-args", args_info->probe_args_orig, 0); + if (args_info->output_args_given) + write_into_file(outfile, "output-args", args_info->output_args_orig, 0); + if (args_info->list_output_modules_given) + write_into_file(outfile, "list-output-modules", 0, 0 ); + if (args_info->list_probe_modules_given) + write_into_file(outfile, "list-probe-modules", 0, 0 ); + if (args_info->config_given) + write_into_file(outfile, "config", args_info->config_orig, 0); + if (args_info->quiet_given) + write_into_file(outfile, "quiet", 0, 0 ); + if (args_info->summary_given) + write_into_file(outfile, "summary", 0, 0 ); + if (args_info->verbosity_given) + write_into_file(outfile, "verbosity", args_info->verbosity_orig, 0); + if (args_info->help_given) + write_into_file(outfile, "help", 0, 0 ); + if (args_info->version_given) + write_into_file(outfile, "version", 0, 0 ); + + + i = EXIT_SUCCESS; + return i; +} + +int +cmdline_parser_file_save(const char *filename, struct gengetopt_args_info *args_info) +{ + FILE *outfile; + int i = 0; + + outfile = fopen(filename, "w"); + + if (!outfile) + { + fprintf (stderr, "%s: cannot open file for writing: %s\n", CMDLINE_PARSER_PACKAGE, filename); + return EXIT_FAILURE; + } + + i = cmdline_parser_dump(outfile, args_info); + fclose (outfile); + + return i; +} + +void +cmdline_parser_free (struct gengetopt_args_info *args_info) +{ + cmdline_parser_release (args_info); +} + +/** @brief replacement of strdup, which is not standard */ +char * +gengetopt_strdup (const char *s) +{ + char *result = 0; + if (!s) + return result; + + result = (char*)malloc(strlen(s) + 1); + if (result == (char*)0) + return (char*)0; + strcpy(result, s); + return result; +} + +int +cmdline_parser (int argc, char **argv, struct gengetopt_args_info *args_info) +{ + return cmdline_parser2 (argc, argv, args_info, 0, 1, 1); +} + +int +cmdline_parser_ext (int argc, char **argv, struct gengetopt_args_info *args_info, + struct cmdline_parser_params *params) +{ + int result; + result = cmdline_parser_internal (argc, argv, args_info, params, 0); + + if (result == EXIT_FAILURE) + { + cmdline_parser_free (args_info); + exit (EXIT_FAILURE); + } + + return result; +} + +int +cmdline_parser2 (int argc, char **argv, struct gengetopt_args_info *args_info, int override, int initialize, int check_required) +{ + int result; + struct cmdline_parser_params params; + + params.override = override; + params.initialize = initialize; + params.check_required = check_required; + params.check_ambiguity = 0; + params.print_errors = 1; + + result = cmdline_parser_internal (argc, argv, args_info, ¶ms, 0); + + if (result == EXIT_FAILURE) + { + cmdline_parser_free (args_info); + exit (EXIT_FAILURE); + } + + return result; +} + +int +cmdline_parser_required (struct gengetopt_args_info *args_info, const char *prog_name) +{ + FIX_UNUSED (args_info); + FIX_UNUSED (prog_name); + return EXIT_SUCCESS; +} + + +static char *package_name = 0; + +/** + * @brief updates an option + * @param field the generic pointer to the field to update + * @param orig_field the pointer to the orig field + * @param field_given the pointer to the number of occurrence of this option + * @param prev_given the pointer to the number of occurrence already seen + * @param value the argument for this option (if null no arg was specified) + * @param possible_values the possible values for this option (if specified) + * @param default_value the default value (in case the option only accepts fixed values) + * @param arg_type the type of this option + * @param check_ambiguity @see cmdline_parser_params.check_ambiguity + * @param override @see cmdline_parser_params.override + * @param no_free whether to free a possible previous value + * @param multiple_option whether this is a multiple option + * @param long_opt the corresponding long option + * @param short_opt the corresponding short option (or '-' if none) + * @param additional_error possible further error specification + */ +static +int update_arg(void *field, char **orig_field, + unsigned int *field_given, unsigned int *prev_given, + char *value, const char *possible_values[], + const char *default_value, + cmdline_parser_arg_type arg_type, + int check_ambiguity, int override, + int no_free, int multiple_option, + const char *long_opt, char short_opt, + const char *additional_error) +{ + char *stop_char = 0; + const char *val = value; + int found; + char **string_field; + FIX_UNUSED (field); + + stop_char = 0; + found = 0; + + if (!multiple_option && prev_given && (*prev_given || (check_ambiguity && *field_given))) + { + if (short_opt != '-') + fprintf (stderr, "%s: `--%s' (`-%c') option given more than once%s\n", + package_name, long_opt, short_opt, + (additional_error ? additional_error : "")); + else + fprintf (stderr, "%s: `--%s' option given more than once%s\n", + package_name, long_opt, + (additional_error ? additional_error : "")); + return 1; /* failure */ + } + + FIX_UNUSED (default_value); + + if (field_given && *field_given && ! override) + return 0; + if (prev_given) + (*prev_given)++; + if (field_given) + (*field_given)++; + if (possible_values) + val = possible_values[found]; + + switch(arg_type) { + case ARG_INT: + if (val) *((int *)field) = strtol (val, &stop_char, 0); + break; + case ARG_STRING: + if (val) { + string_field = (char **)field; + if (!no_free && *string_field) + free (*string_field); /* free previous string */ + *string_field = gengetopt_strdup (val); + } + break; + default: + break; + }; + + /* check numeric conversion */ + switch(arg_type) { + case ARG_INT: + if (val && !(stop_char && *stop_char == '\0')) { + fprintf(stderr, "%s: invalid numeric value: %s\n", package_name, val); + return 1; /* failure */ + } + break; + default: + ; + }; + + /* store the original value */ + switch(arg_type) { + case ARG_NO: + break; + default: + if (value && orig_field) { + if (no_free) { + *orig_field = value; + } else { + if (*orig_field) + free (*orig_field); /* free previous string */ + *orig_field = gengetopt_strdup (value); + } + } + }; + + return 0; /* OK */ +} + + +int +cmdline_parser_internal ( + int argc, char **argv, struct gengetopt_args_info *args_info, + struct cmdline_parser_params *params, const char *additional_error) +{ + int c; /* Character of the parsed option. */ + + int error = 0; + struct gengetopt_args_info local_args_info; + + int override; + int initialize; + int check_required; + int check_ambiguity; + + package_name = argv[0]; + + override = params->override; + initialize = params->initialize; + check_required = params->check_required; + check_ambiguity = params->check_ambiguity; + + if (initialize) + cmdline_parser_init (args_info); + + cmdline_parser_init (&local_args_info); + + optarg = 0; + optind = 0; + opterr = params->print_errors; + optopt = '?'; + + while (1) + { + int option_index = 0; + + static struct option long_options[] = { + { "target-port", 1, NULL, 'p' }, + { "output-file", 1, NULL, 'o' }, + { "blacklist-file", 1, NULL, 'b' }, + { "whitelist-file", 1, NULL, 'w' }, + { "max-targets", 1, NULL, 'n' }, + { "max-results", 1, NULL, 'N' }, + { "max-runtime", 1, NULL, 't' }, + { "rate", 1, NULL, 'r' }, + { "bandwidth", 1, NULL, 'B' }, + { "cooldown-time", 1, NULL, 'c' }, + { "seed", 1, NULL, 'e' }, + { "sender-threads", 1, NULL, 'T' }, + { "probes", 1, NULL, 'P' }, + { "dryrun", 0, NULL, 'd' }, + { "source-port", 1, NULL, 's' }, + { "source-ip", 1, NULL, 'S' }, + { "gateway-mac", 1, NULL, 'G' }, + { "interface", 1, NULL, 'i' }, + { "probe-module", 1, NULL, 'M' }, + { "output-module", 1, NULL, 'O' }, + { "probe-args", 1, NULL, 0 }, + { "output-args", 1, NULL, 0 }, + { "list-output-modules", 0, NULL, 0 }, + { "list-probe-modules", 0, NULL, 0 }, + { "config", 1, NULL, 'C' }, + { "quiet", 0, NULL, 'q' }, + { "summary", 0, NULL, 'g' }, + { "verbosity", 1, NULL, 'v' }, + { "help", 0, NULL, 'h' }, + { "version", 0, NULL, 'V' }, + { 0, 0, 0, 0 } + }; + + c = getopt_long (argc, argv, "p:o:b:w:n:N:t:r:B:c:e:T:P:ds:S:G:i:M:O:C:qgv:hV", long_options, &option_index); + + if (c == -1) break; /* Exit from `while (1)' loop. */ + + switch (c) + { + case 'p': /* TCP port number to scan (for SYN scans). */ + + + if (update_arg( (void *)&(args_info->target_port_arg), + &(args_info->target_port_orig), &(args_info->target_port_given), + &(local_args_info.target_port_given), optarg, 0, 0, ARG_INT, + check_ambiguity, override, 0, 0, + "target-port", 'p', + additional_error)) + goto failure; + + break; + case 'o': /* Output file. */ + + + if (update_arg( (void *)&(args_info->output_file_arg), + &(args_info->output_file_orig), &(args_info->output_file_given), + &(local_args_info.output_file_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "output-file", 'o', + additional_error)) + goto failure; + + break; + case 'b': /* File of subnets to exclude, in CIDR notation, e.g. 192.168.0.0/16. */ + + + if (update_arg( (void *)&(args_info->blacklist_file_arg), + &(args_info->blacklist_file_orig), &(args_info->blacklist_file_given), + &(local_args_info.blacklist_file_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "blacklist-file", 'b', + additional_error)) + goto failure; + + break; + case 'w': /* File of subnets to constrain scan to, in CIDR notation, e.g. 192.168.0.0/16. */ + + + if (update_arg( (void *)&(args_info->whitelist_file_arg), + &(args_info->whitelist_file_orig), &(args_info->whitelist_file_given), + &(local_args_info.whitelist_file_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "whitelist-file", 'w', + additional_error)) + goto failure; + + break; + case 'n': /* Cap number of targets to probe (as a number or a percentage of the address space). */ + + + if (update_arg( (void *)&(args_info->max_targets_arg), + &(args_info->max_targets_orig), &(args_info->max_targets_given), + &(local_args_info.max_targets_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "max-targets", 'n', + additional_error)) + goto failure; + + break; + case 'N': /* Cap number of results to return. */ + + + if (update_arg( (void *)&(args_info->max_results_arg), + &(args_info->max_results_orig), &(args_info->max_results_given), + &(local_args_info.max_results_given), optarg, 0, 0, ARG_INT, + check_ambiguity, override, 0, 0, + "max-results", 'N', + additional_error)) + goto failure; + + break; + case 't': /* Cap length of time for sending packets. */ + + + if (update_arg( (void *)&(args_info->max_runtime_arg), + &(args_info->max_runtime_orig), &(args_info->max_runtime_given), + &(local_args_info.max_runtime_given), optarg, 0, 0, ARG_INT, + check_ambiguity, override, 0, 0, + "max-runtime", 't', + additional_error)) + goto failure; + + break; + case 'r': /* Set send rate in packets/sec. */ + + + if (update_arg( (void *)&(args_info->rate_arg), + &(args_info->rate_orig), &(args_info->rate_given), + &(local_args_info.rate_given), optarg, 0, 0, ARG_INT, + check_ambiguity, override, 0, 0, + "rate", 'r', + additional_error)) + goto failure; + + break; + case 'B': /* Set send rate in bits/second (supports suffixes G, M and K). */ + + + if (update_arg( (void *)&(args_info->bandwidth_arg), + &(args_info->bandwidth_orig), &(args_info->bandwidth_given), + &(local_args_info.bandwidth_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "bandwidth", 'B', + additional_error)) + goto failure; + + break; + case 'c': /* How long to continue receiving after sending last probe. */ + + + if (update_arg( (void *)&(args_info->cooldown_time_arg), + &(args_info->cooldown_time_orig), &(args_info->cooldown_time_given), + &(local_args_info.cooldown_time_given), optarg, 0, "8", ARG_INT, + check_ambiguity, override, 0, 0, + "cooldown-time", 'c', + additional_error)) + goto failure; + + break; + case 'e': /* Seed used to select address permutation. */ + + + if (update_arg( (void *)&(args_info->seed_arg), + &(args_info->seed_orig), &(args_info->seed_given), + &(local_args_info.seed_given), optarg, 0, 0, ARG_INT, + check_ambiguity, override, 0, 0, + "seed", 'e', + additional_error)) + goto failure; + + break; + case 'T': /* Threads used to send packets. */ + + + if (update_arg( (void *)&(args_info->sender_threads_arg), + &(args_info->sender_threads_orig), &(args_info->sender_threads_given), + &(local_args_info.sender_threads_given), optarg, 0, "1", ARG_INT, + check_ambiguity, override, 0, 0, + "sender-threads", 'T', + additional_error)) + goto failure; + + break; + case 'P': /* Number of probes to send to each IP. */ + + + if (update_arg( (void *)&(args_info->probes_arg), + &(args_info->probes_orig), &(args_info->probes_given), + &(local_args_info.probes_given), optarg, 0, "1", ARG_INT, + check_ambiguity, override, 0, 0, + "probes", 'P', + additional_error)) + goto failure; + + break; + case 'd': /* Don't actually send packets. */ + + + if (update_arg( 0 , + 0 , &(args_info->dryrun_given), + &(local_args_info.dryrun_given), optarg, 0, 0, ARG_NO, + check_ambiguity, override, 0, 0, + "dryrun", 'd', + additional_error)) + goto failure; + + break; + case 's': /* Source port(s) for scan packets. */ + + + if (update_arg( (void *)&(args_info->source_port_arg), + &(args_info->source_port_orig), &(args_info->source_port_given), + &(local_args_info.source_port_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "source-port", 's', + additional_error)) + goto failure; + + break; + case 'S': /* Source address(es) for scan packets. */ + + + if (update_arg( (void *)&(args_info->source_ip_arg), + &(args_info->source_ip_orig), &(args_info->source_ip_given), + &(local_args_info.source_ip_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "source-ip", 'S', + additional_error)) + goto failure; + + break; + case 'G': /* Specify gateway MAC address. */ + + + if (update_arg( (void *)&(args_info->gateway_mac_arg), + &(args_info->gateway_mac_orig), &(args_info->gateway_mac_given), + &(local_args_info.gateway_mac_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "gateway-mac", 'G', + additional_error)) + goto failure; + + break; + case 'i': /* Specify network interface to use. */ + + + if (update_arg( (void *)&(args_info->interface_arg), + &(args_info->interface_orig), &(args_info->interface_given), + &(local_args_info.interface_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "interface", 'i', + additional_error)) + goto failure; + + break; + case 'M': /* Select probe module. */ + + + if (update_arg( (void *)&(args_info->probe_module_arg), + &(args_info->probe_module_orig), &(args_info->probe_module_given), + &(local_args_info.probe_module_given), optarg, 0, "tcp_synscan", ARG_STRING, + check_ambiguity, override, 0, 0, + "probe-module", 'M', + additional_error)) + goto failure; + + break; + case 'O': /* Select output module. */ + + + if (update_arg( (void *)&(args_info->output_module_arg), + &(args_info->output_module_orig), &(args_info->output_module_given), + &(local_args_info.output_module_given), optarg, 0, "simple_file", ARG_STRING, + check_ambiguity, override, 0, 0, + "output-module", 'O', + additional_error)) + goto failure; + + break; + case 'C': /* Read a configuration file, which can specify any of these options. */ + + + if (update_arg( (void *)&(args_info->config_arg), + &(args_info->config_orig), &(args_info->config_given), + &(local_args_info.config_given), optarg, 0, "/etc/zmap/zmap.conf", ARG_STRING, + check_ambiguity, override, 0, 0, + "config", 'C', + additional_error)) + goto failure; + + break; + case 'q': /* Do not print status updates. */ + + + if (update_arg( 0 , + 0 , &(args_info->quiet_given), + &(local_args_info.quiet_given), optarg, 0, 0, ARG_NO, + check_ambiguity, override, 0, 0, + "quiet", 'q', + additional_error)) + goto failure; + + break; + case 'g': /* Print configuration and summary at end of scan. */ + + + if (update_arg( 0 , + 0 , &(args_info->summary_given), + &(local_args_info.summary_given), optarg, 0, 0, ARG_NO, + check_ambiguity, override, 0, 0, + "summary", 'g', + additional_error)) + goto failure; + + break; + case 'v': /* Level of log detail (0-5). */ + + + if (update_arg( (void *)&(args_info->verbosity_arg), + &(args_info->verbosity_orig), &(args_info->verbosity_given), + &(local_args_info.verbosity_given), optarg, 0, "3", ARG_INT, + check_ambiguity, override, 0, 0, + "verbosity", 'v', + additional_error)) + goto failure; + + break; + case 'h': /* Print help and exit. */ + + + if (update_arg( 0 , + 0 , &(args_info->help_given), + &(local_args_info.help_given), optarg, 0, 0, ARG_NO, + check_ambiguity, override, 0, 0, + "help", 'h', + additional_error)) + goto failure; + + break; + case 'V': /* Print version and exit. */ + + + if (update_arg( 0 , + 0 , &(args_info->version_given), + &(local_args_info.version_given), optarg, 0, 0, ARG_NO, + check_ambiguity, override, 0, 0, + "version", 'V', + additional_error)) + goto failure; + + break; + + case 0: /* Long option with no short option */ + /* Arguments to pass to probe module. */ + if (strcmp (long_options[option_index].name, "probe-args") == 0) + { + + + if (update_arg( (void *)&(args_info->probe_args_arg), + &(args_info->probe_args_orig), &(args_info->probe_args_given), + &(local_args_info.probe_args_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "probe-args", '-', + additional_error)) + goto failure; + + } + /* Arguments to pass to output module. */ + else if (strcmp (long_options[option_index].name, "output-args") == 0) + { + + + if (update_arg( (void *)&(args_info->output_args_arg), + &(args_info->output_args_orig), &(args_info->output_args_given), + &(local_args_info.output_args_given), optarg, 0, 0, ARG_STRING, + check_ambiguity, override, 0, 0, + "output-args", '-', + additional_error)) + goto failure; + + } + /* List available output modules. */ + else if (strcmp (long_options[option_index].name, "list-output-modules") == 0) + { + + + if (update_arg( 0 , + 0 , &(args_info->list_output_modules_given), + &(local_args_info.list_output_modules_given), optarg, 0, 0, ARG_NO, + check_ambiguity, override, 0, 0, + "list-output-modules", '-', + additional_error)) + goto failure; + + } + /* List available probe modules. */ + else if (strcmp (long_options[option_index].name, "list-probe-modules") == 0) + { + + + if (update_arg( 0 , + 0 , &(args_info->list_probe_modules_given), + &(local_args_info.list_probe_modules_given), optarg, 0, 0, ARG_NO, + check_ambiguity, override, 0, 0, + "list-probe-modules", '-', + additional_error)) + goto failure; + + } + + break; + case '?': /* Invalid option. */ + /* `getopt_long' already printed an error message. */ + goto failure; + + default: /* bug: option not considered. */ + fprintf (stderr, "%s: option unknown: %c%s\n", CMDLINE_PARSER_PACKAGE, c, (additional_error ? additional_error : "")); + abort (); + } /* switch */ + } /* while */ + + + + + cmdline_parser_release (&local_args_info); + + if ( error ) + return (EXIT_FAILURE); + + return 0; + +failure: + + cmdline_parser_release (&local_args_info); + return (EXIT_FAILURE); +} + +#ifndef CONFIG_FILE_LINE_SIZE +#define CONFIG_FILE_LINE_SIZE 2048 +#endif +#define ADDITIONAL_ERROR " in configuration file " + +#define CONFIG_FILE_LINE_BUFFER_SIZE (CONFIG_FILE_LINE_SIZE+3) +/* 3 is for "--" and "=" */ + +static int +_cmdline_parser_configfile (const char *filename, int *my_argc) +{ + FILE* file; + char my_argv[CONFIG_FILE_LINE_BUFFER_SIZE+1]; + char linebuf[CONFIG_FILE_LINE_SIZE]; + int line_num = 0; + int result = 0, equal; + char *fopt, *farg; + char *str_index; + size_t len, next_token; + char delimiter; + + if ((file = fopen(filename, "r")) == 0) + { + fprintf (stderr, "%s: Error opening configuration file '%s'\n", + CMDLINE_PARSER_PACKAGE, filename); + return EXIT_FAILURE; + } + + while ((fgets(linebuf, CONFIG_FILE_LINE_SIZE, file)) != 0) + { + ++line_num; + my_argv[0] = '\0'; + len = strlen(linebuf); + if (len > (CONFIG_FILE_LINE_BUFFER_SIZE-1)) + { + fprintf (stderr, "%s:%s:%d: Line too long in configuration file\n", + CMDLINE_PARSER_PACKAGE, filename, line_num); + result = EXIT_FAILURE; + break; + } + + /* find first non-whitespace character in the line */ + next_token = strspn (linebuf, " \t\r\n"); + str_index = linebuf + next_token; + + if ( str_index[0] == '\0' || str_index[0] == '#') + continue; /* empty line or comment line is skipped */ + + fopt = str_index; + + /* truncate fopt at the end of the first non-valid character */ + next_token = strcspn (fopt, " \t\r\n="); + + if (fopt[next_token] == '\0') /* the line is over */ + { + farg = 0; + equal = 0; + goto noarg; + } + + /* remember if equal sign is present */ + equal = (fopt[next_token] == '='); + fopt[next_token++] = '\0'; + + /* advance pointers to the next token after the end of fopt */ + next_token += strspn (fopt + next_token, " \t\r\n"); + + /* check for the presence of equal sign, and if so, skip it */ + if ( !equal ) + if ((equal = (fopt[next_token] == '='))) + { + next_token++; + next_token += strspn (fopt + next_token, " \t\r\n"); + } + str_index += next_token; + + /* find argument */ + farg = str_index; + if ( farg[0] == '\"' || farg[0] == '\'' ) + { /* quoted argument */ + str_index = strchr (++farg, str_index[0] ); /* skip opening quote */ + if (! str_index) + { + fprintf + (stderr, + "%s:%s:%d: unterminated string in configuration file\n", + CMDLINE_PARSER_PACKAGE, filename, line_num); + result = EXIT_FAILURE; + break; + } + } + else + { /* read up the remaining part up to a delimiter */ + next_token = strcspn (farg, " \t\r\n#\'\""); + str_index += next_token; + } + + /* truncate farg at the delimiter and store it for further check */ + delimiter = *str_index, *str_index++ = '\0'; + + /* everything but comment is illegal at the end of line */ + if (delimiter != '\0' && delimiter != '#') + { + str_index += strspn(str_index, " \t\r\n"); + if (*str_index != '\0' && *str_index != '#') + { + fprintf + (stderr, + "%s:%s:%d: malformed string in configuration file\n", + CMDLINE_PARSER_PACKAGE, filename, line_num); + result = EXIT_FAILURE; + break; + } + } + + noarg: + if (!strcmp(fopt,"include")) { + if (farg && *farg) { + result = _cmdline_parser_configfile(farg, my_argc); + } else { + fprintf(stderr, "%s:%s:%d: include requires a filename argument.\n", + CMDLINE_PARSER_PACKAGE, filename, line_num); + } + continue; + } + len = strlen(fopt); + strcat (my_argv, len > 1 ? "--" : "-"); + strcat (my_argv, fopt); + if (len > 1 && ((farg && *farg) || equal)) + strcat (my_argv, "="); + if (farg && *farg) + strcat (my_argv, farg); + ++(*my_argc); + + cmd_line_list_tmp = (struct line_list *) malloc (sizeof (struct line_list)); + cmd_line_list_tmp->next = cmd_line_list; + cmd_line_list = cmd_line_list_tmp; + cmd_line_list->string_arg = gengetopt_strdup(my_argv); + } /* while */ + + if (file) + fclose(file); + return result; +} + +int +cmdline_parser_configfile ( + const char *filename, + struct gengetopt_args_info *args_info, + int override, int initialize, int check_required) +{ + struct cmdline_parser_params params; + + params.override = override; + params.initialize = initialize; + params.check_required = check_required; + params.check_ambiguity = 0; + params.print_errors = 1; + + return cmdline_parser_config_file (filename, args_info, ¶ms); +} + +int +cmdline_parser_config_file (const char *filename, + struct gengetopt_args_info *args_info, + struct cmdline_parser_params *params) +{ + int i, result; + int my_argc = 1; + char **my_argv_arg; + char *additional_error; + + /* store the program name */ + cmd_line_list_tmp = (struct line_list *) malloc (sizeof (struct line_list)); + cmd_line_list_tmp->next = cmd_line_list; + cmd_line_list = cmd_line_list_tmp; + cmd_line_list->string_arg = gengetopt_strdup (CMDLINE_PARSER_PACKAGE); + + result = _cmdline_parser_configfile(filename, &my_argc); + + if (result != EXIT_FAILURE) { + my_argv_arg = (char **) malloc((my_argc+1) * sizeof(char *)); + cmd_line_list_tmp = cmd_line_list; + + for (i = my_argc - 1; i >= 0; --i) { + my_argv_arg[i] = cmd_line_list_tmp->string_arg; + cmd_line_list_tmp = cmd_line_list_tmp->next; + } + + my_argv_arg[my_argc] = 0; + + additional_error = (char *)malloc(strlen(filename) + strlen(ADDITIONAL_ERROR) + 1); + strcpy (additional_error, ADDITIONAL_ERROR); + strcat (additional_error, filename); + result = + cmdline_parser_internal (my_argc, my_argv_arg, args_info, + params, + additional_error); + + free (additional_error); + free (my_argv_arg); + } + + free_cmd_list(); + if (result == EXIT_FAILURE) + { + cmdline_parser_free (args_info); + exit (EXIT_FAILURE); + } + + return result; +} diff --git a/src/zopt.ggo b/src/zopt.ggo new file mode 100644 index 0000000..b52a4e1 --- /dev/null +++ b/src/zopt.ggo @@ -0,0 +1,119 @@ +# ZMap Copyright 2013 Regents of the University of Michigan + +# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 + +# zmap option description to be processed by gengetopt + +package "zmap" +version "1.0.0" +purpose "A fast Internet-wide scanner." + +section "Basic arguments" + +option "target-port" p "TCP port number to scan (for SYN scans)" + typestr="port" + optional int +option "output-file" o "Output file" + typestr="name" + optional string +option "blacklist-file" b "File of subnets to exclude, in CIDR notation, e.g. 192.168.0.0/16" + typestr="path" + optional string +option "whitelist-file" w "File of subnets to constrain scan to, in CIDR notation, e.g. 192.168.0.0/16" + typestr="path" + optional string + +section "Scan options" + +option "max-targets" n "Cap number of targets to probe (as a number or a percentage of the address space)" + typestr="n" + optional string +option "max-results" N "Cap number of results to return" + typestr="n" + optional int +option "max-runtime" t "Cap length of time for sending packets" + typestr="ses" + optional int +option "rate" r "Set send rate in packets/sec" + typestr="pps" + optional int +option "bandwidth" B "Set send rate in bits/second (supports suffixes G, M and K)" + typestr="bps" + optional string +option "cooldown-time" c "How long to continue receiving after sending last probe" + typestr="secs" + default="8" + optional int +option "seed" e "Seed used to select address permutation" + typestr="n" + optional int +option "sender-threads" T "Threads used to send packets" + typestr="n" + default="1" + optional int +option "probes" P "Number of probes to send to each IP" + typestr="n" + default="1" + optional int +option "dryrun" d "Don't actually send packets" + optional + +section "Network options" + +option "source-port" s "Source port(s) for scan packets" + typestr="port|range" + optional string +option "source-ip" S "Source address(es) for scan packets" + typestr="ip|range" + optional string +option "gateway-mac" G "Specify gateway MAC address" + typestr="addr" + optional string +option "interface" i "Specify network interface to use" + typestr="name" + optional string + +section "Advanced options" + +option "probe-module" M "Select probe module" + typestr="name" + default="tcp_synscan" + optional string +option "output-module" O "Select output module" + typestr="name" + default="simple_file" + optional string +option "probe-args" - "Arguments to pass to probe module" + typestr="args" + optional string +option "output-args" - "Arguments to pass to output module" + typestr="args" + optional string +option "list-output-modules" - "List available output modules" + optional +option "list-probe-modules" - "List available probe modules" + optional + +section "Additional options" + +option "config" C "Read a configuration file, which can specify any of these options" + typestr="filename" + default="/etc/zmap/zmap.conf" + optional string +option "quiet" q "Do not print status updates" + optional +option "summary" g "Print configuration and summary at end of scan" + optional +option "verbosity" v "Level of log detail (0-5)" + typestr="n" + default="3" + optional int +option "help" h "Print help and exit" + optional +option "version" V "Print version and exit" + optional + +text "\nExamples:\n\ + zmap -p 443 (scans the whole Internet for hosts with port 443 open)\n\ + zmap -N 5 -B 10M -p 80 -o - (find 5 HTTP servers, scanning at 10 Mb/s)" + diff --git a/src/zopt.h b/src/zopt.h new file mode 100644 index 0000000..30f30ee --- /dev/null +++ b/src/zopt.h @@ -0,0 +1,298 @@ +/** @file zopt.h + * @brief The header file for the command line option parser + * generated by GNU Gengetopt version 2.22.5 + * http://www.gnu.org/software/gengetopt. + * DO NOT modify this file, since it can be overwritten + * @author GNU Gengetopt by Lorenzo Bettini */ + +#ifndef ZOPT_H +#define ZOPT_H + +/* If we use autoconf. */ +#ifdef HAVE_CONFIG_H +#include "config.h" +#endif + +#include /* for FILE */ + +#ifdef __cplusplus +extern "C" { +#endif /* __cplusplus */ + +#ifndef CMDLINE_PARSER_PACKAGE +/** @brief the program name (used for printing errors) */ +#define CMDLINE_PARSER_PACKAGE "zmap" +#endif + +#ifndef CMDLINE_PARSER_PACKAGE_NAME +/** @brief the complete program name (used for help and version) */ +#define CMDLINE_PARSER_PACKAGE_NAME "zmap" +#endif + +#ifndef CMDLINE_PARSER_VERSION +/** @brief the program version */ +#define CMDLINE_PARSER_VERSION "1.0.0" +#endif + +/** @brief Where the command line options are stored */ +struct gengetopt_args_info +{ + int target_port_arg; /**< @brief TCP port number to scan (for SYN scans). */ + char * target_port_orig; /**< @brief TCP port number to scan (for SYN scans) original value given at command line. */ + const char *target_port_help; /**< @brief TCP port number to scan (for SYN scans) help description. */ + char * output_file_arg; /**< @brief Output file. */ + char * output_file_orig; /**< @brief Output file original value given at command line. */ + const char *output_file_help; /**< @brief Output file help description. */ + char * blacklist_file_arg; /**< @brief File of subnets to exclude, in CIDR notation, e.g. 192.168.0.0/16. */ + char * blacklist_file_orig; /**< @brief File of subnets to exclude, in CIDR notation, e.g. 192.168.0.0/16 original value given at command line. */ + const char *blacklist_file_help; /**< @brief File of subnets to exclude, in CIDR notation, e.g. 192.168.0.0/16 help description. */ + char * whitelist_file_arg; /**< @brief File of subnets to constrain scan to, in CIDR notation, e.g. 192.168.0.0/16. */ + char * whitelist_file_orig; /**< @brief File of subnets to constrain scan to, in CIDR notation, e.g. 192.168.0.0/16 original value given at command line. */ + const char *whitelist_file_help; /**< @brief File of subnets to constrain scan to, in CIDR notation, e.g. 192.168.0.0/16 help description. */ + char * max_targets_arg; /**< @brief Cap number of targets to probe (as a number or a percentage of the address space). */ + char * max_targets_orig; /**< @brief Cap number of targets to probe (as a number or a percentage of the address space) original value given at command line. */ + const char *max_targets_help; /**< @brief Cap number of targets to probe (as a number or a percentage of the address space) help description. */ + int max_results_arg; /**< @brief Cap number of results to return. */ + char * max_results_orig; /**< @brief Cap number of results to return original value given at command line. */ + const char *max_results_help; /**< @brief Cap number of results to return help description. */ + int max_runtime_arg; /**< @brief Cap length of time for sending packets. */ + char * max_runtime_orig; /**< @brief Cap length of time for sending packets original value given at command line. */ + const char *max_runtime_help; /**< @brief Cap length of time for sending packets help description. */ + int rate_arg; /**< @brief Set send rate in packets/sec. */ + char * rate_orig; /**< @brief Set send rate in packets/sec original value given at command line. */ + const char *rate_help; /**< @brief Set send rate in packets/sec help description. */ + char * bandwidth_arg; /**< @brief Set send rate in bits/second (supports suffixes G, M and K). */ + char * bandwidth_orig; /**< @brief Set send rate in bits/second (supports suffixes G, M and K) original value given at command line. */ + const char *bandwidth_help; /**< @brief Set send rate in bits/second (supports suffixes G, M and K) help description. */ + int cooldown_time_arg; /**< @brief How long to continue receiving after sending last probe (default='8'). */ + char * cooldown_time_orig; /**< @brief How long to continue receiving after sending last probe original value given at command line. */ + const char *cooldown_time_help; /**< @brief How long to continue receiving after sending last probe help description. */ + int seed_arg; /**< @brief Seed used to select address permutation. */ + char * seed_orig; /**< @brief Seed used to select address permutation original value given at command line. */ + const char *seed_help; /**< @brief Seed used to select address permutation help description. */ + int sender_threads_arg; /**< @brief Threads used to send packets (default='1'). */ + char * sender_threads_orig; /**< @brief Threads used to send packets original value given at command line. */ + const char *sender_threads_help; /**< @brief Threads used to send packets help description. */ + int probes_arg; /**< @brief Number of probes to send to each IP (default='1'). */ + char * probes_orig; /**< @brief Number of probes to send to each IP original value given at command line. */ + const char *probes_help; /**< @brief Number of probes to send to each IP help description. */ + const char *dryrun_help; /**< @brief Don't actually send packets help description. */ + char * source_port_arg; /**< @brief Source port(s) for scan packets. */ + char * source_port_orig; /**< @brief Source port(s) for scan packets original value given at command line. */ + const char *source_port_help; /**< @brief Source port(s) for scan packets help description. */ + char * source_ip_arg; /**< @brief Source address(es) for scan packets. */ + char * source_ip_orig; /**< @brief Source address(es) for scan packets original value given at command line. */ + const char *source_ip_help; /**< @brief Source address(es) for scan packets help description. */ + char * gateway_mac_arg; /**< @brief Specify gateway MAC address. */ + char * gateway_mac_orig; /**< @brief Specify gateway MAC address original value given at command line. */ + const char *gateway_mac_help; /**< @brief Specify gateway MAC address help description. */ + char * interface_arg; /**< @brief Specify network interface to use. */ + char * interface_orig; /**< @brief Specify network interface to use original value given at command line. */ + const char *interface_help; /**< @brief Specify network interface to use help description. */ + char * probe_module_arg; /**< @brief Select probe module (default='tcp_synscan'). */ + char * probe_module_orig; /**< @brief Select probe module original value given at command line. */ + const char *probe_module_help; /**< @brief Select probe module help description. */ + char * output_module_arg; /**< @brief Select output module (default='simple_file'). */ + char * output_module_orig; /**< @brief Select output module original value given at command line. */ + const char *output_module_help; /**< @brief Select output module help description. */ + char * probe_args_arg; /**< @brief Arguments to pass to probe module. */ + char * probe_args_orig; /**< @brief Arguments to pass to probe module original value given at command line. */ + const char *probe_args_help; /**< @brief Arguments to pass to probe module help description. */ + char * output_args_arg; /**< @brief Arguments to pass to output module. */ + char * output_args_orig; /**< @brief Arguments to pass to output module original value given at command line. */ + const char *output_args_help; /**< @brief Arguments to pass to output module help description. */ + const char *list_output_modules_help; /**< @brief List available output modules help description. */ + const char *list_probe_modules_help; /**< @brief List available probe modules help description. */ + char * config_arg; /**< @brief Read a configuration file, which can specify any of these options (default='/etc/zmap/zmap.conf'). */ + char * config_orig; /**< @brief Read a configuration file, which can specify any of these options original value given at command line. */ + const char *config_help; /**< @brief Read a configuration file, which can specify any of these options help description. */ + const char *quiet_help; /**< @brief Do not print status updates help description. */ + const char *summary_help; /**< @brief Print configuration and summary at end of scan help description. */ + int verbosity_arg; /**< @brief Level of log detail (0-5) (default='3'). */ + char * verbosity_orig; /**< @brief Level of log detail (0-5) original value given at command line. */ + const char *verbosity_help; /**< @brief Level of log detail (0-5) help description. */ + const char *help_help; /**< @brief Print help and exit help description. */ + const char *version_help; /**< @brief Print version and exit help description. */ + + unsigned int target_port_given ; /**< @brief Whether target-port was given. */ + unsigned int output_file_given ; /**< @brief Whether output-file was given. */ + unsigned int blacklist_file_given ; /**< @brief Whether blacklist-file was given. */ + unsigned int whitelist_file_given ; /**< @brief Whether whitelist-file was given. */ + unsigned int max_targets_given ; /**< @brief Whether max-targets was given. */ + unsigned int max_results_given ; /**< @brief Whether max-results was given. */ + unsigned int max_runtime_given ; /**< @brief Whether max-runtime was given. */ + unsigned int rate_given ; /**< @brief Whether rate was given. */ + unsigned int bandwidth_given ; /**< @brief Whether bandwidth was given. */ + unsigned int cooldown_time_given ; /**< @brief Whether cooldown-time was given. */ + unsigned int seed_given ; /**< @brief Whether seed was given. */ + unsigned int sender_threads_given ; /**< @brief Whether sender-threads was given. */ + unsigned int probes_given ; /**< @brief Whether probes was given. */ + unsigned int dryrun_given ; /**< @brief Whether dryrun was given. */ + unsigned int source_port_given ; /**< @brief Whether source-port was given. */ + unsigned int source_ip_given ; /**< @brief Whether source-ip was given. */ + unsigned int gateway_mac_given ; /**< @brief Whether gateway-mac was given. */ + unsigned int interface_given ; /**< @brief Whether interface was given. */ + unsigned int probe_module_given ; /**< @brief Whether probe-module was given. */ + unsigned int output_module_given ; /**< @brief Whether output-module was given. */ + unsigned int probe_args_given ; /**< @brief Whether probe-args was given. */ + unsigned int output_args_given ; /**< @brief Whether output-args was given. */ + unsigned int list_output_modules_given ; /**< @brief Whether list-output-modules was given. */ + unsigned int list_probe_modules_given ; /**< @brief Whether list-probe-modules was given. */ + unsigned int config_given ; /**< @brief Whether config was given. */ + unsigned int quiet_given ; /**< @brief Whether quiet was given. */ + unsigned int summary_given ; /**< @brief Whether summary was given. */ + unsigned int verbosity_given ; /**< @brief Whether verbosity was given. */ + unsigned int help_given ; /**< @brief Whether help was given. */ + unsigned int version_given ; /**< @brief Whether version was given. */ + +} ; + +/** @brief The additional parameters to pass to parser functions */ +struct cmdline_parser_params +{ + int override; /**< @brief whether to override possibly already present options (default 0) */ + int initialize; /**< @brief whether to initialize the option structure gengetopt_args_info (default 1) */ + int check_required; /**< @brief whether to check that all required options were provided (default 1) */ + int check_ambiguity; /**< @brief whether to check for options already specified in the option structure gengetopt_args_info (default 0) */ + int print_errors; /**< @brief whether getopt_long should print an error message for a bad option (default 1) */ +} ; + +/** @brief the purpose string of the program */ +extern const char *gengetopt_args_info_purpose; +/** @brief the usage string of the program */ +extern const char *gengetopt_args_info_usage; +/** @brief all the lines making the help output */ +extern const char *gengetopt_args_info_help[]; + +/** + * The command line parser + * @param argc the number of command line options + * @param argv the command line options + * @param args_info the structure where option information will be stored + * @return 0 if everything went fine, NON 0 if an error took place + */ +int cmdline_parser (int argc, char **argv, + struct gengetopt_args_info *args_info); + +/** + * The command line parser (version with additional parameters - deprecated) + * @param argc the number of command line options + * @param argv the command line options + * @param args_info the structure where option information will be stored + * @param override whether to override possibly already present options + * @param initialize whether to initialize the option structure my_args_info + * @param check_required whether to check that all required options were provided + * @return 0 if everything went fine, NON 0 if an error took place + * @deprecated use cmdline_parser_ext() instead + */ +int cmdline_parser2 (int argc, char **argv, + struct gengetopt_args_info *args_info, + int override, int initialize, int check_required); + +/** + * The command line parser (version with additional parameters) + * @param argc the number of command line options + * @param argv the command line options + * @param args_info the structure where option information will be stored + * @param params additional parameters for the parser + * @return 0 if everything went fine, NON 0 if an error took place + */ +int cmdline_parser_ext (int argc, char **argv, + struct gengetopt_args_info *args_info, + struct cmdline_parser_params *params); + +/** + * Save the contents of the option struct into an already open FILE stream. + * @param outfile the stream where to dump options + * @param args_info the option struct to dump + * @return 0 if everything went fine, NON 0 if an error took place + */ +int cmdline_parser_dump(FILE *outfile, + struct gengetopt_args_info *args_info); + +/** + * Save the contents of the option struct into a (text) file. + * This file can be read by the config file parser (if generated by gengetopt) + * @param filename the file where to save + * @param args_info the option struct to save + * @return 0 if everything went fine, NON 0 if an error took place + */ +int cmdline_parser_file_save(const char *filename, + struct gengetopt_args_info *args_info); + +/** + * Print the help + */ +void cmdline_parser_print_help(void); +/** + * Print the version + */ +void cmdline_parser_print_version(void); + +/** + * Initializes all the fields a cmdline_parser_params structure + * to their default values + * @param params the structure to initialize + */ +void cmdline_parser_params_init(struct cmdline_parser_params *params); + +/** + * Allocates dynamically a cmdline_parser_params structure and initializes + * all its fields to their default values + * @return the created and initialized cmdline_parser_params structure + */ +struct cmdline_parser_params *cmdline_parser_params_create(void); + +/** + * Initializes the passed gengetopt_args_info structure's fields + * (also set default values for options that have a default) + * @param args_info the structure to initialize + */ +void cmdline_parser_init (struct gengetopt_args_info *args_info); +/** + * Deallocates the string fields of the gengetopt_args_info structure + * (but does not deallocate the structure itself) + * @param args_info the structure to deallocate + */ +void cmdline_parser_free (struct gengetopt_args_info *args_info); + +/** + * The config file parser (deprecated version) + * @param filename the name of the config file + * @param args_info the structure where option information will be stored + * @param override whether to override possibly already present options + * @param initialize whether to initialize the option structure my_args_info + * @param check_required whether to check that all required options were provided + * @return 0 if everything went fine, NON 0 if an error took place + * @deprecated use cmdline_parser_config_file() instead + */ +int cmdline_parser_configfile (const char *filename, + struct gengetopt_args_info *args_info, + int override, int initialize, int check_required); + +/** + * The config file parser + * @param filename the name of the config file + * @param args_info the structure where option information will be stored + * @param params additional parameters for the parser + * @return 0 if everything went fine, NON 0 if an error took place + */ +int cmdline_parser_config_file (const char *filename, + struct gengetopt_args_info *args_info, + struct cmdline_parser_params *params); + +/** + * Checks that all the required options were specified + * @param args_info the structure to check + * @param prog_name the name of the program that will be used to print + * possible errors + * @return + */ +int cmdline_parser_required (struct gengetopt_args_info *args_info, + const char *prog_name); + + +#ifdef __cplusplus +} +#endif /* __cplusplus */ +#endif /* ZOPT_H */