cookie.c

dos下开发TCP网络的库文件部分

/******************************************************************************
    COOKIE - read and print a witty saying from internet

    By: Jim Martin                      Internet: [e-mail]
        Dormitory Networking Project    UUCP: {backbone}!rutgers!jim
        Rutgers University              Phone: [phone]

    Uses the WATTCP kernal.
******************************************************************************/

#include <stdio.h>
#include <stdlib.h>
#include <tcp.h>


#define COOKIE_PORT 17

int cookie( longword host )
{
	static udp_Socket sock;
    	udp_Socket *s;
    	char buffer[ 2049 ];
    	int status;
	int i;
	int fortunelen;

    	s = &sock;
    	status = 0;

	if ( host != 0 )
	{
		status=udp_open( s, 0, host, COOKIE_PORT, NULL );
	}
	else
	{
		if (_last_cookie == 0)
		{
			puts("Sorry, I can't seem to remember where my cookie jars are. ");
			puts("Could you tell me where one is? (Hint: host on the comand line)");
			exit(3);
		}
		for (i=0; i < _last_cookie; i++)
		{
			if ( (status=udp_open( s, 0, _cookie[i], COOKIE_PORT, NULL )) != 0 )
			{
				break;
			}
		}
	}

	if ( status == 0 )
	{
		puts("None of the cookie jars are open!");
		return( 1 );
	}

	sock_write( s, "\n", 1 );


    	while ( 1 )
	{
		sock_tick( s, &status );

		if (sock_dataready( s ) )
		{
			fortunelen=sock_fastread( s, buffer, sizeof( buffer ));
			buffer[fortunelen]='\0';
			printf("%s\n", buffer);
			sock_close(s);
			return(0);
		}
    	}

sock_err:
    switch (status)
	{
		case 1 : /* foreign host closed */
			 return(0);
		case -1: /* timeout */
			 printf("\nConnection timed out!");
             printf("ERROR: %s\n\r", sockerr( s ));
			 return(1);
    	}
        return (0);
}


int main(int argc, char **argv )
{
	int status;
	longword host;

	if (argc > 2)
	{
		puts("Quote of the Day (Cookie) - retrieves a witty message");
		puts("Usage: COOKIE [server]");
		exit( 3 );
	}

	sock_init();

	if ( argc == 1)
	{
		status = cookie ((longword) NULL);
	}
	else
	{
		if ( (host = resolve( argv[1])) != 0uL )
		{
			status = cookie( host );
		}
		else
		{
			printf("Could not resolve host '%s'\n", argv[1]);
			status = 3;
		}
	}

	exit( status );
        return (0);  /* not reached */
}