C / C++ / MFC

BOOL CServer::SocketForDownload(int port, CString DirectoryPath)
{
   m_iPort = port;	
   m_sDirectoryPath = DirectoryPath;
   m_pDLThread = AfxBeginThread(DLThreadProc, this);
   return SOCKET_OK;
}

UINT DLThreadProc(LPVOID pParam)
{
   CServer *pNW = (CServer*)pParam;
   SOCKET DLSocket;
   SOCKADDR_IN	DLAddr;  

   DLSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
   if(DLSocket == INVALID_SOCKET)
   {
      // error handling code
   }
   int optval, rc;
   optval = 1;
   rc = setsockopt(DLSocket, SOL_SOCKET, SO_REUSEADDR, (char *)&optval, sizeof(optval));
	
   DLAddr.sin_family = AF_INET;
   DLAddr.sin_port = htons(pNW->m_iPort);
   DLAddr.sin_addr.s_addr = htonl(INADDR_ANY);

   if(bind(DLSocket, (SOCKADDR*) &DLAddr, sizeof(DLAddr)) == SOCKET_ERROR)
   {
      // error handling code
   }

   if (listen(DLSocket, 5) == SOCKET_ERROR)
   {
      // error handling code
   }
   int len;
   SOCKET AcceptSocket;
   while(TRUE)
   {
        len = sizeof(DLAddr);
        AcceptSocket = SOCKET_ERROR;
        while (AcceptSocket == SOCKET_ERROR)
        {
 //        AcceptSocket = accept(DLSocket, NULL, NULL );
           AcceptSocket = accept(DLSocket, (SOCKADDR*) &DLAddr, &len);
	// the above two are all the same result, hang there sometimes     
           if(AcceptSocket == INVALID_SOCKET)
           {
	     // error handling code
           }
        }
        DLSocket = AcceptSocket; 
        break;
    }
  
    // transfer part start
    // transfer code
    // transfer part end	
	
    shutdown(AcceptSocket, SD_SEND);
//  shutdown(DLSocket, SD_SEND);
	
    int returncode;	
    returncode = closesocket(AcceptSocket);
//  returncode = closesocket(DLSocket);
    
    AfxEndThread(SOCKET_OK);
    return SOCKET_OK;
}

BOOL CClient::SocketForUpload(char *ip, char *filepath)
{
    m_sFilePath = filepath;
    m_sIP = ip;
    m_pULThread = AfxBeginThread(ULThreadProc, this);
    return SOCKET_OK;
}
UINT ULThreadProc(LPVOID pParam)
{
    CClient *pNW = (CClient *)pParam;
    SOCKET DLSocket;
    SOCKADDR_IN	DLAddr;

    if((DLSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET)
    {
        // error handling code	
    }
    Sleep(500);

    DLAddr.sin_family = AF_INET;
    DLAddr.sin_port = htons(pNW->m_iPort - 1);
    DLAddr.sin_addr.s_addr = inet_addr(pNW->m_sIP);

    if(connect(DLSocket, (SOCKADDR *) &DLAddr, sizeof(DLAddr)) == SOCKET_ERROR)
    {
       // error handling code
    }
     
    // transfer part start
    // transfer code
    // transfer part end

    shutdown(DLSocket, SD_SEND);
    int returncode = closesocket(DLSocket);
    
    AfxEndThread(SOCKET_CLOSE);
    return SOCKET_OK;
}