www.pudn.com > tcpsocket.rar > recvDlg.cpp, change:1999-01-10,size:6965b
// recvDlg.cpp : implementation file
//
#include "stdafx.h"
#include "recv.h"
#include "recvDlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
volatile int threadController;
char* name;
/////////////////////////////////////////////////////////////////////////////
// CRecvDlg dialog
CRecvDlg::CRecvDlg(CWnd* pParent /*=NULL*/)
: CDialog(CRecvDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CRecvDlg)
m_tmp = _T("");
//}}AFX_DATA_INIT
// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CRecvDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CRecvDlg)
DDX_Control(pDX, IDC_LIST1, m_list);
DDX_Text(pDX, IDC_EDIT1, m_tmp);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CRecvDlg, CDialog)
//{{AFX_MSG_MAP(CRecvDlg)
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_BN_CLICKED(IDC_BUTTON1, OnButton1)
ON_BN_CLICKED(IDC_BUTTON2, OnButton2)
ON_COMMAND(ID_MENUITEM32773, OnMenuitem32773)
ON_COMMAND(ID_MENUITEM32772, OnMenuitem32772)
//}}AFX_MSG_MAP
ON_MESSAGE(WM_THREADENDED, OnThreadended)
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CRecvDlg message handlers
BOOL CRecvDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// Set the icon for this dialog. The framework does this automatically
// when the application's main window is not a dialog
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
// TODO: Add extra initialization here
name = new char[256];
return TRUE; // return TRUE unless you set the focus to a control
}
// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.
void CRecvDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
// Center icon in client rectangle
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// Draw the icon
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
// The system calls this to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CRecvDlg::OnQueryDragIcon()
{
return (HCURSOR) m_hIcon;
}
void CRecvDlg::OnButton1()
{
WORD wVersionRequested;
WSAData wsaData;
struct sockaddr_in client;
struct sockaddr_in server;
char buf[256];
int s;
int clientfd;
int err;
//start socket
wVersionRequested = MAKEWORD( 2, 0 );
err = WSAStartup( wVersionRequested, &wsaData );
if ( err != 0 ) {
MessageBox("0","000",MB_OK);}
if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 0 ) {
MessageBox("1","000",MB_OK);
WSACleanup( );
return;
}
//create socket
if ((s=socket(AF_INET, SOCK_STREAM,0))<0){
MessageBox("err","sockr err",MB_OK);
}
else
MessageBox("¹þ","sockr create ok",MB_OK);
//bind socket
server.sin_family = AF_INET;
server.sin_port = htons(1024);
server.sin_addr.s_addr = INADDR_ANY;
if (bind(s, (struct sockaddr *)&server, sizeof(server)) == SOCKET_ERROR){
MessageBox("err","bind err",MB_OK);
}
else
MessageBox("¹þ","bind ok",MB_OK);
int length = sizeof(server);
if(getsockname(s,(struct sockaddr *)&server,&length) < 0){
MessageBox("","getsockname_err",MB_OK);
}
else
MessageBox("¹þ","getsockname ok",MB_OK);
if(listen(s,1)!=0){
MessageBox("err","listen err",MB_OK);
}
else
MessageBox("ok","listen ok",MB_OK);
int tmp=sizeof(server);
int flag=0;
int clientnamelen=sizeof(client);
int pktlentmp;
int tmpcount = 0;
CString tmpstr1;
// LPCTSTR tmpstr1;
// while(1){
for(int ii=0;ii<5;ii++){
if((clientfd=accept(s,(struct sockaddr *)&client,&clientnamelen))==-1){
MessageBox("err","accept err",MB_OK);
}
// else
// MessageBox("ok","accept ok",MB_OK);
// int pktlentmp;
// while(1){
if ((pktlentmp = recv(clientfd,buf,256,0))<=0){
MessageBox("err","recv err",MB_OK);
break;
}
else{
// MessageBox(buf,"recv ok",MB_OK);
tmpcount++;
tmpstr1.Format("%s+%d",buf,tmpcount);// = buf;
m_list.AddString(tmpstr1);
UpdateData(false);
// break;
}
}
closesocket(clientfd);
closesocket(s);
WSACleanup( );
}
HANDLE ThreadProc(LPVOID param)
{
WORD wVersionRequested;
WSAData wsaData;
struct sockaddr_in client;
struct sockaddr_in server;
char buf[256];
int s;
int clientfd;
int err;
//start socket
wVersionRequested = MAKEWORD( 2, 0 );
err = WSAStartup( wVersionRequested, &wsaData );
if ( err != 0 ) return 0;
if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 0 ) {
WSACleanup( );
return 0;
}
//create socket
if ((s=socket(AF_INET, SOCK_STREAM,0))<0){
return 0;
}
//bind socket
server.sin_family = AF_INET;
server.sin_port = htons(1024);
server.sin_addr.s_addr = INADDR_ANY;
if (bind(s, (struct sockaddr *)&server, sizeof(server)) == SOCKET_ERROR){
return 0;
}
int length = sizeof(server);
if(getsockname(s,(struct sockaddr *)&server,&length) < 0){
return 0;
}
if(listen(s,1)!=0){
return 0;
}
int tmp=sizeof(server);
int flag=0;
int clientnamelen=sizeof(client);
int pktlentmp;
int tmpcount = 0;
CString tmpstr1;
// while(1){
for(int ii=0;ii<5;ii++){
if((clientfd=accept(s,(struct sockaddr *)&client,&clientnamelen))==-1){
return 0;
}
if ((pktlentmp = recv(clientfd,buf,256,0))<=0){
return 0;
break;
}
else{
tmpcount++;
tmpstr1.Format("%s+%d",buf,tmpcount);
strcpy(name,tmpstr1);
Sleep(20);
::PostMessage((HWND)param, WM_THREADENDED, 0, 0);
}
}
closesocket(clientfd);
closesocket(s);
WSACleanup( );
::MessageBox((HWND)param, "Thread stopped.", "Thread", MB_OK); return NULL;
return 0;
}
void CRecvDlg::OnButton2()
{
HWND hWnd = GetSafeHwnd();
threadController = 1;
AfxBeginThread((AFX_THREADPROC)ThreadProc, hWnd, THREAD_PRIORITY_NORMAL);
}
void CRecvDlg::OnMenuitem32773()
{
threadController = 0;
}
void CRecvDlg::OnMenuitem32772()
{
HWND hWnd = GetSafeHwnd();
threadController = 1;
AfxBeginThread((AFX_THREADPROC)ThreadProc, hWnd, THREAD_PRIORITY_NORMAL);
}
LONG CRecvDlg::OnThreadended(WPARAM wParam, LPARAM lParam)
{
// CString tmp;
// tmp.Format("%d",wParam);
// AfxMessageBox(tmp);
// AfxMessageBox("Thread ended.");
m_list.AddString(_T(name));
UpdateData(false);
return 0;
}