datagram.c

杭州立宇泰豪华型44B0开发板

/* UDP datagram utility */

#define VERSION "0.06"

#include <stdio.h>
#include <ctype.h>
#include <stdlib.h>
#include <string.h>
//#include <signal.h>
#include <time.h>

#include "..\Target\44blib.h"

#include "ether.h"
#include "netutil.h"
#include "net.h"
#include "ip.h"
#include "udp.h"

//#define CFGFILE     "tcplean.cfg"   /* Default config filename */
//#define CFGEXT      ".cfg"          /* Default config extension */
#define REMOTEIP    "192.168.253.3"
#define LOCALIP     "192.168.253.2"
#define IDCFG       "a"
#define NETCFG      "ether ne 0x08000000"
//#define IPCFG       "192.168.35.2"
#define MASKCFG     "255.255.255.0"
#define GATECFG     "192.168.253.3"

#define MAXNETCFG   40              /* Max length of a net config string */
#define DEFPORT     DAYPORT         /* Default port number */
#define ARPTIME     1000            /* ARP retry timer (msec) */
#define COMMANDLEN  80
#define TRYTIME     1000
#define MAXTRIES    10

#define ARP_TX      1               /* Client state: sending ARP */
#define ARP_RX      2               /*               received ARP */
#define CLIENT_DONE 3               /*               completed */

GENFRAME genframe;                  /* Frame for network Tx/Rx */
ARPKT arpkt;
IPKT  ipkt;
ICMPKT icmpkt;
UDPKT udpkt;

//char cfgfile[MAXPATH+5]=CFGFILE;    /* Config filename */
//char netcfg[MAXNETCFG+1]="??";      /* Network config string */
extern BYTE bcast[MACLEN];          /* Broadcast Ethernet addr */
extern BYTE zermac[MACLEN];         /* All-zero MAC addr */
NODE  locnode;                      /* My Ethernet and IP addresses */
NODE remnode;                       /* Remote node */
WORD remport;                       /* Remote port number */
int breakflag;                      /* Flag to indicate ctrl-break pressed */
int client;                         /* Flag to indicate client is active */
int binmode;                        /* Flag to enable binary mode */
//char infile[MAXPATH+1];             /* Filename for I/P data */
const char UDP_text[]= "This is a test text!";

/* Debug flags */
extern int netdebug;                /* Net packet display */
extern int udpdebug;                /* UDP frame display */
extern int ndrivers;

/* Function pointer: upcall from TCP/IP stack */
extern NODE *(*get_locnode_n)(int n);           /* Get local node */

/* Prototypes */
WORD read_netconfig(NODE *np);
NODE *locnode_n(int n);
int do_receive(GENFRAME *gfp);
int udp_receive(GENFRAME *gfp, int len);
void udp_transmit(GENFRAME *gfp, NODE *sp, NODE *dp, void *dat, int len);
void udp_transmit_text(GENFRAME *gfp, NODE *sp, NODE *dp, char *in);
void do_poll(GENFRAME *gfp);
void disp_usage(void);
void disp_data(BYTE *data, int len);
WORD str2service(char *str);
void break_handler(int sig);
void Datagram_test(void);

void Datagram_test(void)
{
    int args=0, err=0, tries=0, st, cstate=ARP_TX, lastcstate=0,temp;
    char c,*p,temps[18],*cmd="";
	char argv[256];
    WORD dtype;
    GENFRAME *gfp;
    char *in=0;
	//char *in=UDP_text;

    Uart_Printf("\nDATAGRAM v" VERSION "");          /* Sign on */
    get_locnode_n = locnode_n;                  /* Set upcall ptr to func */
    //signal(SIGINT, break_handler);              /* Trap ctrl-C */
    remnode.port = DEFPORT;                     /* Preset remote port number */
    locnode.port = MINEPORT;
	Uart_Printf("\nEnter Command line:(Example, datagram 192.168.253.3 echo hello)\n");
	p=argv;
	client=0;
	breakflag=0;
	ndrivers=0;
	Uart_GetString(p);
	p=argv;
	//Analyse Command Line-------------------------------------------
    while (*p++!='\0')                       /* Process command-line args */
    {
        if ((c=*p) == '-')
        {
            switch (*(p+1))
            {
            case 'b':
			case 'B':                           /* -B: binary mode */
                binmode = 1;
                break;
		    
            case 'u':
            case 'U':                           /* -U: display UDP segments */
                udpdebug = 1;
                break;
			case 'v':
            case 'V':                           /* -V: verbose packet display */
                netdebug |= 1;
                break;
			case 'x':
            case 'X':                           /* -X: hex packet display */
                netdebug |= 2;
                break;
            default:
                err = 1;
            }
        }
        else if ((client==0) && (c>=0x30)&&(c<=0x39))       /* If client mode.. */
        {
            remnode.ip = atoip(p);     /* Get destination IP address */
			while((*p++)!=' ');
			p--;
            client++;
        }
        else if (client==1)
        {
            if ((remport=str2service(p))!=0)
                remnode.port = remport;         /* ..then port number */
            else
            {
                Uart_Printf("Unrecognised port number/service '%s'\n", argv[args]);
                err++;
            }
			while(((*p++)!=' ')&&((*p++)!='\0'));
			p--;
            client++;
        }
        else if (client == 2)
        {
            cmd = p;//argv[args];                   /* ..then command string */
            while(((*p++)!=' ')&&((*p++)!='\0'));
			p--;
            client++;
        }
    }

	//init. ethernet controller--------------------------------------
    if (err)                                    /* Prompt user if error */
        disp_usage();
    else if (!(dtype=read_netconfig(&locnode)))//初始化本地节点，启动以太网控制器
        //Uart_Printf("Invalid configuration '%s'\n", cfgfile);
        Uart_Printf("Initialize node failed!");
    else
    {
        remnode.dtype = genframe.g.dtype = dtype;   /* Set frame driver type */
        gfp = &genframe;                            /* Get pointer to frame */
        Uart_Printf("IP %s", ipstr(locnode.ip, temps));
        Uart_Printf(" mask %s", ipstr(locnode.mask, temps));
        if (locnode.gate)
            Uart_Printf(" gate %s", ipstr(locnode.gate, temps));
        if (dtype & DTYPE_ETHER)
            Uart_Printf(" Ethernet %s", ethstr(locnode.mac, temps));
        else if (dtype & DTYPE_SLIP)
            cstate = ARP_RX;
        Uart_Printf("\nPress EXINTn to exit\n");
        if (client)
        {
            Uart_Printf("Contacting %s...\n\n", ipstr(remnode.ip, temps));
            memcpy((BYTE *)(remnode.mac), bcast, MACLEN);
        }
		else
			Uart_Printf("Enter Server mode...\n\n");
				
        //mstimeout(&mstime, 0);
        //Delay(0);
        while (!breakflag)                      /* Main loop.. */
        {
            if (client /*&& (mstimeout(&mstime, TRYTIME)*/ || (cstate!=lastcstate))
            {
				if (tries++ > MAXTRIES)         /* Giving up? */
						breakflag = 1;
				else if (cstate == ARP_TX)      /* (Re)transmit ARP? */
                {
					put_frame(gfp, make_arp(gfp, &locnode, &remnode, ARPREQ));

                }
                else if (cstate == ARP_RX)      /* ARP response? */
                {
                    if (in){
                        udp_transmit_text(gfp, &locnode, &remnode, in);
                    	}
                    else{
                          temp=strlen(cmd);                    
                          udp_transmit(gfp, &locnode, &remnode, cmd, temp);//strlen(cmd));

                    	}
                }
                else if (cstate == CLIENT_DONE) /* UDP response? */
                {
                    do_poll(gfp); 
					while (st = do_receive(gfp))
                    {
                        do_poll(gfp); 
                    }
					breakflag = 1;
                }
                lastcstate = cstate;            /* Record state-change */
            }
			do_poll(gfp);                       /* Poll net drivers */
			st = do_receive(gfp);               /* Receive frames */
            cstate = st ? st : cstate;          /* ..maybe change state */
            if (c=Readkey())             /* if Any exintX pressed,break... */
                breakflag = 1;
        }
        close_net(dtype);                       /* Shut down net driver */
    }