利用流式套接字傳輸資料檔案
阿新 • • 發佈：2019-02-03
伺服器端：
1>建立一個基於對話方塊的應用程式StreamSocketServer
2>設計介面，將“確定”按鈕的標題改為“啟動”
3>雙擊“啟動按鈕”，新增以下程式碼，初始化網路，啟動伺服器監聽
void CStreamSocketServerDlg::OnOK()
{
// TODO: Add extra validation here
WSADATA wsaData;
/*
typedef struct WSAData
{
WORD wVersion; //為希望使用的Winsock版本
WORD wHeighVersion; //返回現有的Winsock庫的最高版本 4>註冊自定義訊息NETWORK_EVENT, 並新增其相應處理函式OnNetEvent
在StreamSocketServerDlg.h中新增成員函式：
void OnNetEvent(WPARAM wParam, LPARAM lParam);在StreamSocketServerDlg.cpp中實現：
void CStreamSocketServerDlg::OnNetEvent(WPARAM wParam, LPARAM lParam)
{
int iEvent = WSAGETSELECTEVENT(lParam); //呼叫Winsock API函式，得到網路事件型別
SOCKET CurSock = (SOCKET)wParam; //呼叫Winsock API函式，得到此事件的客戶端套接字
switch(iEvent)
{
case FD_ACCEPT: //客戶端連線請求事件
TRACE("Get client call!\n");
OnAccept(CurSock);
break;
case FD_CLOSE: //客戶端斷開事件
OnClose(CurSock);
break;
case FD_READ: //客戶端網路包到達事件
OnReceive(CurSock);
break;
case FD_WRITE: //傳送網路資料事件
break;
default: break;
}
}5>新增客戶端的連線請求處理函式
void CStreamSocketServerDlg::OnAccept(SOCKET CurSock)
{
SOCKET TempSock;
struct socaaddr_in ConSocketAddr;
int addrlen;
addrlen = sizeof(ConSocketAddr);
TempSock = accept(CurSock, (struct sockaddr*)&ConSocketAddr, &addrlen);
/*
用於接受連線請求
SCOKET accept(
SCOKET s, //Socket的識別碼
struct sockaddr FAR * addr, //存放連線的客戶端地址
int FAR* addrlen //地址長度
)
*/
if(ConnectSock == INVALID_SOCKET)
MessageBox("INVALID_SOCKET");
TRACE("建立新連線，sock:%d\n", TempSock); //輸出除錯資訊
inet_ntoa(ConSocketAddr.sin_addr);
ConnectSock = TempSock;
char *filename = "c:\\001.doc";
if(!SendFile(filename, ConnectSock))
MessageBox("傳送失敗！");
}6>添加發送資料檔案函式
BOOL CStreamSocketServerDlg::SendFile(char *name, SOCKET conn)
{
char *FileName = name;
SOCKET TcpConn = conn;
CFile file;
if(!file.Open(FileName, CFile::modeRead))
{
printf("開啟%s失敗！\n", FileName);
return FALSE;
}
int NumBytes; //用來儲存每次傳送時資料塊的大小
UINT Total = 0; //用來儲存套接字已經傳送的總的位元組數
int BufSize = 1024; //傳送緩衝區的大小
int Size = BufSize; //讀取檔案的大小
LPBYTE pBuf = new BYTE[BufSize]; //傳送緩衝區
DWORD dwTemp = 0;
UINT FileLength = file.GetLength(); //得到檔案的大小併發送出去
send(TcpConn, (char *)&FileLength, 4, 0);
/*
int send( //返回傳送的字元總數
SOCKET s,
const char FAR *buf, //存放要傳送的資料的暫存區
int len, //buf的長度
int flags //此函式被呼叫的方式
)
*/
file.SeekToBegin();
while(Total < FileLength)
{
if((int)(FileLength - Total) < Size)
Size = FileLength - Total;
Size = file.Read(pBuf, Size);
/*
virtual UINT Read(
void* lpBuf, //是把資源讀入哪裡
UINT nCount //是讀入的位元組數
);
*/
NumBytes = send(TcpConn, (char *)pBuf, Size, 0);
if(NumBytes == SOCKET_ERROR)
{
send(TcpConn, "ERROR", sizeof("ERROR")+1, 0);
return FALSE;
}
Total += NumBytes;
file.Seek(Total, CFile::begin);
}
delete[] pBuf;
file.Close();
closesocket(TcpConn);
return TRUE;
}客戶端：
1>新建一個基於對話方塊的程式StreamSocketClient
2>設計介面，新增編輯框m_CtrlPAddress輸入IP地址，m_iPort用於輸入埠，將“確定”按鈕改為“連線”，“取消”改為“關閉”
3>雙擊“連線”，新增以下程式碼：
void CStreamSocketClientDlg::OnOK()
{
// TODO: Add extra validation here
UpdateData(FALSE); //初始化對話方塊中的資料
WORD wVersionRequested;
WSADATA wsaData;
int err;
wVersionRequested = MAKEWORD(2, 2); //連線兩個給定的無符號引數
err = WSAStartUp(wVersionReqested, &wsaData);
if(err != 0)
return;
if(LOBYTE(wsaData.wVersion) != 1) || HIBYTE(wsaData.wVersion) != 1) //LOBYTE()得到一個16bit數
最低(最右邊)那個位元組
{
WSACleanup();
return;
}
SOCKET TcpClient = socket(AF_INET, SOCK_STREAM, 0);
SOCKADDR_IN SerAddr;
SerAddr.sin_family AF_INET;
SerAddr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
/*
struct in_addr{
union{
struct{ u_char s_b1, s_b2, s_b3, s_b4;} S_un_b;
struct{u_short S_w1, S_w2} S_un_w;
u_long S_addr;
}S_un;
};
*/
SerAddr.sin_port = htos(m_iPort);
connect(TcpClient, (SOCKADDR*)&SerAddr, sizeof(SerAddr)); //與伺服器端建立連線
/*
int connect(
SOCKET s, //伺服器端Socket的識別碼
const struct sockaddr FAR* name, //Socket想要連線的對方地址
int namelen //地址長度
)
*/
long FileLength;
recv(TcpClient, (char *)&FileLength, sizeof(long), 0);
/*
int recv(
SOCKET s,
char FAR *buf, //存放接收到資料的暫存區
int len, //buf的長度
int flags //此函式的呼叫方式
)
*/
if(RecvFile("c:\\003.doc", TcpClient, FileLength))
MessageBox("傳輸結束！\n");
else
MessageBox("傳輸失敗！\n");
CDialog::OnOK();
}4>新增接收檔案函式
BOOL CStreamSocketClientDlg::RecvFile(char *name, SOCKET conn, UINT filelen)
{
char *FileName = name;
SOCKET client = conn;
CFile file;
if(!file.Open(FileName, CFile::modeWrite))
{
printf("開啟%s失敗！\n", FileName);
return FALSE;
}
int NumBytes; //用來儲存每次接收時資料塊的大小
UINT Total = 0; //用來儲存套接字已經接收的總的位元組數
int BufSize = 1024; //接收緩衝區的大小
int Size = BufSize; //寫檔案的大小
LPBYTE pBuf = new BYTE[BufSize]; //接收緩衝區
DWORD dwTemp = 0;
file.SeekToBegin();
while(Total < filelen)
{
if((int)(filelen - Total) < Size)
Size = filelen - Total;
NumBytes = recv(client, (char *)pBuf, Size, 0);
if(NumBytes == SOCKET_ERROR)
{
printf("接收失敗！\n");
return FALSE;
}
file.Write(pBuf, NumBytes);
/*
virtual void Write(
const void* lpBuf, //儲存要寫的東西的字串或者字元陣列
UINT nCount ); //要從這個字串或者字元陣列中寫多少個字元到檔案中
*/
Total += NumBytes;
file.Seek(Total, CFile::begin);
}
delete[] pBuf;
file.Close();
closesocket(client);
return TRUE;
}介面：
