#include <windows.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <nb30.h>
#pragma comment(lib,"netapi32.lib")
#define MAXSESSIONS 2
#define MAXNAMES 4

struct message{
	int name;
	char msg[10];
}message1;

UCHAR LanaEnum(LANA_ENUM *lenum)
{
	NCB ncb;
	UCHAR ucRc;
	memset(&ncb,0,sizeof(NCB));
	ncb.ncb_command=NCBENUM;
	ncb.ncb_buffer=(PUCHAR)lenum;
	ncb.ncb_length=sizeof(LANA_ENUM);
	ucRc=Netbios(&ncb);
	return(ucRc);
}

UCHAR RestNCB(UCHAR lana)
{
	NCB ncb;
	UCHAR ucRc;
	memset(&ncb,0,sizeof(NCB));
	ncb.ncb_command=NCBRESET;
	ncb.ncb_callname[0]=MAXSESSIONS;
	ncb.ncb_callname[2]=MAXNAMES;
//	ncb.ncb_callname[3]=TRUE;
	ncb.ncb_lana_num=lana;
	ucRc=Netbios(&ncb);
	return(ucRc);
}

UCHAR AddName(UCHAR *Name,UCHAR Lana,UCHAR *NameNum)
{
	NCB ncb;
	memset(&ncb,0,sizeof(NCB));
	ncb.ncb_command=NCBADDNAME;
	ncb.ncb_lana_num=Lana;
	strcpy((char*)ncb.ncb_name,(const char*)Name);

	
	Netbios(&ncb);
	*NameNum=ncb.ncb_num;
	return(ncb.ncb_cmd_cplt);
}

UCHAR DeleteName(UCHAR *Name,UCHAR Lana)
{
	NCB ncb;
	memset(&ncb,0,sizeof(NCB));
	ncb.ncb_command=NCBDELNAME;
	ncb.ncb_lana_num=Lana;
	strcpy((char*)ncb.ncb_name,(const char*)Name);
	Netbios(&ncb);
	return(ncb.ncb_cmd_cplt);
}

UCHAR DatagramSend(PNCB pncb,UCHAR DestName[],UCHAR Num,UCHAR *pBuffer,WORD Length,UCHAR Lana)
{
	memset(pncb,0,sizeof(NCB));
	pncb->ncb_command=NCBDGSEND;
	strcpy((char *)pncb->ncb_callname,(const char *)DestName);
	pncb->ncb_lana_num=Lana;
	pncb->ncb_num=Num;
	pncb->ncb_buffer=pBuffer;
	pncb->ncb_length=Length;
	Netbios(pncb);
	return(pncb->ncb_cmd_cplt);


}

UCHAR RecvechoMessage(PNCB pncb,UCHAR Num,UCHAR *pBuffer,WORD Length,UCHAR Lana)
{
	memset(pncb,0,sizeof(NCB));
	pncb->ncb_command=NCBDGRECVBC;
	pncb->ncb_lana_num=Lana;
	pncb->ncb_num=Num;
	pncb->ncb_buffer=pBuffer;
	pncb->ncb_length=Length;
	Netbios(pncb);
	return(pncb->ncb_cmd_cplt);
}



UCHAR SendMessage(PNCB pncb,UCHAR Num,UCHAR *pBuffer,WORD Length,UCHAR Lana)
{
	memset(pncb,0,sizeof(NCB));
	pncb->ncb_command=NCBDGSENDBC;
	pncb->ncb_lana_num=Lana;
	pncb->ncb_num=Num;
	pncb->ncb_buffer=pBuffer;
	pncb->ncb_length=Length;
	Netbios(pncb);
	return(pncb->ncb_cmd_cplt);
}


UCHAR WaitTimeRequest(PNCB pncb,UCHAR Num,UCHAR *pBuffer,UCHAR Lana)
{
	memset(pncb,0,sizeof(NCB));
	pncb->ncb_command=NCBDGRECV;
	pncb->ncb_lana_num=Lana;
	pncb->ncb_num=Num;
	pncb->ncb_buffer=pBuffer;
	pncb->ncb_length=512;
	Netbios(pncb);
	return(pncb->ncb_cmd_cplt);
}


void main()
{
	NCB ClientNcb;
	UCHAR LocalName[16],RemoteName[16],p[30];
	UCHAR LocalNameNum;
	LANA_ENUM lenum;

	strcpy((char *)LocalName,"swandive");
	strcpy((char *)RemoteName,"swandive1");

	if(LanaEnum(&lenum)!=NRC_GOODRET)
	{
		printf("lanaEnum Error!\n");
		return;
	}
//	printf("LanaEnum OK!\n");

	if(RestNCB(lenum.lana[0])!=NRC_GOODRET)
	{
		printf("RestNCB Error!\n");
		return ;
	}
	printf("please just a wait!\n");

	if(AddName(LocalName,lenum.lana[0],&LocalNameNum)!=NRC_GOODRET)
	{
		printf("AddName Error!\n");
		return;
	}
//	printf("AddName OK!\n");


	for(int z=0;z<sizeof(message1.msg);z++)	
		message1.msg[z]='0';

	printf("Please input one number!\n");
	scanf("%d",&message1.name);

	printf("Please input some words!\n");
	scanf("%s",message1.msg);
	
	int count=0;
	for(int k=0;k<sizeof(message1.msg);k++)
		if(message1.msg[k]!='0')
			count++;

	
	for(int i=0;i<count;i++)		
	{
		p[i]=message1.msg[i];
	}
	p[count-1]=(UCHAR)message1.name;
	p[count]='\0';
	printf("%d\n",p[count-1]);
	printf("%c\n",p[count]);
	if(DatagramSend(&ClientNcb,RemoteName,LocalNameNum,(UCHAR *)p,sizeof(p),lenum.lana[0])!=NRC_GOODRET)
	{
		printf("Send Error!\n");
	}
	printf("Send OK!\n");	

	for(;;)
	{

		if(RecvechoMessage(&ClientNcb,LocalNameNum,(UCHAR *)p,sizeof(p),lenum.lana[0])!=NRC_GOODRET)
		{
			printf("RecvEchoMessage Error!\n");
			return;
		}
		printf("RecvEchoMessage ok!\n");
		if(ClientNcb.ncb_cmd_cplt==0)
		{
			for(int j=0;j<count+5;j++)
				message1.msg[j]=p[j];
			message1.msg[count+5]='\0';
			UCHAR pp[9];
			for(int t=0;t<8;t++)
				pp[t]=p[count+5+t];
			pp[8]='\0';

			message1.name=(int)p[count+6+t];

			printf("message is:%s\n",message1.msg);
			printf("%s",pp);
			printf("%d\n",message1.name);
		}
	
	}
	
}