winsockcomm_new.cpp

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

                if(sock == INVALID_SOCKET) 
                {
                    Print(TEXT("ERROR: accept() failed with error = %d\r\n"), WSAGetLastError());
                    goto Cleanup;
                }

                Print(TEXT("Accepted TCP connection from socket 0x%08x\r\n"), sock);


    //
    // Receive data from a client
    //

    cbTotalRecvd = 0;
    do
    {
        cbRemoteAddrSize = sizeof(ssRemoteAddr);
        cbXfer = recvfrom(sock, pBuf + cbTotalRecvd, sizeof(pBuf) - cbTotalRecvd, 0,
            (SOCKADDR *)&ssRemoteAddr, &cbRemoteAddrSize);
        cbTotalRecvd += cbXfer;
    } while(cbXfer > 0 && cbTotalRecvd < sizeof(pBuf));

    if(cbXfer == SOCKET_ERROR)
    {
        Print(TEXT("ERROR: Couldn't receive the data! Error = %d\r\n"), WSAGetLastError());
        goto Cleanup;
    }
    else if(cbXfer == 0)
    {
        Print(TEXT("ERROR: Didn't get all the expected data from the client!\r\n"));
        goto Cleanup;
    }

    cbRemoteAddrSize = sizeof(ssRemoteAddr);
    getpeername(sock, (SOCKADDR *)&ssRemoteAddr, &cbRemoteAddrSize);
    
    if (getnameinfo((SOCKADDR *)&ssRemoteAddr, cbRemoteAddrSize,
        szRemoteAddrString, sizeof(szRemoteAddrString), NULL, 0, NI_NUMERICHOST) != 0)
        strcpy(szRemoteAddrString, "");

    Print(TEXT("SUCCESS - Received %d bytes from client %hs\r\n"), cbTotalRecvd, szRemoteAddrString);

    //
    // Echo the data back to the client
    //

    cbXfer = 0;
    cbXfer = sendto(sock, pBuf, cbTotalRecvd, 0, (SOCKADDR *)&ssRemoteAddr, cbRemoteAddrSize);

    if(cbXfer != cbTotalRecvd)
        Print(TEXT("ERROR: Couldn't send the data! error = %d\r\n"), WSAGetLastError());
    else
        Print(TEXT("SUCCESS - Echo'd %d bytes back to the client\r\n"), cbXfer);
}
Cleanup:

    closesocket(SockServ);

    if(sock != INVALID_SOCKET)
    {
        shutdown(sock, SD_BOTH);
        closesocket(sock);
    }

    return 0;
} 

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

DWORD WINAPI ClientStart(PVOID pArg)
{
    SOCKET gClientSocket = INVALID_SOCKET;
	char *szRemoteName;
    char *szPort = DEFAULT_PORT;
    ADDRINFO Hints; 
    TIMEVAL timeout = {TIMEOUT_SECS, TIMEOUT_USECS};
    HWND hDlg;

    hDlg = (HWND)pArg;
	szRemoteName = gIPaddress;

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

    if(getaddrinfo(szRemoteName, gPortNumber, &Hints, &gServerAddrInfo))
    {
        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
    //
            gClientSocket = socket(gServerAddrInfo->ai_family, gServerAddrInfo->ai_socktype, gServerAddrInfo->ai_protocol);
            if (gClientSocket != INVALID_SOCKET)
            {
                if (gServerAddrInfo->ai_socktype == SOCK_STREAM)
                {
                    if(connect(gClientSocket, gServerAddrInfo->ai_addr, gServerAddrInfo->ai_addrlen) == SOCKET_ERROR)
                    {
                        // Connect failed, let's close this socket and try again on the next address in the list
                        closesocket(gClientSocket);
//						MessageBox(NULL, TEXT("connect succeed"), TEXT(""), MB_OK);
                    }
                }
                // connect() succeeded or we're a UDP socket
            }
			else 
				MessageBox(NULL, TEXT("connect failed"), TEXT(""), MB_OK);
	return 1;
}
  



DWORD WINAPI ClientThread (PVOID pArg)
{
	char *szRemoteName;
    char *szPort = DEFAULT_PORT;
    SOCKADDR_STORAGE ssRemoteAddr;
    int cbXfer, cbTotalRecvd, cbRemoteAddrSize;
    char szRemoteAddrString[128];
    TIMEVAL timeout = {TIMEOUT_SECS, TIMEOUT_USECS};
    char pRecvBuf[BUFFER_SIZE];

    HWND hDlg;
    hDlg = (HWND)pArg;
	szRemoteName = gIPaddress;


    //
    // Send data to the server
    //

//MessageBox(NULL, TEXT("Start to sendto"), TEXT(""), MB_OK);

    cbXfer = 0;
    cbXfer = sendto(gClientSocket, pBuf, sizeof(pBuf), 0, gServerAddrInfo->ai_addr, gServerAddrInfo->ai_addrlen);

    if(cbXfer != sizeof(pBuf))
    {
        Print(TEXT("ERROR: Couldn't send the data! error = %d\r\n"), WSAGetLastError());
        return 0;
    }

    if (getnameinfo(gServerAddrInfo->ai_addr, gServerAddrInfo->ai_addrlen,
        szRemoteAddrString, sizeof(szRemoteAddrString), NULL, 0, NI_NUMERICHOST) != 0)
        strcpy(szRemoteAddrString, "");
    
    Print(TEXT("SUCCESS - Sent %d bytes to address %hs\r\n"), cbXfer, szRemoteAddrString);

    //
    // Receive the echo'd data back from the server
    //
    cbTotalRecvd = 0;
    do
    {
        cbRemoteAddrSize = sizeof(ssRemoteAddr);
        cbXfer = recvfrom(gClientSocket, pRecvBuf + cbTotalRecvd, sizeof(pRecvBuf) - cbTotalRecvd, 0, 
            (SOCKADDR *)&ssRemoteAddr, &cbRemoteAddrSize);
        cbTotalRecvd += cbXfer;
    } while(cbXfer > 0 && cbTotalRecvd < sizeof(pRecvBuf));

    if(cbXfer == SOCKET_ERROR)
    {
        Print(TEXT("ERROR: Couldn't receive the data! Error = %d\r\n"), WSAGetLastError());
        return 0;
    }
    else if(cbTotalRecvd != sizeof(pRecvBuf))
    {
        Print(TEXT("ERROR: Server didn't send back all the expected data!\r\n"));
        return 0;
    }

    memset(&ssRemoteAddr, 0, sizeof(ssRemoteAddr));
    cbRemoteAddrSize = sizeof(ssRemoteAddr);
    getpeername(gClientSocket, (SOCKADDR *)&ssRemoteAddr, &cbRemoteAddrSize);

    if (getnameinfo((SOCKADDR *)&ssRemoteAddr, cbRemoteAddrSize,
        szRemoteAddrString, sizeof(szRemoteAddrString), NULL, 0, NI_NUMERICHOST) != 0)
        strcpy(szRemoteAddrString, "");

    Print(TEXT("SUCCESS - Received %d bytes back from address %hs\r\n"), cbTotalRecvd, szRemoteAddrString);

    return 1;
} 

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

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;
}

//==============================================================================
//==============================================================================
//==============================================================================
int _tmainClient(void)
{
    SOCKET sock = INVALID_SOCKET;
    int nFamily = DEFAULT_FAMILY;
    int nSockType = DEFAULT_SOCKTYPE;
//    char szRemoteName[64];
    char szRemoteName[]="192.168.100.48";
    char *szPort = DEFAULT_PORT;
    SOCKADDR_STORAGE ssRemoteAddr;
    int cbXfer, cbTotalRecvd, cbRemoteAddrSize;
    WSADATA wsaData;
    ADDRINFO Hints, *AddrInfo = NULL, *AI;
    char szRemoteAddrString[128];
    fd_set fdReadSet;
    TIMEVAL timeout = {TIMEOUT_SECS, TIMEOUT_USECS};
    char pRecvBuf[BUFFER_SIZE];

/*
    if(argc < 2)
    {
        Print(TEXT("Server name/address parameter required.  Ex: clnt 123.45.67.89"));
        goto Cleanup;
    }
    else
    {
        // we use the first argument as the server name/address
#if defined UNICODE
        wcstombs(szRemoteName, argv[1], sizeof(szRemoteName));
#else
        strncpy(szRemoteName, argv[1], sizeof(szRemoteName));
#endif

        Print(TEXT("Communicating with server - %hs\r\n"), szRemoteName);
    }
*/
    //
    // Resolve the server name/address
    //

//MessageBox(NULL, TEXT("Start Client"), TEXT(""), MB_OK);


    memset(&Hints, 0, sizeof(Hints));
    Hints.ai_family = nFamily;
    Hints.ai_socktype = nSockType;

    if(getaddrinfo(szRemoteName, szPort, &Hints, &AddrInfo))
    {
        Print(TEXT("ERROR: Couldn't get resolve the server name/address!"));
        goto Cleanup;
    }

    //
    // Attempt to connect to each address until we find one that succeeds
    //
	AI = AddrInfo;
            sock = socket(AI->ai_family, AI->ai_socktype, AI->ai_protocol);
            if (sock != INVALID_SOCKET)
            {
                if (AI->ai_socktype == SOCK_STREAM)
                {
                    if(connect(sock, AI->ai_addr, AI->ai_addrlen) == SOCKET_ERROR)
                    {
                        // Connect failed, let's close this socket and try again on the next address in the list
                        closesocket(sock);
//						MessageBox(NULL, TEXT("connect succeed"), TEXT(""), MB_OK);
                    }
                }
                // connect() succeeded or we're a UDP socket
            }
			else 
				MessageBox(NULL, TEXT("connect failed"), TEXT(""), MB_OK);

  

    //
    // Send data to the server
    //

//MessageBox(NULL, TEXT("Start to sendto"), TEXT(""), MB_OK);

    cbXfer = 0;
    cbXfer = sendto(sock, pBuf, sizeof(pBuf), 0, AI->ai_addr, AI->ai_addrlen);

    if(cbXfer != sizeof(pBuf))
    {
        Print(TEXT("ERROR: Couldn't send the data! error = %d\r\n"), WSAGetLastError());
        goto Cleanup;
    }

    if (getnameinfo(AI->ai_addr, AI->ai_addrlen,
        szRemoteAddrString, sizeof(szRemoteAddrString), NULL, 0, NI_NUMERICHOST) != 0)
        strcpy(szRemoteAddrString, "");
    
    Print(TEXT("SUCCESS - Sent %d bytes to address %hs\r\n"), cbXfer, szRemoteAddrString);

    //
    // Receive the echo'd data back from the server
    //
// Vincent said : these can not be omitted, and why it failed
/*
    FD_ZERO(&fdReadSet);
    FD_SET(sock, &fdReadSet);
    if(select(0, &fdReadSet, NULL, NULL, &timeout) != 1)
    {
        Print(TEXT("ERROR: Server hasn't responded in %d milliseconds\r\n"), 
            ((timeout.tv_sec * 1000) + (timeout.tv_sec / 1000)));
        goto Cleanup;
    }
*/

    cbTotalRecvd = 0;
    do
    {
        cbRemoteAddrSize = sizeof(ssRemoteAddr);
// If we added below two MessageBox calls and omit select() call, nothing error, if marked them, it failed.
// It seems to fast to call recvfrom(), we should add code to check if sock is available
//MessageBox(NULL, TEXT("client call recvfrom"), TEXT(""), MB_OK);
        cbXfer = recvfrom(sock, pRecvBuf + cbTotalRecvd, sizeof(pRecvBuf) - cbTotalRecvd, 0, 
            (SOCKADDR *)&ssRemoteAddr, &cbRemoteAddrSize);
//MessageBox(NULL, TEXT("client received"), TEXT(""), MB_OK);
        cbTotalRecvd += cbXfer;
    } while(cbXfer > 0 && cbTotalRecvd < sizeof(pRecvBuf));

    if(cbXfer == SOCKET_ERROR)
    {
        Print(TEXT("ERROR: Couldn't receive the data! Error = %d\r\n"), WSAGetLastError());
        goto Cleanup;
    }
    else if(cbTotalRecvd != sizeof(pRecvBuf))
    {
        Print(TEXT("ERROR: Server didn't send back all the expected data!\r\n"));
        goto Cleanup;
    }

    if(nSockType == SOCK_STREAM)
    {
        memset(&ssRemoteAddr, 0, sizeof(ssRemoteAddr));
        cbRemoteAddrSize = sizeof(ssRemoteAddr);
        getpeername(sock, (SOCKADDR *)&ssRemoteAddr, &cbRemoteAddrSize);
    }

    if (getnameinfo((SOCKADDR *)&ssRemoteAddr, cbRemoteAddrSize,
        szRemoteAddrString, sizeof(szRemoteAddrString), NULL, 0, NI_NUMERICHOST) != 0)
        strcpy(szRemoteAddrString, "");

    Print(TEXT("SUCCESS - Received %d bytes back from address %hs\r\n"), cbTotalRecvd, szRemoteAddrString);

Cleanup:

    if(sock != INVALID_SOCKET)
    {
        shutdown(sock, SD_BOTH);
        closesocket(sock);
    }

    if(AddrInfo)
        freeaddrinfo(AddrInfo);

//    WSACleanup();

    return 0;
}