Windows 基于udp的socket编程（课程代码基础上修改）

/*
 * Udpclient.cpp : 定义控制台应用程序的入口点。
 *
 */


/*
 * author: zchimp
 * date: 2016.03.10
 */
#include "stdafx.h"
#pragma comment(lib, "Ws2_32.lib")


#include <winsock2.h>
#include <ws2tcpip.h>
#include <stdio.h>
#include <windows.h>


int _tmain( int argc, _TCHAR* argv[] )
{
/* step 1: 初始化Windows Socket 2.2库 */
	WSADATA wsaData;
	int	ret = WSAStartup( MAKEWORD( 2, 2 ), &wsaData );
	if ( ret != 0 )
	{
		printf( "装载Winsock库出错\n" );
		WSACleanup();
		return(-1);
	}


/* step 2: 创建客户端SOCKET套接字 */
	SOCKET sockClient = socket( AF_INET, SOCK_DGRAM, 0 );
	if ( INVALID_SOCKET == sockClient ) /* 无效套接字 */
	{
		printf( "创建SOCKET套接字出错\n" );
		WSACleanup();
		return(-1);
	}


/* step3:设置对方的IP地址和端口等属性 */


	SOCKADDR_IN addrServ;
	addrServ.sin_addr.S_un.S_addr	= inet_addr( "127.0.0.1" );
	addrServ.sin_family		= AF_INET;
	addrServ.sin_port		= htons( 4999 );


	char sendmsg[1024] = { 0 };


	while ( true )
	{
		char sendmsg[1024] = { 0 };
		printf( "please input your account(username+sercret+information) to register. \n" );
		scanf_s( "%s", sendmsg, 1024 );
		if ( strcmp( sendmsg, "quit" ) == 0 )
		{
			sendto( sockClient, sendmsg, sizeof(sendmsg), 0, (sockaddr *) &addrServ, sizeof(SOCKADDR) );
			break;
		}


		sendto( sockClient, sendmsg, sizeof(sendmsg), 0, (sockaddr *) &addrServ, sizeof(SOCKADDR) );
	}
/* sendto(sockClient, "Hello\n", strlen("Hello\n") + 1, 0, (SOCKADDR*)&addrServ, sizeof(SOCKADDR)); */


	closesocket( sockClient );


	system( "PAUSE" );
	WSACleanup();


	return(0);
}


/*
 * UdpServer.cpp : 定义控制台应用程序的入口点。
 *
 */


/*
 * author: zchimp
 * date: 2016.03.10
 */


#include "stdafx.h"


#pragma comment(lib, "Ws2_32.lib")


#include <winsock2.h>
#include <ws2tcpip.h>
#include <stdio.h>
#include <windows.h>


int _tmain( int argc, _TCHAR* argv[] )
{
/* step1:初始化Windows Socket 2.2库 */
	WSADATA wsaData;
	int	ret = WSAStartup( MAKEWORD( 2, 2 ), &wsaData );
	if ( ret != 0 )
	{
		printf( "装载Winsock库出错\n" );
		WSACleanup();
		return(-1);
	}


/* step2:创建一个socket */
	SOCKET sockSrv = socket( AF_INET, SOCK_DGRAM, 0 );
	if ( INVALID_SOCKET == sockSrv )
	{
		printf( "创建SOCKET套接字出错\n" );
		WSACleanup();
		return(-1);
	}


/* step3:绑定IP地址、端口等信息到socket上 */


	SOCKADDR_IN addrServ;
	addrServ.sin_addr.S_un.S_addr	= htonl( INADDR_ANY );
	addrServ.sin_family		= AF_INET;
	addrServ.sin_port		= htons( 4999 );


	bind( sockSrv, (SOCKADDR *) &addrServ, sizeof(SOCKADDR) );


	SOCKADDR_IN	addrClient;
	int		length = sizeof(SOCKADDR);
/* char recvBuf[100]; */


	while ( true )
	{
		char	*name	= NULL;
		char	*secret = NULL;
		char	*info	= NULL;
		char	*strsv	= NULL;
		char	recvBuf[100];
		recvfrom( sockSrv, recvBuf, 100, 0, (SOCKADDR *) &addrClient, &length );

		printf( "%s\n", recvBuf );
		if ( strcmp( recvBuf, "quit" ) == 0 ) /* 判断是否退出 */
		{
			break;
		}
/* 解析客户端发送的信息 */
		name	= strtok_s( recvBuf, "+", &strsv );
		secret	= strtok_s( NULL, "+", &strsv );
		info	= strtok_s( NULL, "+", &strsv );
/* 判断数据格式是否符合格式 */
		if ( !name || !secret )
		{
/*
 * char sendmsg[] = "Please input infomation with the right format!\n";
 * sendto(sListen, sendmsg, sizeof(sendmsg), 0, (sockaddr*)&clientAddr, nclientAddr);
 */
			continue;
		}
/*
 * 进行用户文件处理操作
 * FILE *cfPtr=NULL;
 */
		FILE	*cf;
		char	fname[64] = "", fsecret[64] = "", finfo[512] = "";
		if ( !(cf = fopen( "D:\\sfile.txt", "r+" ) ) )
		{
			printf( "Source file cannot be opened\n" );
/* exit(1); */
		}
/* 检查文件中是否有相同的用户名 */
		while ( !feof( cf ) )
		{
			fscanf( cf, "%s%s%s", fname, fsecret, finfo );
			if ( strcmp( fname, name ) == 0 )
			{
				printf( "name has existed! \n" );
/*
 * char sendmsg[] = "name has existed!\n";
 * sendto(sListen, sendmsg, sizeof(sendmsg), 0, (sockaddr*)&clientAddr, nclientAddr);
 */
				break;
			}
		}
/*
 * 将用户注册信息存放到用户文件中
 * 返回注册成功信息
 */
		if ( feof( cf ) )
		{
			fprintf_s( cf, "\n%s %s %s", name, secret, info );
			printf( "REGISTER SUCCESS. \n" );
/*
 * char sendmsg[] = "register success！\n";
 * sendto(sListen, sendmsg, sizeof(sendmsg), 0, (sockaddr*)&clientAddr, nclientAddr);
 */
		}
/* fclose(cfPtr); */
		fclose( cf );
/*
 * }
 * 关闭与客户通讯的套接字
 */
		printf( "ending...\n" );
		closesocket( sockSrv );
		break;
	}
/* recvfrom(sockSrv, recvBuf, 100, 0, (SOCKADDR*)&addrClient, &length); */


/* printf("%s", recvBuf); */


	closesocket( sockSrv );


	WSACleanup();


	system( "PAUSE" );
	return(0);
}