国庆作业6

#include <myhead.h>
#include <sqlite3.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr," __%d__", __LINE__);\
	perror(msg);\
}while(0)

#define PORT 2222
#define IP "192.168.1.16"

int do_select(sqlite3* db);
int do_insert(sqlite3* db);

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");

	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	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, (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__);

	sqlite3 *db = NULL;
	if(sqlite3_open("./name.db", &db) != SQLITE_OK)
	{
		ERR_MSG("sqlite3_open");
		return -1;
	}
	printf("database open success __%d__\n", __LINE__);

	char sql[128] = "create table if not exists name1 (name char, passwd char);";
	char *errmsg = NULL;

	if(sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		ERR_MSG("sqlite3_exec");
		return -1;
	}
	printf("create table name1 success __%d__\n", __LINE__);

	char choose = 0;
	while(1)
	{
		printf("----------------------------------------\n");
        printf("-----------------1、插入----------------\n");
        printf("-----------------2、查询----------------\n");
        printf("-----------------3、退出----------------\n");

		printf("请输入>>> ");
		choose = getchar();
		while(getchar() != 10);

		switch(choose)
		{
		case '1':
			do_insert(db);
			break;
		case '2':
			do_select(db);
			break;
		case '3':
			goto END;
		}
	}

END:

	if(sqlite3_close(db) != SQLITE_OK)
	{
		ERR_MSG("sqlite3_close");
		return -1;
	}
	printf("database close success __%d__\n", __LINE__);

	close(sfd);
	return 0;
}

int do_select(sqlite3* db)
{
	char sql[128] = "select * from name1;";
	char** pres = NULL;
	int row, column;
	char* errmsg = NULL;

	if(sqlite3_get_table(db, sql, &pres, &row, &column, &errmsg) != SQLITE_OK)
	{
		ERR_MSG("sqlite3_get_table");
		return -1;
	}
	printf("查询完毕\n");

	printf("row=%d column=%d pres=%p\n", row, column, pres);

	for(int i=0;i<(row+1)*column;i++)
	{
		printf("%s\t", pres[i]);

		if((i+1)%column == 0)
			putchar('\n');

	}

	printf("====================\n");
	for(int i=0;i<row+1;i++)
	{
		for(int j=0;j<column;j++)
		{
			printf("%s\t", pres[(i*column)+j]);
		}
		putchar(10);
	}

	sqlite3_free_table(pres);

	return 0;
}

int do_insert(sqlite3* db)
{
	char name[20] = "";
	char passwd[20] = "";

	printf("请输入name passwd >>> ");
	scanf("%s %s", name, passwd);
	while(getchar()!=10);

	char sql[128] = "";
	char *errmsg = NULL;

	sprintf(sql, "insert int name1 values(\"%s\", \"%s\");", name, passwd);
	printf("sql = %s\n", sql);

	if(sqlite3_exec(db, sql, NULL, NULL, &errmsg) != SQLITE_OK)
	{
		ERR_MSG("sqlite3_exec");
		return -1;
	}
	printf("insert success\n");

	return 0;
}

#include <myhead.h>

#define ERR_MSG(msg) do{\
	fprintf(stderr," __%d__", __LINE__);\
	perror(msg);\
}while(0)

#define SER_PORT 2222
#define SER_IP "192.168.1.16"

int main(int argc, const char *argv[])
{
	int cfd = socket(AF_INET, SOCK_STREAM, 0);
	if(cfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success\n");

	int reuse = 1;
	if(setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口复用\n");

	struct sockaddr_in sin;
	sin.sin_family      = AF_INET;
	sin.sin_port        = htons(SER_PORT);
	sin.sin_addr.s_addr = inet_addr(SER_IP);

	if(connect(cfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	char buf[128] = "";
	//char name[30] = "";
	//char passwd[30] = "";

	while(1)
	{
		bzero(buf, sizeof(buf));
		printf("请输入name passwd >>> ");
		fgets(buf, sizeof(buf), stdin);
		buf[strlen(buf)-1] = '\0';

		if(send(cfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");



	}

	close(cfd);
	return 0;
}