Linux 自虐之路(三): 一个简单的TCP通信例子

环境：服务器端用linux c编写，运行于linux mint18.

           客户端用vs控制台编写，运行于Win 7.

Server端：

main.c文件

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pthread.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <unistd.h>  /* for func: close */
#include <ctype.h>

#include "base_type.h"
#include "util_string.h"

#define MAX_BUF_LEN 1000
const INT g_iServerPort = 8000;

extern BOOL str_toUpper(IN CHAR *srcStr,OUT CHAR *dstStr);

int main()
{
struct sockaddr_in stSockIn;
struct sockaddr_in stSockClient;
INT sock_fd = 0;            /* 监听socket */
INT sock_ConFd =0;          /* 连接socket */
INT addr_size = 0;
CHAR RecvBuf[MAX_BUF_LEN]={0};
CHAR SendBuf[MAX_BUF_LEN]={0};

sock_fd = socket(AF_INET,SOCK_STREAM,0);
if ( -1 == sock_fd)
{
perror("Create socket failed.\n");
return -1;
}
memset(&stSockIn,0,sizeof(struct sockaddr_in));
memset(&stSockClient,0,sizeof(struct sockaddr_in));
stSockIn.sin_family = AF_INET;
stSockIn.sin_addr.s_addr = INADDR_ANY;
stSockIn.sin_port = htons(g_iServerPort);

if ( -1 == bind(sock_fd,(struct sockaddr *)&stSockIn,sizeof(stSockIn)) )
{
perror("Bind socket failed.\n");
return 0;
}
if ( -1 == listen(sock_fd,30) )
{
perror("Listen failed.\n");
return 0;
}
printf("---------server is accepting connection.------\n");
while ( 1 )
{
sock_ConFd= accept(sock_fd,(struct sockaddr *)&stSockClient,&addr_size);
if ( -1 == sock_ConFd)
{
perror("accept failed.\n");
break;
}
else
{
/* 获取已连接的client的地址信息 */
printf("The client IP(%s):Port(%d) connected.\n",
inet_ntoa(stSockClient.sin_addr),
ntohs(stSockClient.sin_port));
}
if ( -1 == recv(sock_ConFd,RecvBuf,MAX_BUF_LEN,0) )
{
perror("Server Recv failed.\n");
break;
}
printf("recv data : %s.\n",RecvBuf);

str_toUpper(RecvBuf,SendBuf);
/* send */
if ( -1 == send(sock_ConFd,SendBuf,strlen(SendBuf),0))
{
perror("Server send data failed!\n");
break;
}
close(sock_ConFd);
}
close(sock_fd);
return 0;
}

base_type.h文件

#ifndef BASE_TYPE_H
#define BASE_TYPE_H

#define UCHAR unsigned char
#define CHAR char
#define UINT unsigned int
#define INT int
#define ULONG unsigned long

typedef enum tagBOOL
{
FALSE=0,
TRUE
}BOOL;
#ifndef IN
#define IN
#endif
#ifndef OUT
#define OUT
#endif
#ifndef INOUT
#define INOUT
#endif

#endif

util_string.h

#ifndef UTIL_STRING_H
#define UTIL_STRING_H

BOOL str_toUpper(IN CHAR *srcStr,OUT CHAR *dstStr);
BOOL str_toLower(IN CHAR * srcStr, OUT CHAR * dstStr);

#endif

util_string.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>

#include "util_string.h"
#include "base_type.h"

/*****************************************************************************
* 函 数 名  : str_toUpper
* 负 责 人  : zhaogang
* 创建日期  : 2016年9月15日
* 函数功能  : 将str中的小写字符转大写字符
* 输入参数  : IN CHAR * srcStr   源str
OUT CHAR * dstStr  结果str（调用者自己保证内存已分配）
* 输出参数  : 无
* 返 回 值  :
* 调用关系  :
* 其    它  :

*****************************************************************************/
BOOL str_toUpper(IN CHAR * srcStr, OUT CHAR * dstStr)
{
INT iIndex = 0;
if ( (NULL == srcStr) || (NULL == dstStr))
{
printf("Invalid params: p1(%p),p2(%p).",srcStr,dstStr);
return FALSE;
}
INT iSrcLen = strlen(srcStr);
for (iIndex = 0; iIndex <iSrcLen; iIndex++)
{
dstStr[iIndex] = (CHAR)toupper(srcStr[iIndex]);
}
dstStr[iIndex]='\0';
return TRUE;
}

/*****************************************************************************
* 函 数 名  : str_toLower
* 负 责 人  : zhaogang
* 创建日期  : 2016年9月15日
* 函数功能  : 将源字符串所有大写转小写
* 输入参数  : IN CHAR * srcStr   原始字符串
OUT CHAR * dstStr  转换后字符串
* 输出参数  : 无
* 返 回 值  :
* 调用关系  :
* 其    它  :

*****************************************************************************/
BOOL str_toLower(IN CHAR * srcStr, OUT CHAR * dstStr)
{
INT iIndex =0;
INT iSrcLen = 0;
if ( (NULL == srcStr) || (NULL == dstStr))
{
printf("Invalid params: p1(%p),p2(%p).",srcStr,dstStr);
return FALSE;
}
iSrcLen = strlen(srcStr);
for (iIndex = 0; iIndex<srcStr ; iIndex++)
{
dstStr[iIndex] = (CHAR)(tolower(srcStr[iIndex]));
}
dstStr[iIndex]='\0';
return TRUE;
}

client端：（win32 --直接抄的网上的）

#include <WINSOCK2.H>
#include <STDIO.H>

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

#define MAX_BUF_LEN 1000

const int g_iServerPort = 8000;
const char g_szServerIP[] = "192.168.16.112";

int main(int argc, char* argv[])
{
WORD sockVersion = MAKEWORD(2, 2);
WSADATA data;
if (WSAStartup(sockVersion, &data) != 0)
{
return 0;
}

SOCKET sclient = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sclient == INVALID_SOCKET)
{
printf("invalid socket !");
return 0;
}

sockaddr_in serAddr;
serAddr.sin_family = AF_INET;
serAddr.sin_port = htons(g_iServerPort);
serAddr.sin_addr.S_un.S_addr = inet_addr(g_szServerIP);
if (connect(sclient, (sockaddr *)&serAddr, sizeof(serAddr)) == SOCKET_ERROR)
{
printf("connect error !");
closesocket(sclient);
return 0;
}
char * sendData = "this is client send data!";
send(sclient, sendData, strlen(sendData), 0);

char recData[MAX_BUF_LEN] = {0};
int ret = recv(sclient, recData, MAX_BUF_LEN, 0);
if (ret > 0)
{
recData[ret] = 0x00;
printf(recData);
}
closesocket(sclient);
WSACleanup();
return 0;
}

测试结果：



说明：

由于我有两台物理机，一台装的win7，一台装的linux mint，因此Client端连接的IP的地址是服务器的IP地址，如果装的虚拟机（双系统），用环回地址127.0.0.1测试。我主要是学习不同操作系统下内核编程，客户端直接copy过来，希望观众查找资料深入理解TCP连接步骤。

此外，写那么多文件是为了日后扩展，毕竟已经上班了，再不能没有规范的乱写代码了。谢谢。