[国嵌攻略][183][账号管理子系统设计]
登录模块设计
1.输入用户名和密码
2.根据用户名从数据库提取密码
3.比较用户输入密码和数据库提取密码，以决定是否登录成功
编译客户端程序
arm-linux-gcc
-L ../../008/openssl-1.0.0s/_install/lib/ -lssl -lcrypto
-I ../../008/openssl-1.0.0s/_install/include/
-L ../../010/sqlite-autoconf-3070800/_install/lib/ -lsqlite3
-I ../../010/sqlite-autoconf-3070800/_install/include/
client.c -o client
创建数据库
./db client.db "create table login(username varchar(20), password varchar(20));”
插入数据
./db client.db "insert into login values('yuan','123456');"
./db client.db "insert into login values('chen','654321');"
client.c
/******************************************************************** *名称:client.c *作者:D *时间:2016.04.03 *功能:网络安全传输系统客户端 *********************************************************************/ /******************************************************************** *头文件 *********************************************************************/ #include <stdio.h> #include <stdlib.h> #include <string.h> #include <fcntl.h> #include <sys/stat.h> #include <sys/socket.h> #include <netinet/in.h> #include <openssl/ssl.h> #include <openssl/err.h> #include <sqlite3.h> /******************************************************************** *宏定义 *********************************************************************/ #define SERVER_PORT 3333 //网络端口 /******************************************************************** *函数原型 *********************************************************************/ int main(int argc, char **argv); int login(); int callback(void *passback, int argc, char **argv, char **colName); void menu(SSL *ssl); void upload(SSL *ssl); void download(SSL *ssl); void quit(SSL *ssl); /******************************************************************** *名称:main *参数: * argc 参数数量 * argv 参数列表 *返回: * stat 0 成功 * -1 失败 *功能:主函数 *********************************************************************/ int main(int argc, char **argv){ //参数检查 if(argc != 2){ printf("Usage:\n\t./client <ip address>\n"); return -1; } //用户登录 if(login() != 0){ printf("Wrong username or password!\n"); return -1; } //创建 SSL SSL_CTX *ctx; SSL_library_init(); OpenSSL_add_all_algorithms(); SSL_load_error_strings(); ctx = SSL_CTX_new(SSLv23_client_method()); //客户端模式 //创建标识 int clientfd; clientfd = socket(AF_INET, SOCK_STREAM, 0); if(clientfd == -1){ printf("Can not create socket!\n"); return -1; } //建立连接 struct sockaddr_in serverAddr; int isConnect; serverAddr.sin_family = AF_INET; //设置协议 serverAddr.sin_port = htons(SERVER_PORT); //设置端口 serverAddr.sin_addr.s_addr = inet_addr(argv[1]); //设置地址 bzero(serverAddr.sin_zero, 8); //设置为零 isConnect = connect(clientfd, (struct sockaddr *)&serverAddr, sizeof(struct sockaddr)); if(isConnect == -1){ printf("Can not connect to server!\n"); return -1; } //连接 SSL SSL *ssl; ssl = SSL_new(ctx); SSL_set_fd(ssl, clientfd); SSL_connect(ssl); //显示菜单 menu(ssl); //关闭 SSL SSL_shutdown(ssl); SSL_free(ssl); //关闭连接 close(clientfd); //释放 SSL SSL_CTX_free(ctx); return 0; } /******************************************************************** *名称:login *参数: * none *返回: * success 0 成功 * other 失败 *功能:用户登录 *********************************************************************/ int login(){ //输入用户名称 char username[20]; printf("Username: "); scanf("%s", &username); //输入用户密码 char password[20]; printf("Password: "); scanf("%s", &password); //打开数据库 sqlite3 *db; sqlite3_open("client.db", &db); //提取用户密码 char satement[80]; char passback[20]; sprintf(satement, "select password from login where username='%s';", username); sqlite3_exec(db, satement, callback, (void *)passback, NULL); //关闭数据库 sqlite3_close(db); //比较用户密码 int success; success = strcmp(password, passback); return success; } int callback(void *passback, int argc, char **argv, char **colName){ //提取用户密码 int i; for(i = 0; i < argc; i++){ if(strcmp(colName[i], "password") == 0){ strcpy((char *)passback, argv[i]); } } return 0; } /******************************************************************** *名称:menu *参数: * ssl 客户端标志 *返回: * none *功能:显示菜单 *********************************************************************/ void menu(SSL *ssl){ //显示菜单 while(1){ //打印菜单 printf("\n"); printf("**********************************************************************\n"); printf("* Client *\n"); printf("*[1]Upload Files *\n"); printf("*[2]Download Files *\n"); printf("*[3]Exit *\n"); printf("**********************************************************************\n"); printf("Please Select: "); //输入命令 int num; scanf("%d", &num); //处理命令 switch(num){ //上传文件 case 1: upload(ssl); break; //下载文件 case 2: download(ssl); break; //退出程序 case 3: quit(ssl); break; //错误命令 default: printf("\nPlease input again!\n"); break; } //是否退出 if(num == 3){ break; } } } /******************************************************************** *名称:upload *参数: * ssl 客户端标志 *返回: * none *功能:上传文件 *********************************************************************/ void upload(SSL *ssl){ //输入文件名称 char filename[20]; printf("\nUpload file: "); scanf("%s", &filename); //打开上传文件 int fd; fd = open(filename, O_RDONLY); if(fd == -1){ printf("Can not open file!\n"); return ; } //发送上传命令 char cmd = 'U'; SSL_write(ssl, (void *)&cmd, sizeof(cmd)); //发送文件名称 int namesize; namesize = strlen(filename) + 1; //加上字符串接收符 SSL_write(ssl, (void *)&namesize, sizeof(namesize)); SSL_write(ssl, (void *)&filename, namesize); //发送文件长度 struct stat fstat; int isState; isState = stat(filename, &fstat); if(isState == -1){ printf("Can not get file state!\n"); return ; } SSL_write(ssl, (void *)&(fstat.st_size), sizeof(fstat.st_size)); //发送文件内容 char buf[1024]; int num; num = read(fd, (void *)buf, sizeof(buf)); while(num > 0){ //发送文件内容 SSL_write(ssl, (void *)&buf, num); //读取文件内容 num = read(fd, (void *)buf, sizeof(buf)); } //关闭上传文件 close(fd); } /******************************************************************** *名称:download *参数: * ssl 客户端标志 *返回: * none *功能:下载文件 *********************************************************************/ void download(SSL *ssl){ //输入文件名称 char filename[20]; printf("\nDownload file: "); scanf("%s", &filename); //创建下载文件 int fd; fd = open(filename, O_RDWR | O_CREAT, 0777); if(fd == -1){ printf("Can not create file!\n"); return ; } //发送下载命令 char cmd = 'D'; SSL_write(ssl, (void *)&cmd, sizeof(cmd)); //发送文件名称 int namesize; namesize = strlen(filename) + 1; //加上字符串接收符 SSL_write(ssl, (void *)&namesize, sizeof(namesize)); SSL_write(ssl, (void *)&filename, namesize); //接收文件长度 int fileszie; SSL_read(ssl, &fileszie, sizeof(fileszie)); //接收文件内容 char buf[1024]; int num; num = SSL_read(ssl, (void *)buf, sizeof(buf)); while(num > 0){ //写入接收内容 write(fd, (void *)&buf, num); //是否接收结束 fileszie = fileszie - num; if(fileszie == 0){ break; } //接收文件内容 num = SSL_read(ssl, (void *)buf, sizeof(buf)); } //关闭下载文件 close(fd); } /******************************************************************** *名称:quit *参数: * ssl 客户端标志 *返回: * none *功能:退出程序 *********************************************************************/ void quit(SSL *ssl){ //发送退出命令 char cmd = 'Q'; SSL_write(ssl, (void *)&cmd, sizeof(cmd)); //清除屏幕显示 system("clear"); }
