/*
 * Copyright (C) 1994-1996, Michael S. Durkin & Tsoft.  All Rights Reserved.
 *    Please see legalese in README.1ST or contact us for details:
 *        Contact:  Mike Durkin        mdurkin@tsoft.net (email preferred)
 *                  c/o Tsoft
 *                  P.O. Box 14897
 *                  Berkeley, CA 94712    USA
 */

/*

#      sends numbered UDP datagrams to a remote host.
#
#      Usage:
#          udp-send [-n <datagrams_number>] 
#                   [-s <datagrams_size>] 
#                   [-p <port_number>]
#                   host
#
#          <datagrams_number> defaults to 16, <datagram_size> to 32
#          and <port_number> to 9870.
#
# Renaud Waldura # on Sep 18 16:51:24 1995
# converted to trumpet abi by Dan Kegel and Chris Thomas 6/7/1996
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <conio.h>

#include <tcpabi/tcpabi.h>

#define TRUE (1)
#define FALSE (0)

#define TIMEOUT		20

#define BUFLEN 512
static char inbuf[BUFLEN];

int datagrams_n = 16;
int datagrams_size = 32;
int port = 9870;

static void print_usage( char *progname )
{
	printf(
"     Usage:\n"
"	  %s [-n <datagrams_number>]\n"
"			   [-s <datagrams_size>]\n"
"			   [-p <port_number>]\n"
"			   host\n", progname);
}

static int send_packets(char *hostname)
{
	unsigned int send_handle;
	unsigned long dst_ip;
	unsigned dst_port;
	unsigned src_port;

	int n, i;
	int err;
	int namelen, inlen;

	// open a handle to send packets to port 9870
	dst_ip = htonl(aton(hostname));
	dst_port = 9870;
	src_port = 0;

	send_handle = tcpabi_udp_open(dst_ip, dst_port, src_port, UDP_OPEN_NORMAL, NULL);
	if (send_handle == NULL_HANDLE) {
		printf("error opening send_handle!\n");
		return FALSE;
	}
	printf("send_handle opened!\n");

	for (n = 1; n<datagrams_n; n++) {
		int i;
		for (i=0; i<datagrams_size-1; i++)
			inbuf[i] = '*';
		sprintf(inbuf+i, ":%d", n);
		namelen = strlen(inbuf);
		printf("sending %s, len %d\n", inbuf, namelen);
		err = tcpabi_udp_send(send_handle, inbuf, namelen, TTLTOS_NORMAL, 0xF001,
							UDP_SEND_NORMAL);
		if (err != (namelen)) {
			printf("failed!\n");
			printf("chat: UDP put to '%s' failed, _tcpabi_errno %d\n\n",
				hostname, _tcpabi_errno);
			tcpabi_udp_close(send_handle, UDP_CLOSE_NORMAL );
			return TRUE;     // error
		}
	}
	
	tcpabi_udp_close(send_handle, UDP_CLOSE_NORMAL);
	return TRUE;

}

static int send_to( char *host )
{
	unsigned long hostip;
	unsigned sd;            // descriptor
	int namelen, inlen;
	int n;

	if((hostip = resolve( host )) == 0L) {
		printf("\nchat: Can't resolve '%s' to IP address\n\n", host);
		return TRUE;     // error
	}
	
	if( (sd = tcpabi_udp_open( hostip, port, 0,
								UDP_OPEN_NORMAL, NULL )) == NULL_HANDLE ) {
		printf("\nchat: UDP open to '%s' port %d failed\n\n", host, port);
		return TRUE;     // error
	}
	printf("\n[%s]\n", host);
	
	for (n = 1; n<datagrams_n; n++) {
		int i;
		for (i=0; i<datagrams_size-1; i++)
			inbuf[i] = '*';
		sprintf(inbuf+i, ":%d", n);
		namelen = strlen(inbuf);
		printf("sending %s, len %d\n", inbuf, namelen);
		if (tcpabi_udp_send( sd, inbuf, namelen, TTLTOS_NORMAL, 0xF001,
							UDP_SEND_NORMAL) != (namelen) ) {
			printf("chat: UDP put to '%s' failed, _tcpabi_errno %d\n\n",
				host, _tcpabi_errno);
			tcpabi_udp_close( sd, UDP_CLOSE_NORMAL );
			return TRUE;     // error
		}
	}

   tcpabi_udp_close( sd, UDP_CLOSE_NORMAL );
   return FALSE;     // no error
}

#ifdef __386__
#define DPMI_INT 0x31
#include <dos.h>
/*-------------------------------------------------------------------------
 Get a block of DOS memory.
 Return flat pointer to it, or NULL on failure.
 Also place segmented real mode pointer in *seg, *off, and protected mode
 selector in *sel.  (The selector is only needed to free the memory,
 thanks to the flat memory model of DOS4G.)
-------------------------------------------------------------------------*/
void *my_dos_malloc(size_t len, short *seg, short *off, short *sel)
{
	union REGS regs;

	// Note: The DPMI service expects the number of paragraphs not bytes.
	regs.w.ax = (short) 0x0100;
	regs.w.bx = (len + 0xf) >> 4;
	int386(DPMI_INT, &regs, &regs);
	if (regs.w.cflag) {
		// Abort!   Out of DOS memory!
		return NULL;
	}

	// Result is in AX:DX (seg:sel)
	*seg = regs.w.ax;
	*off = 0;
	*sel = regs.w.dx;
	return (void *) (regs.w.ax << 4);
}
#endif

int main( int argc, char *argv[] )
{
   int i;
   int exitcode = 0;
   short selector;

   puts("udpsend test (for UDP ABI) -- Copyright (C) 1994, Michael S. Durkin");

   if( argc == 1 ) {
      print_usage( argv[0] );
      exit(1);
   }
   for (i = 1; i<argc; i++) {
	   if (argv[i][0] == '-') {
		   switch (argv[i][1]) {
		   case 'n':	datagrams_n = atoi(argv[i]+2); break;
		   case 's':	datagrams_size = atoi(argv[i]+2); break;
		   case 'p':	port = atoi(argv[i]+2); break;
		   default:		print_usage(argv[0]); exit(1);
		   }
	   }

   }

   if( !tcpabi_installed() ) {
      printf("\n   UDP ABI not found\n");
      exit(2);
   }
#ifdef __386__
	scratch_flat = my_dos_malloc(BUFLEN, &scratch_seg, &scratch_off, &selector);
	if (!scratch_flat) {
		printf("couldn't allocate DOS memory!\n");
		exit(3);
	}
#endif

   //exitcode &= send_to( argv[1] );
   send_packets(argv[1]);
   exit(exitcode);
   return 0;
}