diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-17 07:42:04 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-17 07:42:04 +0000 |
commit | 0d47952611198ef6b1163f366dc03922d20b1475 (patch) | |
tree | 3d840a3b8c0daef0754707bfb9f5e873b6b1ac13 /nsock/examples | |
parent | Initial commit. (diff) | |
download | nmap-0d47952611198ef6b1163f366dc03922d20b1475.tar.xz nmap-0d47952611198ef6b1163f366dc03922d20b1475.zip |
Adding upstream version 7.94+git20230807.3be01efb1+dfsg.upstream/7.94+git20230807.3be01efb1+dfsgupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'nsock/examples')
-rw-r--r-- | nsock/examples/Makefile | 29 | ||||
-rw-r--r-- | nsock/examples/README | 9 | ||||
-rw-r--r-- | nsock/examples/nsock_telnet.c | 259 | ||||
-rw-r--r-- | nsock/examples/nsock_test_timers.c | 162 |
4 files changed, 459 insertions, 0 deletions
diff --git a/nsock/examples/Makefile b/nsock/examples/Makefile new file mode 100644 index 0000000..de40593 --- /dev/null +++ b/nsock/examples/Makefile @@ -0,0 +1,29 @@ +CC = gcc +CCOPT = +DEFS = +NBASEDIR=../../nbase +NSOCKLIB=../src/libnsock.a +NBASELIB=$(NBASEDIR)/libnbase.a +OPENSSLLIB=-lssl -lpcap -lcrypto +INCLS = -I../include -I$(NBASEDIR) +CFLAGS = -I/usr/local/include -Wall -g $(CCOPT) $(DEFS) $(INCLS) +LDFLAGS = +PCAPBASEDIR=../../libpcap +PCAPLIB=$(PCAPBASEDIR)/libpcap.a +RM = rm -f + +TARGETS = nsock_test_timers nsock_telnet + +all: $(TARGETS) + +nsock_telnet: nsock_telnet.o $(NSOCKLIB) + $(CC) -o $@ $(CFLAGS) nsock_telnet.o $(NSOCKLIB) $(NBASELIB) $(OPENSSLLIB) + +nsock_test_timers: nsock_test_timers.o $(NSOCKLIB) + $(CC) -o $@ $(CFLAGS) nsock_test_timers.o $(NSOCKLIB) $(NBASELIB) $(OPENSSLLIB) + +nsock_pcap: nsock_pcap.o $(NSOCKLIB) $(PCAPLIB) + $(CC) -o $@ $(CFLAGS) nsock_pcap.o $(NSOCKLIB) $(NBASELIB) $(OPENSSLLIB) $(PCAPLIB) + +clean: + $(RM) *.o $(TARGETS) diff --git a/nsock/examples/README b/nsock/examples/README new file mode 100644 index 0000000..d3c66f5 --- /dev/null +++ b/nsock/examples/README @@ -0,0 +1,9 @@ +$Id$ + +Here are some example programs, they weren't really written to +present nice and elegant use of the nsock library. They were +actually written for my testing purposes during development. So don't +be surprised if you see code that looks strange, or even downright +evil -- I'm testing the code reliability. + +-Fyodor
\ No newline at end of file diff --git a/nsock/examples/nsock_telnet.c b/nsock/examples/nsock_telnet.c new file mode 100644 index 0000000..cb2a7c0 --- /dev/null +++ b/nsock/examples/nsock_telnet.c @@ -0,0 +1,259 @@ +/*************************************************************************** + * nsock_telnet.c -- A simple "telnet" client -- a trivial example of * + * using the nsock parallel socket event library * + * * + ***********************IMPORTANT NSOCK LICENSE TERMS*********************** + * + * The nsock parallel socket event library is (C) 1999-2023 Nmap Software LLC + * This library is free software; you may redistribute and/or modify it under + * the terms of the GNU General Public License as published by the Free Software + * Foundation; Version 2. This guarantees your right to use, modify, and + * redistribute this software under certain conditions. If this license is + * unacceptable to you, Nmap Software LLC may be willing to sell alternative + * licenses (contact sales@nmap.com ). + * + * As a special exception to the GPL terms, Nmap Software LLC grants permission + * to link the code of this program with any version of the OpenSSL library + * which is distributed under a license identical to that listed in the included + * docs/licenses/OpenSSL.txt file, and distribute linked combinations including + * the two. You must obey the GNU GPL in all respects for all of the code used + * other than OpenSSL. If you modify this file, you may extend this exception to + * your version of the file, but you are not obligated to do so. + * + * If you received these files with a written license agreement stating terms + * other than the (GPL) terms above, then that alternative license agreement + * takes precedence over this comment. + * + * Source is provided to this software because we believe users have a right to + * know exactly what a program is going to do before they run it. This also + * allows you to audit the software for security holes. + * + * Source code also allows you to port Nmap to new platforms, fix bugs, and add + * new features. You are highly encouraged to send your changes to the + * dev@nmap.org mailing list for possible incorporation into the main + * distribution. By sending these changes to Fyodor or one of the Insecure.Org + * development mailing lists, or checking them into the Nmap source code + * repository, it is understood (unless you specify otherwise) that you are + * offering the Nmap Project (Nmap Software LLC) the unlimited, non-exclusive + * right to reuse, modify, and relicense the code. Nmap will always be available + * Open Source, but this is important because the inability to relicense code + * has caused devastating problems for other Free Software projects (such as KDE + * and NASM). We also occasionally relicense the code to third parties as + * discussed above. If you wish to specify special license conditions of your + * contributions, just say so when you send them. + * + * This program is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU General Public License v2.0 for more + * details (http://www.gnu.org/licenses/gpl-2.0.html). + * + ***************************************************************************/ + + +#include "nsock.h" + +#include <stdio.h> +#include <stdlib.h> +#include <sys/socket.h> +#include <netinet/in.h> +#include <arpa/inet.h> +#include <netdb.h> +#include <unistd.h> +#include <stdlib.h> +#include <string.h> +#include <stdio.h> +#include <assert.h> +#include <sys/time.h> + +#include <openssl/ssl.h> +#include <openssl/err.h> +/* #include <nbase.h> */ + +/* from nbase.h */ +int socket_errno(); + + +extern char *optarg; + +extern int optind; + +struct telnet_state { + nsock_iod tcp_nsi; + nsock_iod stdin_nsi; + nsock_event_id latest_readtcpev; + nsock_event_id latest_readstdinev; + void *ssl_session; +}; + +/* Tries to resolve given hostname and stores + result in ip . returns 0 if hostname cannot + be resolved */ +int resolve(char *hostname, struct in_addr *ip) { + struct hostent *h; + + if (!hostname || !*hostname) { + fprintf(stderr, "NULL or zero-length hostname passed to resolve(). Quitting.\n"); + exit(1); + } + + if (inet_aton(hostname, ip)) + return 1; /* damn, that was easy ;) */ + if ((h = gethostbyname(hostname))) { + memcpy(ip, h->h_addr_list[0], sizeof(struct in_addr)); + return 1; + } + return 0; +} + +void telnet_event_handler(nsock_pool nsp, nsock_event nse, void *mydata) { + nsock_iod nsi = nse_iod(nse); + enum nse_status status = nse_status(nse); + enum nse_type type = nse_type(nse); + struct sockaddr_in peer; + struct telnet_state *ts; + int nbytes; + char *str; + int read_timeout = -1; + int write_timeout = 2000; + ts = (struct telnet_state *)mydata; + + printf("telnet_event_handler: Received callback of type %s with status %s\n", nse_type2str(type), nse_status2str(status)); + + if (status == NSE_STATUS_SUCCESS) { + switch (type) { + case NSE_TYPE_CONNECT: + case NSE_TYPE_CONNECT_SSL: + nsock_iod_get_communication_info(nsi, NULL, NULL, NULL, (struct sockaddr *)&peer, sizeof peer); + printf("Successfully connected %sto %s:%hu -- start typing lines\n", (type == NSE_TYPE_CONNECT_SSL) ? "(SSL!) " : "", inet_ntoa(peer.sin_addr), ntohs(peer.sin_port)); + /* First of all, lets add STDIN to our list of watched filehandles */ + if ((ts->stdin_nsi = nsock_iod_new2(nsp, STDIN_FILENO, NULL)) == NULL) { + fprintf(stderr, "Failed to create stdin msi\n"); + exit(1); + } + + /* Now lets read from stdin and the network, line buffered (by nsock) */ + ts->latest_readtcpev = nsock_readlines(nsp, ts->tcp_nsi, telnet_event_handler, read_timeout, ts, 1); + ts->latest_readstdinev = nsock_readlines(nsp, ts->stdin_nsi, telnet_event_handler, read_timeout, ts, 1); + break; + case NSE_TYPE_READ: + str = nse_readbuf(nse, &nbytes); + if (nsi == ts->tcp_nsi) { + printf("%s", str); + /* printf("Read from tcp socket (%d bytes):\n%s", nbytes, str); */ + ts->latest_readtcpev = nsock_readlines(nsp, ts->tcp_nsi, telnet_event_handler, read_timeout, ts, 1); + } else { + /* printf("Read from stdin (%d bytes):\n%s", nbytes, str); */ + nsock_write(nsp, ts->tcp_nsi, telnet_event_handler, write_timeout, ts, str, nbytes); + ts->latest_readstdinev = nsock_readlines(nsp, ts->stdin_nsi, telnet_event_handler, read_timeout, ts, 1); + } + break; + case NSE_TYPE_WRITE: + /* Nothing to do, really */ + break; + case NSE_TYPE_TIMER: + break; + default: + fprintf(stderr, "telnet_event_handler: Got bogus type -- quitting\n"); + exit(1); + break; + } + } else if (status == NSE_STATUS_EOF) { + printf("Got EOF from %s\nCancelling outstanding readevents.\n", (nsi == ts->tcp_nsi) ? "tcp socket" : "stdin"); + /* One of these is the event I am currently handling! But I wanted to + be evil when testing this out... */ + if (nsock_event_cancel(nsp, ts->latest_readtcpev, 1) != 0) { + printf("Cancelled tcp event: %li\n", ts->latest_readtcpev); + } + if (nsock_event_cancel(nsp, ts->latest_readstdinev, 1) != 0) { + printf("Cancelled stdin event: %li\n", ts->latest_readstdinev); + } + } else if (status == NSE_STATUS_ERROR) { + if (nsock_iod_check_ssl(nsi)) { + printf("SSL %s failed: %s\n", nse_type2str(type), ERR_error_string(ERR_get_error(), NULL)); + } else { + int err; + + err = nse_errorcode(nse); + printf("%s failed: (%d) %s\n", nse_type2str(type), err, strerror(err)); + } + } + return; +} + +void usage() { + fprintf(stderr, "\nUsage: nsock_telnet [-s] <hostnameorip> [portnum]\n" " Where -s enables SSL for the connection\n\n"); + exit(1); +} + +int main(int argc, char *argv[]) { + struct in_addr target; + nsock_pool nsp; + nsock_event_id ev; + unsigned short portno; + enum nsock_loopstatus loopret; + struct telnet_state ts; + int c; + int usessl = 0; + struct timeval now; + struct sockaddr_in taddr; + + ts.stdin_nsi = NULL; + + while ((c = getopt(argc, argv, "s")) != -1) { + switch (c) { + case 's': + usessl = 1; + break; + default: + usage(); + break; + } + } + + if (argc - optind <= 0 || argc - optind > 2) + usage(); + + + if (!resolve(argv[optind], &target)) { + fprintf(stderr, "Failed to resolve target host: %s\nQUITTING.\n", argv[optind]); + exit(1); + } + optind++; + + if (optind < argc) + portno = atoi(argv[optind]); + else + portno = 23; + + /* OK, we start with creating a p00l */ + if ((nsp = nsock_pool_new(NULL)) == NULL) { + fprintf(stderr, "Failed to create new pool. QUITTING.\n"); + exit(1); + } + + gettimeofday(&now, NULL); + + if ((ts.tcp_nsi = nsock_iod_new(nsp, NULL)) == NULL) { + fprintf(stderr, "Failed to create new nsock_iod. QUITTING.\n"); + exit(1); + } + + taddr.sin_family = AF_INET; + taddr.sin_addr = target; + taddr.sin_port = portno; + + if (usessl) { + ts.ssl_session = NULL; + ev = nsock_connect_ssl(nsp, ts.tcp_nsi, telnet_event_handler, 10000, &ts, (struct sockaddr *)&taddr, sizeof taddr, IPPROTO_TCP, portno, ts.ssl_session); + } else + ev = nsock_connect_tcp(nsp, ts.tcp_nsi, telnet_event_handler, 10000, &ts, (struct sockaddr *)&taddr, sizeof taddr, portno); + + printf("The event id is %lu -- initiating l00p\n", ev); + + /* Now lets get this party started right! */ + loopret = nsock_loop(nsp, -1); + + printf("nsock_loop returned %d\n", (int)loopret); + + return 0; +} diff --git a/nsock/examples/nsock_test_timers.c b/nsock/examples/nsock_test_timers.c new file mode 100644 index 0000000..24972e2 --- /dev/null +++ b/nsock/examples/nsock_test_timers.c @@ -0,0 +1,162 @@ +/*************************************************************************** + * nsock_test_timers.c -- A test program to exercise the nsock timer * + * routines. * + * * + ***********************IMPORTANT NSOCK LICENSE TERMS*********************** + * + * The nsock parallel socket event library is (C) 1999-2023 Nmap Software LLC + * This library is free software; you may redistribute and/or modify it under + * the terms of the GNU General Public License as published by the Free Software + * Foundation; Version 2. This guarantees your right to use, modify, and + * redistribute this software under certain conditions. If this license is + * unacceptable to you, Nmap Software LLC may be willing to sell alternative + * licenses (contact sales@nmap.com ). + * + * As a special exception to the GPL terms, Nmap Software LLC grants permission + * to link the code of this program with any version of the OpenSSL library + * which is distributed under a license identical to that listed in the included + * docs/licenses/OpenSSL.txt file, and distribute linked combinations including + * the two. You must obey the GNU GPL in all respects for all of the code used + * other than OpenSSL. If you modify this file, you may extend this exception to + * your version of the file, but you are not obligated to do so. + * + * If you received these files with a written license agreement stating terms + * other than the (GPL) terms above, then that alternative license agreement + * takes precedence over this comment. + * + * Source is provided to this software because we believe users have a right to + * know exactly what a program is going to do before they run it. This also + * allows you to audit the software for security holes. + * + * Source code also allows you to port Nmap to new platforms, fix bugs, and add + * new features. You are highly encouraged to send your changes to the + * dev@nmap.org mailing list for possible incorporation into the main + * distribution. By sending these changes to Fyodor or one of the Insecure.Org + * development mailing lists, or checking them into the Nmap source code + * repository, it is understood (unless you specify otherwise) that you are + * offering the Nmap Project (Nmap Software LLC) the unlimited, non-exclusive + * right to reuse, modify, and relicense the code. Nmap will always be available + * Open Source, but this is important because the inability to relicense code + * has caused devastating problems for other Free Software projects (such as KDE + * and NASM). We also occasionally relicense the code to third parties as + * discussed above. If you wish to specify special license conditions of your + * contributions, just say so when you send them. + * + * This program is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU General Public License v2.0 for more + * details (http://www.gnu.org/licenses/gpl-2.0.html). + * + ***************************************************************************/ + + +#include "nsock.h" + +#include <stdio.h> +#include <stdlib.h> +#include <sys/socket.h> +#include <netinet/in.h> +#include <arpa/inet.h> +#include <netdb.h> +#include <unistd.h> +#include <stdlib.h> +#include <time.h> +#include <assert.h> + +nsock_event_id ev_ids[2048]; + +int num_ids = 0; + +nsock_event_id request_timer(nsock_pool nsp, nsock_ev_handler handler, int timeout_msecs, void *userdata) { + nsock_event_id id; + + id = nsock_timer_create(nsp, handler, timeout_msecs, userdata); + printf("%ld: Created timer ID %li for %d ms from now\n", time(NULL), id, timeout_msecs); + + return id; + +} + +int try_cancel_timer(nsock_pool nsp, int idx, int notify) { + int res; + + printf("%ld:Attempting to cancel id %li (idx %d) %s notify.\n", time(NULL), ev_ids[idx], idx, ((notify) ? "WITH" : "WITHOUT")); + res = nsock_event_cancel(nsp, ev_ids[idx], notify); + printf("Kill of %li %s\n", ev_ids[idx], (res == 0) ? "FAILED" : "SUCCEEDED"); + return res; +} + +void timer_handler(nsock_pool nsp, nsock_event nse, void *mydata) { + enum nse_status status = nse_status(nse); + enum nse_type type = nse_type(nse); + int rnd, rnd2; + + printf("%ld:timer_handler: Received callback of type %s; status %s; id %li\n", time(NULL), nse_type2str(type), nse_status2str(status), nse_id(nse)); + + rnd = rand() % num_ids; + rnd2 = rand() % 3; + + if (num_ids > (sizeof(ev_ids) / sizeof(nsock_event_id)) - 3) { + printf("\n\nSUCCEEDED DUE TO CREATING ENOUGH EVENTS THAT IT WAS GOING TO OVERFLOW MY BUFFER :)\n\n"); + exit(0); + } + + if (status == NSE_STATUS_SUCCESS) { + switch (rnd2) { + case 0: + /* do nothing */ + /* Actually I think I'll create two timers :) */ + ev_ids[num_ids++] = request_timer(nsp, timer_handler, rand() % 3000, NULL); + ev_ids[num_ids++] = request_timer(nsp, timer_handler, rand() % 3000, NULL); + break; + case 1: + /* Kill another id (which may or may not be active */ + try_cancel_timer(nsp, rnd, rand() % 2); + break; + case 2: + /* Create a new timer */ + ev_ids[num_ids++] = request_timer(nsp, timer_handler, rand() % 3000, NULL); + break; + default: + assert(0); + } + } +} + +int main(int argc, char *argv[]) { + nsock_pool nsp; + enum nsock_loopstatus loopret; + int num_loops = 0; + + srand(time(NULL)); + /* OK, we start with creating a p00l */ + if ((nsp = nsock_pool_new(NULL)) == NULL) { + fprintf(stderr, "Failed to create new pool. QUITTING.\n"); + exit(1); + } + + ev_ids[num_ids++] = request_timer(nsp, timer_handler, 1800, NULL); + ev_ids[num_ids++] = request_timer(nsp, timer_handler, 800, NULL); + ev_ids[num_ids++] = request_timer(nsp, timer_handler, 1300, NULL); + ev_ids[num_ids++] = request_timer(nsp, timer_handler, 0, NULL); + ev_ids[num_ids++] = request_timer(nsp, timer_handler, 100, NULL); + + /* Now lets get this party started right! */ + while (num_loops++ < 5) { + loopret = nsock_loop(nsp, 1500); + if (loopret == NSOCK_LOOP_TIMEOUT) + printf("Finished l00p #%d due to l00p timeout :) I may do another\n", num_loops); + else if (loopret == NSOCK_LOOP_NOEVENTS) { + printf("SUCCESS -- NO EVENTS LEFT\n"); + exit(0); + } else { + printf("nsock_loop FAILED!\n"); + exit(1); + } + } + printf("Trying to kill my msp!\n"); + nsock_pool_delete(nsp); + printf("SUCCESS -- completed %d l00ps.\n", num_loops); + + return 0; +} |