-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtcp_shark_client.c
171 lines (171 loc) · 4.3 KB
/
tcp_shark_client.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
#include<my_head.h>
#include <netinet/ip.h>
#include <sys/select.h>
#include <sqlite3.h>
typedef struct mseeage
{
char option;
char name[32];
char text[256];
}msg_t;
void print2(void)
{
printf("-------------------------------------------------\n");
printf("1.存钱 2.取钱 3.查看存款 4.查看存取记录 5.退出\n");
printf("-------------------------------------------------\n");
printf("请输入您的选择\n");
}
void print1(void)
{
printf("------------------------\n");
printf("1.注册 2.登录 3.退出\n");
printf("------------------------\n");
printf("请输入您的选择\n");
}
msg_t msg;
int do_register(int sockfd,msg_t *msg,sqlite3 *db);//声明注册函数
int do_login(int sockfd,msg_t *msg,sqlite3 *db);//声明登录函数
int do_quit(int sockfd,msg_t *msg,sqlite3 *db);//声明退出函数
int main(int argc, char const *argv[])
{
sqlite3 *my_db;
if(sqlite3_open("shark.db",&my_db)!=SQLITE_OK)
{
printf("error code:%d\n",sqlite3_errcode(my_db));
printf("error:%s\n",sqlite3_errmsg(my_db));
exit(1);
}
if(3 != argc)
{
printf("Usage : %s <ip> <port>\n", argv[0]);
exit(1);
}
int sockfd = 0;
if(-1 == (sockfd = socket(AF_INET, SOCK_STREAM, 0)))
ERR_LOG("socket error");
struct sockaddr_in serveraddr;
memset(&serveraddr, 0, sizeof(serveraddr));
serveraddr.sin_family = AF_INET;
serveraddr.sin_port = htons(atoi(argv[2]));
serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
socklen_t serveraddr_len = sizeof(serveraddr);
int ret=0;
if(-1==(ret=(connect(sockfd,(struct sockaddr *)&serveraddr,serveraddr_len))))
{
ERR_LOG("connect error");
}
while(1)
{
NEXT2:
print1();
scanf("%c",&msg.option);
getchar();
if(msg.option>'3' || msg.option<'1')
{
printf("请输入正确的选择\n");
goto NEXT2;
}
switch(msg.option)
{
case '1'://注册分支
do_register(sockfd,&msg,my_db);
break;
case '2'://登录分支
do_login(sockfd,&msg,my_db);
break;
case '3'://退出分支
do_quit(sockfd,&msg,my_db);
break;
}
}
return 0;
}
int do_quit(int sockfd,msg_t *msg,sqlite3 *db)
{
send(sockfd,msg,sizeof(msg_t),0);
exit(1);
}
int do_register(int sockfd,msg_t *msg,sqlite3 *db)
{
int ret=0;
printf("请输入注册姓名\n");
scanf("%s",msg->name);
getchar();
printf("请输入注册密码\n");
scanf("%s",msg->text);
getchar();
send(sockfd,msg,sizeof(msg_t),0);
recv(sockfd,msg,sizeof(msg_t),0);
printf("%s\n",msg->text);
return 0;
}
int do_login(int sockfd,msg_t *msg,sqlite3 *db)
{
char sqlbuff[300]={0};
char buff[100]={0};
int ret=99;
NEXT:
printf("请输入登录用户名称\n");
scanf("%s",msg->name);
getchar();
printf("请输入登录密码\n");
scanf("%s",msg->text);
getchar();
send(sockfd,msg,sizeof(msg_t),0);
recv(sockfd,msg,sizeof(msg_t),0);
printf("%s\n",msg->text);
sprintf(buff,"用户[%s]登录成功",msg->name);
if(strcmp(buff,msg->text))
{
goto NEXT;
}
while(1)
{
NEXT1:
print2();
scanf("%c",&msg->option);
getchar();
if(msg->option>'5' || msg->option<'1')
{
printf("请输入正确的选择\n");
goto NEXT1;
}
switch(msg->option)
{
case '1':
printf("请输入要存入的数目\n");
scanf("%s",msg->text);
getchar();
send(sockfd,msg,sizeof(msg_t),0);
break;
case '2':
printf("请输入要取出的数目\n");
scanf("%s",msg->text);
getchar();
send(sockfd,msg,sizeof(msg_t),0);
break;
case '3':
send(sockfd,msg,sizeof(msg_t),0);
break;
case '4':
send(sockfd,msg,sizeof(msg_t),0);
while(1)
{
recv(sockfd,msg,sizeof(msg_t),0);
if(!strcmp(msg->text,"over"))
{
goto NEXT1;
}
printf("%s\n",msg->text);
}
break;
case '5':
goto NEXT2;
break;
}
recv(sockfd,msg,sizeof(msg_t),0);
printf("%s\n",msg->text);
}
NEXT2:
return 0;
}