sender.cpp

这是一个学习WinSocket编程的程序

// Sender.cpp : Defines the class behaviors for the application.
//

#include "stdafx.h"
#include "Sender.h"

#include "MainFrm.h"
#include "SenderDoc.h"
#include "SenderView.h"

#include "glb_fun.h"
#include "glb_fun.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif


/*sender SOCKET */
int sender_sockfd;
/*receiver SOCKET */
int receiver_sockfd;
/*channel SOCKET */
int channelsender_sockfd,channelreceiver_sockfd;
//地址
struct sockaddr_in sender_addr,receiver_addr,channelsender_addr,channelreceiver_addr;

WSADATA ws;

BOOL	bSending;							//正在发送标志

char	str_send[MAX_MSG_SIZE];				//对话框输入的待发送信息
char	now_send[MAX_FRAME_LEN];			//当前发送的数据帧
int		iSendno;							//当前发送的数据帧号
int		msglen;								//对话框输入的待发送信息长度	

int		nTimer;								//定时器
int		iSendTime;							//时间计数器


/////////////////////////////////////////////////////////////////////////////
// CSenderApp

BEGIN_MESSAGE_MAP(CSenderApp, CWinApp)
	//{{AFX_MSG_MAP(CSenderApp)
	ON_COMMAND(ID_APP_ABOUT, OnAppAbout)
		// NOTE - the ClassWizard will add and remove mapping macros here.
		//    DO NOT EDIT what you see in these blocks of generated code!
	//}}AFX_MSG_MAP
	// Standard file based document commands
	ON_COMMAND(ID_FILE_NEW, CWinApp::OnFileNew)
	ON_COMMAND(ID_FILE_OPEN, CWinApp::OnFileOpen)
	// Standard print setup command
	ON_COMMAND(ID_FILE_PRINT_SETUP, CWinApp::OnFilePrintSetup)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CSenderApp construction

CSenderApp::CSenderApp()
{
	// TODO: add construction code here,
	// Place all significant initialization in InitInstance
	m_bConnect=0;
	bSending=0;
}

/////////////////////////////////////////////////////////////////////////////
// The one and only CSenderApp object

CSenderApp theApp;


CSenderApp::~CSenderApp()
{
	// TODO: add construction code here,
	// Place all significant initialization in InitInstance
	if (m_bConnect)
	{
		closesocket(sender_sockfd);
		WSACleanup( );
	}
}

/////////////////////////////////////////////////////////////////////////////
// CSenderApp initialization

BOOL CSenderApp::InitInstance()
{
	AfxEnableControlContainer();

	// Standard initialization
	// If you are not using these features and wish to reduce the size
	//  of your final executable, you should remove from the following
	//  the specific initialization routines you do not need.

#ifdef _AFXDLL
	Enable3dControls();			// Call this when using MFC in a shared DLL
#else
	Enable3dControlsStatic();	// Call this when linking to MFC statically
#endif

	// Change the registry key under which our settings are stored.
	// TODO: You should modify this string to be something appropriate
	// such as the name of your company or organization.
	SetRegistryKey(_T("Local AppWizard-Generated Applications"));

	LoadStdProfileSettings();  // Load standard INI file options (including MRU)

	// Register the application's document templates.  Document templates
	//  serve as the connection between documents, frame windows and views.

	CSingleDocTemplate* pDocTemplate;
	pDocTemplate = new CSingleDocTemplate(
		IDR_MAINFRAME,
		RUNTIME_CLASS(CSenderDoc),
		RUNTIME_CLASS(CMainFrame),       // main SDI frame window
		RUNTIME_CLASS(CSenderView));
	AddDocTemplate(pDocTemplate);

	// Parse command line for standard shell commands, DDE, file open
	CCommandLineInfo cmdInfo;
	ParseCommandLine(cmdInfo);

	// Dispatch commands specified on the command line
	if (!ProcessShellCommand(cmdInfo))
		return FALSE;

	// The one and only window has been initialized, so show and update it.
	m_pMainWnd->ShowWindow(SW_SHOW);
	m_pMainWnd->UpdateWindow();

	return Connect();
	//return TRUE;
}


BOOL CSenderApp::Connect() 
{
	// TODO: Add your command handler code here
	// TODO: Add your command handler code here

	char	strerr[100];

	if (m_bConnect) return 1;

	int		addrlen;
	int		errno;
	
	char	*senderaddr = "127.0.0.1",*receiveraddr = "127.0.0.1", *channeladdr = "127.0.0.1";
	int		senderport = 8001, channelport_sender = 7001, channelport_receiver = 7002, receiverport = 6001;

	/*创建连接的SOCKET */
	WSAStartup(0x0101,&ws);
	sender_sockfd = socket(AF_INET,SOCK_DGRAM,0);
	
	if(sender_sockfd < 0)
	{/*创建失败 */
		errno=GetLastError();
		wsprintf(strerr,"socker Error:%s\n",strerror(errno));
		AfxMessageBox(strerr);
		return 0;
	}

	/* 初始化发送端地址*/
	addrlen=sizeof(struct sockaddr_in);
	memset(&sender_addr, 0, addrlen);
	sender_addr.sin_family		= AF_INET;
	sender_addr.sin_addr.s_addr = inet_addr(senderaddr);
	sender_addr.sin_port		= htons(senderport);
	if(bind(sender_sockfd, (struct sockaddr*)&sender_addr, addrlen) < 0)
	{ 
		/*棒定失败 */
		errno = GetLastError();
		wsprintf(strerr, "Bind Error:%s\n",strerror(errno));
		AfxMessageBox(strerr);
		return 0;
	}

	/* 初始化信道地址*/
	addrlen = sizeof(struct sockaddr_in);
	memset(&channelsender_addr, 0, addrlen);
	channelsender_addr.sin_family		= AF_INET;
	channelsender_addr.sin_addr.s_addr	= inet_addr(channeladdr);
	channelsender_addr.sin_port			= htons(channelport_sender);
	
	WSAAsyncSelect(sender_sockfd, GetMainWnd()->m_hWnd, WM_MY_CHANNEL_SEND, FD_READ);
	
	m_bConnect = 1;
	return 1;
}



/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
		// No message handlers
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

// App command to run the dialog
void CSenderApp::OnAppAbout()
{
	CAboutDlg aboutDlg;
	aboutDlg.DoModal();
}

/////////////////////////////////////////////////////////////////////////////
// CSenderApp message handlers