winsockcomm.cpp

wince下EVC编译通过的客户端程序

				}
			    if(gAddrInfo)
					freeaddrinfo(gAddrInfo);
				WSACleanup();

				EndDialog(hDlg, LOWORD(wParam));
				return TRUE;
			}
			else if (LOWORD(wParam) == IDC_BUTTON_OPEN)
			{
				// Get Server IP
				cntText = GetDlgItemText(hDlg, IDC_EDIT_IP, tchBuffer, sizeof(tchBuffer));
				gIPaddress = ConvertUnicodeToAscii(tchBuffer, cntText);
				cntText = GetDlgItemText(hDlg, IDC_EDIT_PORT, tchBuffer, sizeof(tchBuffer));
				gPortNumber = ConvertUnicodeToAscii(tchBuffer, cntText);
				switch (SendDlgItemMessage (hDlg, IDC_COMBO_ROLE, CB_GETCURSEL, 0, 0))
				{
					case 0 :
						// Create Server Thread to listen
				        Print(TEXT("======== Server Starts ========\r\n"));
						EnableWindow(GetDlgItem(hDlg, ID_SENDTEXT), FALSE);
						EnableWindow(GetDlgItem(hDlg, ID_SENDBTN), FALSE);

				        if (!GetExitCodeThread (hReadThread, &dwTStat) || (dwTStat != STILL_ACTIVE))
						{
							hReadThread = CreateThread (NULL, 0, ServerThread, hDlg, 0, &dwTStat);
							if (hReadThread)
								CloseHandle (hReadThread);
						}
						break;
					case 1 : // client
						// show something let user now it connected
				        Print(TEXT("======== Client Starts ========\r\n"));
						break;
					default :
						break;
				}
				return TRUE;
			}
			else if (LOWORD(wParam) == ID_SENDBTN)
			{
				ClientStart(hDlg);
				_tmainClient(hDlg);
				SetDlgItemText(hDlg, ID_SENDTEXT, TEXT(""));
				SetFocus (GetDlgItem (hDlg, ID_SENDTEXT));
				return TRUE;
			}
			else
			{
			}
			break;
	}
    return FALSE;
}

BOOL GetLANinfo(HWND hDlg)
{
	ULONG uLong;
	DWORD dw;
	uLong=sizeof(gStrucAdpInfo);
	TCHAR strBuffer[100];

	if ((dw=GetAdaptersInfo(gStrucAdpInfo, &uLong))==ERROR_SUCCESS)
	{
		wsprintf(strBuffer, TEXT("%hs"), gStrucAdpInfo[0].IpAddressList.IpAddress.String);
		// Set Default Port Number and IP and show in edit box
		SetDlgItemText(hDlg, IDC_EDIT_IP, strBuffer);
		SetDlgItemText(hDlg, IDC_EDIT_PORT, TEXT("8000"));
	}
	else
	{
		return 0;
	}
	return 1;
}

//======================================================================================

void Print(TCHAR *pFormat, ...)
{
    va_list ArgList;
    TCHAR Buffer[256];

    va_start (ArgList, pFormat);

    (void)wvsprintf (Buffer, pFormat, ArgList);
    SendDlgItemMessage (gDemoDlg, ID_RCVTEXT, EM_REPLACESEL, 0, (LPARAM)Buffer);

    va_end(ArgList);
}

DWORD WINAPI ServerThread (PVOID pArg)
{
    SOCKADDR_STORAGE ssRemoteAddr;
    int cbRemoteAddrSize, cbXfer, cbTotalRecvd;
    ADDRINFO Hints, *AddrInfo = NULL, *AI;
    char szRemoteAddrString[128];
    HWND hDlg;

	char chBuffer[100];


    hDlg = (HWND)pArg;
    gClientSock = INVALID_SOCKET;
	
    // Get get first LAN card and its configurations
    memset(&Hints, 0, sizeof(Hints));
    Hints.ai_family = AF_UNSPEC;
    Hints.ai_socktype = SOCK_STREAM;
    Hints.ai_flags = AI_NUMERICHOST | AI_PASSIVE;
    if(getaddrinfo(gIPaddress, gPortNumber, &Hints, &AddrInfo))
    {
        Print(TEXT("ERROR: getaddrinfo failed with error %d\r\n"), WSAGetLastError());
		return 0;
    }
    // Create a server socket
	AI = AddrInfo;
	gServerSock = socket(AI->ai_family, AI->ai_socktype, AI->ai_protocol);
    if (gServerSock != INVALID_SOCKET)
    {
        if (bind(gServerSock, AI->ai_addr, AI->ai_addrlen) == SOCKET_ERROR)
		{
            closesocket(gServerSock);
			return 0;
		}
        else 
        {
            if (listen(gServerSock, SOMAXCONN) == SOCKET_ERROR)
            {
		        Print(TEXT("Server go to listen mode failed\r\n"));
                closesocket(gServerSock);
				return 0;
            }
            Print(TEXT("Socket 0x%08x ready for connection with %hs family, %hs type, on port %hs\r\n"), 
                       gServerSock, (AI->ai_family == AF_INET) ? "AF_INET" : ((AI->ai_family == AF_INET6) ? "AF_INET6" : "UNKNOWN"),
                        (AI->ai_socktype == SOCK_STREAM) ? "TCP" : ((AI->ai_socktype == SOCK_DGRAM) ? "UDP" : "UNKNOWN"),
                        gPortNumber);
        }
    }
    freeaddrinfo(AddrInfo);

    cbRemoteAddrSize = sizeof(ssRemoteAddr);
    while (1)
	{
	    // Wait for incomming data/connections
        gClientSock = accept(gServerSock, (SOCKADDR*)&ssRemoteAddr, &cbRemoteAddrSize);
        if(gClientSock == INVALID_SOCKET) 
        {
            Print(TEXT("ERROR: accept() failed with error = %d\r\n"), WSAGetLastError());
			return 0;
        }
        Print(TEXT("Accepted TCP connection from socket 0x%08x\r\n"), gClientSock);
	    // Receive data from a client
		memset(chBuffer, 0, sizeof(chBuffer));
	    cbTotalRecvd = 0;
	    do
		{
			cbRemoteAddrSize = sizeof(ssRemoteAddr);
	        cbXfer = recvfrom(gClientSock, chBuffer + cbTotalRecvd, sizeof(chBuffer) - cbTotalRecvd, 0, (SOCKADDR *)&ssRemoteAddr, &cbRemoteAddrSize);
	        cbTotalRecvd += cbXfer;
		} while(cbXfer > 0 && cbTotalRecvd < sizeof(chBuffer));
	    if(cbXfer == SOCKET_ERROR)
		{
	        Print(TEXT("ERROR: Couldn't receive the data! Error = %d\r\n"), WSAGetLastError());
			return 0;
	    }
		else if(cbXfer == 0)
		{
	        Print(TEXT("ERROR: Didn't get all the expected data from the client!\r\n"));
			return 0;
		}
		else
		{
	        cbRemoteAddrSize = sizeof(ssRemoteAddr);
	        getpeername(gClientSock, (SOCKADDR *)&ssRemoteAddr, &cbRemoteAddrSize);
		}
		// get client address
	    if (getnameinfo((SOCKADDR *)&ssRemoteAddr, cbRemoteAddrSize,
				szRemoteAddrString, sizeof(szRemoteAddrString), NULL, 0, NI_NUMERICHOST) != 0)
			strcpy(szRemoteAddrString, "");
	    Print(TEXT("Received from client(%hs) <---- %hs\r\n"), szRemoteAddrString, chBuffer);
	    // Echo the data back to the client
		cbXfer = 0;
		cbXfer = sendto(gClientSock, chBuffer, cbTotalRecvd, 0, (SOCKADDR *)&ssRemoteAddr, cbRemoteAddrSize);
	    if(cbXfer != cbTotalRecvd)
		{
		    Print(TEXT("ERROR: Couldn't send the data! error = %d\r\n"), WSAGetLastError());
			return 0;
		}
		else
			Print(TEXT("Send Echo to client ---->%hs\r\n"), chBuffer);
	}
    return 0;
} 

//============================================================================================


int _tmainClient(PVOID pArg)
{
    int cbXfer, cbTotalRecvd;
	char chBuffer[100], *pBuffer;
	UINT cntText;
	TCHAR tchBuffer[100];
    HWND hDlg;

    hDlg = (HWND)pArg;

    // Send data to the server
	cntText = GetDlgItemText(hDlg, ID_SENDTEXT, tchBuffer, sizeof(tchBuffer));
	pBuffer = ConvertUnicodeToAscii(tchBuffer, cntText);
	memset(chBuffer, 0, sizeof(chBuffer));
	sprintf(chBuffer, "%s", pBuffer);
    cbXfer = 0;
    cbXfer = sendto(gsock, chBuffer, sizeof(chBuffer), 0, gAI->ai_addr, gAI->ai_addrlen);
    if(cbXfer != sizeof(chBuffer))
    {
        Print(TEXT("ERROR: Couldn't send the data! error = %d\r\n"), WSAGetLastError());
		return 0;
    }
    Print(TEXT("Sent to server(%hs) ----> %hs\r\n"), gIPaddress, chBuffer);

    // Receive the echo'd data back from the server
    cbTotalRecvd = 0;
	memset(chBuffer, 0, sizeof(chBuffer));
    do
    {
        cbXfer = recvfrom(gsock, chBuffer + cbTotalRecvd, sizeof(chBuffer) - cbTotalRecvd, 0, NULL, NULL);
        cbTotalRecvd += cbXfer;
    } while(cbXfer > 0 && cbTotalRecvd < sizeof(chBuffer));
    if(cbXfer == SOCKET_ERROR)
    {
        Print(TEXT("ERROR: Couldn't receive the data! Error = %d\r\n"), WSAGetLastError());
		return 0;
    }
    else if(cbTotalRecvd != sizeof(chBuffer))
    {
        Print(TEXT("ERROR: Server didn't send back all the expected data!\r\n"));
		return 0;
    }
    Print(TEXT("Received Echo from server(%hs) <---- %hs\r\n"), gIPaddress, chBuffer);

    return 0;
} 

//=========================================================================
char * ConvertUnicodeToAscii(LPCWSTR p,int l)
{
 char *Out;
 int nChars;

 nChars=WideCharToMultiByte(CP_ACP,
         0,p,l,NULL,0,NULL,NULL);
 if(nChars!=0)
 {
  Out=(char *)malloc(nChars+1);
  if(Out!=NULL)
  {
   memset(Out,0,nChars+1);
   nChars=WideCharToMultiByte(CP_ACP,
           0,p,l,Out,nChars,NULL,NULL);
   if(nChars==0)
   {
    free(Out);
    Out=NULL;
   }
  }
 }
 else
  Out=NULL;

 if(Out==NULL)
 {
  Out=(char *)malloc(sizeof(char));
  *Out=0;
 }

 return Out;
}

DWORD WINAPI ClientStart (PVOID pArg)
{
    ADDRINFO Hints;
	HWND hDlg;

    hDlg = (HWND)pArg;

    memset(&Hints, 0, sizeof(Hints));
    Hints.ai_family = AF_UNSPEC;
    Hints.ai_socktype = SOCK_STREAM;

    if(getaddrinfo(gIPaddress, gPortNumber, &Hints, &gAddrInfo))
    {
        Print(TEXT("ERROR: Couldn't get resolve the server name/address!"));
		return 0;
    }
    // Attempt to connect to each address until we find one that succeeds
	gAI = gAddrInfo;
    gsock = socket(gAI->ai_family, gAI->ai_socktype, gAI->ai_protocol);
    if (gsock != INVALID_SOCKET)
    {
        if (gAI->ai_socktype == SOCK_STREAM)
        {
            if(connect(gsock, gAI->ai_addr, gAI->ai_addrlen) == SOCKET_ERROR)
            {
			    Print(TEXT("Connect to server failed\r\n"));
                closesocket(gsock);
				return 0;
            }
        }
        // connect() succeeded
    }
	else 
	{
	   Print(TEXT("Client create socket failed\r\n"));
	   return 0;
	}

	return 1;
}