summaryrefslogtreecommitdiffstats
path: root/nsock/examples
diff options
context:
space:
mode:
Diffstat (limited to 'nsock/examples')
-rw-r--r--nsock/examples/Makefile29
-rw-r--r--nsock/examples/README9
-rw-r--r--nsock/examples/nsock_telnet.c259
-rw-r--r--nsock/examples/nsock_test_timers.c162
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;
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-17 02:23:53 +0000