用浏览器浏览图片

最新推荐文章于 2024-01-11 02:01:50 发布

原创最新推荐文章于 2024-01-11 02:01:50 发布 · 1.2k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#浏览器 #socket #null #winapi #file

本文介绍了一个基于Windows的简单文件服务器实现案例，该服务器能够监听指定端口接收客户端请求，并从指定目录中发送文件给客户端。文章涵盖了初始化监听套接字、枚举文件夹中的文件、创建线程处理客户端连接等关键步骤。

#include <windowsx.h>

#include <Winsock2.h>

#include<ShlObj.h>

#include<assert.h>

#include<io.h>

#include<ctime>

usingnamespace std;

#include "resource.h"

#pragmacomment(lib,"ws2_32")

#define WM_SOCKET WM_USER +1024

char g_szAllFileName[256][64];

char g_szDirPath[256];

int g_nFileCount=0;

int g_nNext=0;

LRESULT DlgProc(HWND hwnd,UINT uMessage,

WPARAM wParam,LPARAM lParam);

DWORD WINAPI ThreadProc(LPVOID lParam);

SOCKET InitListenSock(UINT nPort);

int InitDirFileName(char *szDirPath);

int EnumAllFile(char *szDirPath);

int WINAPI WinMain(IN HINSTANCE hInstance, IN HINSTANCE hPrevInstance,

IN LPSTR lpCmdLine, IN int nShowCmd )

{

::DialogBox(hInstance,LPCTSTR(IDD_MYDLG),

NULL,DLGPROC(DlgProc));

return TRUE;

}

LRESULT DlgProc(HWND hwnd,UINT uMessage,

WPARAM wParam,LPARAM lParam)

{

static SOCKET listenSock;

static WSADATA wsaData;

static HANDLE pThread[64];

static UINT nThreadCount=0;

staticint len=sizeof(SOCKADDR_IN);

static UINT uPort=5555;

staticchar szDirPath[MAX_PATH];

staticchar szGetBuf[1024];

staticchar szIP[32];

staticchar szPort[6];

staticint nBufLen=0;

static BROWSEINFO bi;

static LPITEMIDLIST pidl;

SOCKADDR_IN newSockaddr;

SOCKET newAccept;

switch(uMessage) {

case WM_INITDIALOG:

srand((unsignedint)time(NULL));

SetWindowText(GetDlgItem(hwnd,IDC_EDT_PORT),"5555");

SetWindowText(GetDlgItem(hwnd,IDC_EDT_FILEDIR),

"F://NET");

SetWindowText(GetDlgItem(hwnd,IDC_EDT_IP),"http://127.0.0.1:5555");

WSAStartup(MAKEWORD(2,2),&wsaData);

return TRUE;

case WM_CLOSE:

closesocket(listenSock);

WSACleanup();

EndDialog(hwnd,wParam);

PostQuitMessage(0);

return TRUE;

case WM_COMMAND:

switch(LOWORD(wParam))

{

case IDC_BT_CHOOSE:

ZeroMemory(&bi,sizeof(bi));

bi.hwndOwner=NULL;

bi.lpszTitle="选择文件夹";

bi.ulFlags=BIF_RETURNONLYFSDIRS;

pidl = SHBrowseForFolder(&bi);

ZeroMemory(szDirPath,MAX_PATH);

SHGetPathFromIDList(pidl, szDirPath);

SetWindowText(GetDlgItem(hwnd,IDC_EDT_FILEDIR),

szDirPath);

return TRUE;

case IDOK:

GetWindowText(GetDlgItem(hwnd,IDC_EDT_FILEDIR),szDirPath,256);

GetWindowText(GetDlgItem(hwnd,IDC_EDT_PORT),szPort,6);

strcpy(szIP,"http://127.0.0.1:");

strcat(szIP,szPort);

SetWindowText(GetDlgItem(hwnd,IDC_EDT_IP),szIP);

if((g_nFileCount =InitDirFileName(szDirPath))==0)

exit(0);

uPort=atoi(szPort);

listenSock=InitListenSock(uPort);

if(WSAAsyncSelect(listenSock,hwnd,WM_SOCKET,FD_ACCEPT|FD_CLOSE)

==SOCKET_ERROR )

{

MessageBox(NULL,"SAAsyncSelect",NULL,MB_OK);

exit(0);

}

if(listen(listenSock,5)==SOCKET_ERROR)

{

MessageBox(NULL,"listen",NULL,MB_OK);

exit(0);

}

::EnableWindow(::GetDlgItem(hwnd,IDOK),false);

return TRUE;

case IDCANCEL:

EndDialog(hwnd,0);

return TRUE;

}

break;

case WM_SOCKET:

if(WSAGETSELECTERROR(lParam))

{

closesocket((SOCKET)wParam);

return TRUE;

}

switch(WSAGETSELECTEVENT(lParam)) {

case FD_ACCEPT:

if((newAccept=accept(listenSock,(sockaddr*)&newSockaddr,&len))==

INVALID_SOCKET)

{

exit(0);

}

WSAAsyncSelect(newAccept,hwnd,WM_SOCKET,FD_CLOSE|FD_READ|FD_WRITE);

return TRUE;

case FD_READ:

nBufLen= recv((SOCKET)wParam,szGetBuf,1024,0);

szGetBuf[nBufLen]='/0';

return TRUE;

case FD_WRITE:

if((pThread[nThreadCount] = CreateThread(NULL,0,

ThreadProc,(LPVOID)wParam, 0, NULL))!=NULL)

++nThreadCount;

if (nThreadCount==64)

{

nThreadCount=0;

}

return TRUE;

case FD_CLOSE:

MessageBox(NULL,"CLOSE",NULL,MB_OK);

closesocket((SOCKET)wParam);

return TRUE;

}

return TRUE;

}

return FALSE;

}

SOCKET InitListenSock(UINT uPort)

{

SOCKET listenSock;

int iRet=0;

if((listenSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP))==INVALID_SOCKET)

{

MessageBox(NULL,"socket","错误",MB_OK);

exit(1);

}

SOCKADDR_IN listenAddr;

listenAddr.sin_family=AF_INET;

listenAddr.sin_port=htons(uPort);

listenAddr.sin_addr.s_addr=htonl(INADDR_ANY);

iRet=bind(listenSock,(sockaddr*)&listenAddr,sizeof(listenAddr));

if (iRet==SOCKET_ERROR)

{

MessageBox(NULL,"bind","错误",MB_OK);

exit(1);

}

return listenSock;

}

int InitDirFileName(char *szDirPath)

{

char *pStr=g_szDirPath;

while (*szDirPath!='/0')

{

if(*szDirPath=='//')

{

*pStr='//';

++pStr;

*pStr='//';

}

else

*pStr=*szDirPath;

++szDirPath;

++pStr;

}

*pStr='//';

++pStr;

*pStr='//';

++pStr;

*pStr='/0';

return EnumAllFile(g_szDirPath);

}

int EnumAllFile(char *szDirPath)

{

int nCount=0;

WIN32_FIND_DATA fd;

ZeroMemory(&fd,sizeof(WIN32_FIND_DATA));

BOOL bFind=TRUE;

char szTempDirPath[256];

strcpy(szTempDirPath,szDirPath);

strcat(szTempDirPath,"*");

HANDLE hFind=FindFirstFile(szTempDirPath,&fd);

assert(hFind);

while( bFind &&nCount<256)

{

if( FindNextFile(hFind, &fd) )

{

if( (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)

&& strcmp(fd.cFileName, ".") && strcmp(fd.cFileName, "..") )

continue;

elseif( strcmp(fd.cFileName, ".") && strcmp(fd.cFileName, "..") )

{

strcpy(g_szAllFileName[nCount],fd.cFileName);

++nCount;

}

else

bFind=FALSE;

}

return nCount;

}

DWORD WINAPI ThreadProc(LPVOID lParam)

{

int iRet=0;

char szFileName[256];

strcpy(szFileName,g_szDirPath);

if (g_nNext==g_nFileCount)

{

g_nNext=0;

}

strcat(szFileName,g_szAllFileName[g_nNext]);

++g_nNext;

int file =open(szFileName,0x100|002);

int len;

char *buf=NULL;

len=filelength(file);

buf=(char*)malloc(len);

if (buf==NULL)

{

free(buf);

exit(0);

}

read(file,buf,len);

SOCKET newAccept=(SOCKET)lParam;

if((iRet=send(newAccept,buf,len,0))==SOCKET_ERROR)

{

free(buf);

exit(0);

}

shutdown(newAccept,SD_RECEIVE);

closesocket(newAccept);

free(buf);

return TRUE;

}