国庆作业6

TCP服务器

#include "head.h"
#define PORT 2580                //端口号
#define IP "192.168.31.219"     //本机IP
int main(int argc, const char *argv[])
{
	sqlite3* db=NULL;
	if(sqlite3_open("./my.db",&db)!=SQLITE_OK)
	{
		fprintf(stderr,"sqlite3_open %d : %s __%d__\n",\
				sqlite3_errcode(db),sqlite3_errmsg(db),__LINE__);
		return -1;
	}
	printf("database open success __%d__\n",__LINE__);
 
	//创建一张表格
	char sql[128]="create table if not exists stu (name char,password int);";
	char* errmsg=NULL;
	if(sqlite3_exec(db,sql,NULL,NULL,&errmsg)!=SQLITE_OK)
	{
		fprintf(stderr,"sqlite3_exec: %s __%d__\n",errmsg,__LINE__);
		return -1;
	}
	printf("create table stu success\n");
 
	//创建流式套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success\n");
	//允许端口快速复用
	int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }                                                                           
    printf("允许端口快速复用成功\n");
 
 
	//填充地址信息结构体给bind函数绑定使用
	//真实地址信息结构体根据地质族指定，AF_INET;man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;               //必须填AF_INET;
	sin.sin_port = htons(PORT);            //端口号网络字节序，1024-49151
	sin.sin_addr.s_addr = inet_addr(IP);   //本机IP的网络字节序，ifconfig
 
	//绑定服务器的地址信息》必须绑定
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
 
	//将套接字转换成被动监听状态
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
 
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	//获取连接成功的客户端信息，生成新的文件描述符
	//该文件描述符才是与客户端通信的文件描述符
	
	//int newfd = accept(sfd,NULL,NULL);
	int newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
	if(newfd<0)
	{
		ERR_MSG("accept");
		return -1;
	}
	printf("[%s:%d] newfd =%d 客户端连接成功__%d__\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);
 
	char buf[128]="";
	ssize_t res=0;
	int flag=0;
	while(1)
	{
		bzero(buf,sizeof(buf));
		//接收数据
		res=recv(newfd,buf,sizeof(buf),0);
		if(res<0)
		{
			ERR_MSG("recv");
			return -1;
		}
		else if(0==res)
		{
			printf("[%s:%d] newfd=%d : 客户端下线 __%d__\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);
			break;
		}
 
		int i;
		char str[128]="";
		for(i=0;buf[i]!=' ';i++)
		{
			str[i]=buf[i];
		}
		str[i]=buf[i];
 
		char sql[128]="select name from stu;";
		char **pre=NULL;
		int row,column;
		char* errmsg=NULL;
		if(sqlite3_get_table(db,sql,&pre,&row,&column,&errmsg)!=SQLITE_OK)
		{
			fprintf(stderr,"sqlite3_get_table: %s \n",errmsg);
			return -1;
		}
		for(int i=0;i<row+1;i++)
		{
			for(int j=0;j<column;j++)
			{
				if(strcmp(str,pre[(i*column)+j])==0)
				{
					printf("重复注册\n");
					flag=1;
				}
			}
		}
		if(flag==0){
			sqlite3_free_table(pre);
 
			char sql1[128]="";
			char* p1=buf+strlen(str);
			sprintf(sql1,"insert into stu values(\"%s\",\"%s\");",str,p1);
			char* errmsg1 = NULL;
			if(sqlite3_exec(db,sql1,NULL,NULL,&errmsg1)!=SQLITE_OK)
			{
				fprintf(stderr,"sqlite3_exec: %s __%d__\n",errmsg1,__LINE__);
				return -1;
			}
			printf("insert success\n");
 
 
			if(strcmp(buf,"quit")==0)
				break;
			//发送数据
			strcat(buf," add success");
			if(send(newfd,buf,sizeof(buf),0)<0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}
		else if(flag==1)
			break;
	}
	//关闭套接字
	close(sfd);
	close(newfd);
	return 0;
}

 TCP客户端

#include "head.h"
#define SER_PORT 2580                //端口号
#define SER_IP "192.168.31.219"     //本机IP
int main(int argc, const char *argv[])
{
	//创建流式套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success\n");
	//允许端口快速复用
	int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }                                                                           
    printf("允许端口快速复用成功\n");
 
 
	//填充地址信息结构体给bind函数绑定使用
	//真实地址信息结构体根据地质族指定，AF_INET;man 7 ip
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;               //必须填AF_INET;
	sin.sin_port = htons(SER_PORT);            //端口号网络字节序，1024-49151
	sin.sin_addr.s_addr = inet_addr(SER_IP);   //本机IP的网络字节序，ifconfig
 
 
	if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");
 
	char buf[128]="";
	ssize_t res=0;
	while(1)
	{
		bzero(buf,sizeof(buf));		
		//发送数据
		printf("请输入账户 密码>> ");
		fgets(buf,sizeof(buf),stdin);
		buf[strlen(buf)-1]='\0';
		if(send(sfd,buf,sizeof(buf),0)<0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");
 
		if(strcmp(buf,"quit")==0)
			break;
 
		bzero(buf,sizeof(buf));
		//接收数据
		res=recv(sfd,buf,sizeof(buf),0);
		if(res<0)
		{
			ERR_MSG("recv");
			return -1;
		}
		else if(0==res)
		{
			printf("[%s:%d] newfd=%d : 服务器下线 __%d__\n",\
					SER_IP,SER_PORT,sfd,__LINE__);
			break;
		}
			printf("[%s:%d] newfd=%d : %s  __%d__\n",\
					SER_IP,SER_PORT,sfd,buf,__LINE__);
	}
	//关闭套接字
	close(sfd);
	return 0;
}