网络版ATM项目的实现——客户端

网络版的ATM实现思路

客户端

#include "customer.h"

static int sockfd;

/*int create_msg_queue_c(int key)
{
    //创建消息队列
    int msgid = msgget(key,IPC_CREAT|0644);
    if(0 > msgid)
    {
        perror("msgget");
        return -1;
    }
    return msgid;
}*/

void show_menu(void)
{
    printf("Welcome to Online Bank !
");
    printf("---------------------
");
    printf("    [1]开户        
");
    printf("    [2]登录        
");
    printf("    [3]解锁        
");
    printf("    [4]销户        
");
    printf("    [5]退出        
");
    printf("---------------------
");
}
//开户界面
void open_menu(void)
{
    
    Msg msg = {99};
    //得到基本信息
    printf("请输入姓名:");
    gets(msg.acc.name);
    printf("请输入身份证号:");
    gets(msg.acc.idcard);
    printf("请输入预存金额:");
    scanf("%f",&msg.acc.money);
    //清空缓冲区
    stdin->_IO_read_ptr = stdin->_IO_read_end;
    strcpy(msg.acc.password,"123456");
    printf("%d
",msg.acc.account);
    //将获得的基本信息发送到服务端
    send(sockfd,&msg,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg,sizeof(Msg),0);
    //perror("msgrcv");
    printf("*****您的id是%d,初始密码为123456,账号内的余额为%f*****
",msg.acc.account,msg.acc.money);
    sleep(2);
}
//显示登陆后的功能界面
void func_menu(void)
{
    printf("Welcome to Online Bank !
");
    printf("---------------------
");
    printf("    [1]存款        
");
    printf("    [2]取款        
");
    printf("    [3]转账        
");
    printf("    [4]查询        
");
    printf("    [5]修密        
");
    printf("    [6]退出        
");
    printf("---------------------
");
}
//退出登录
void logout(int id)
{
    
    Msg msg = {50};
    msg.acc.account = id;
    send(sockfd,&msg,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg,sizeof(Msg),0);
    printf("退出成功！
");
    sleep(2);
}
//存钱界面
void save(int id)
{
    Msg msg = {5};
    msg.acc.account = id;
    printf("请输入要存的金额:");
    scanf("%f",&msg.acc.money);
    //清空缓冲区
    stdin->_IO_read_ptr = stdin->_IO_read_end;
    //将获得的基本信息发送到服务端
    send(sockfd,&msg,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg,sizeof(Msg),0);
    printf("存款成功！
");
    printf("*****您的id是%d,账号内的余额为%f*****
",msg.acc.account,msg.acc.money);
    sleep(2);
}
//取款界面
void take(int id)
{
    Msg msg = {6};
    msg.acc.account = id;
    printf("请输入要取的金额:");
    scanf("%f",&msg.acc.money);
    //清空缓冲区
    stdin->_IO_read_ptr = stdin->_IO_read_end;
    //将获得的基本信息发送到服务端
    send(sockfd,&msg,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg,sizeof(Msg),0);
    if(-1 == msg.acc.flag)
    {
        printf("取款失败！
");
        sleep(2);
    }
    else
    {
        printf("取款成功！
");
        printf("*****您的id是%d,账号内的余额为%f*****
",msg.acc.account,msg.acc.money);
        sleep(2);
    }
}
//转账界面
void trans(int id)
{
    Msg msg1 = {7};
    //Msg msg2 = {17};
    msg1.acc.account = id;
    printf("请输入转账对象的id:");
    scanf("%d",&msg1.acc.target);
    //清空缓冲区
    stdin->_IO_read_ptr = stdin->_IO_read_end;
    printf("请输入要转账的金额:");
    scanf("%f",&msg1.acc.money);
    //清空缓冲区
    stdin->_IO_read_ptr = stdin->_IO_read_end;
    //将获得的基本信息发送到服务端
    send(sockfd,&msg1,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg1,sizeof(Msg),0);
    switch(msg1.acc.flag)
    {
        case -1:printf("您的余额不足！
");sleep(2);break;
        case -2:printf("转账对象不存在！
");sleep(2);break;
        case 0:    printf("转账成功！
");
                printf("*****您的id是%d,账号内的余额为%f*****
",msg1.acc.account,msg1.acc.money);
                sleep(2);break;
    }
}

//查询界面
void show(int id)
{
    system("clear");
    Msg msg = {8};
    msg.acc.account = id;
    //将获得的基本信息发送到服务端
    send(sockfd,&msg,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg,sizeof(Msg),0);
    printf("%s用户您好！
",msg.acc.name);
    printf("账号ID:%d
",msg.acc.account);
    printf("身份证号:%s
",msg.acc.idcard);
    printf("账号余额:%f
",msg.acc.money);
    printf("*****按任意键返回*****
");
    getchar();    
}

//修改密码界面
void change(int id)
{
    Msg msg = {9};
    msg.acc.account = id;
    printf("请输入您的身份账号以用于验证用户信息:");
    gets(msg.acc.idcard);
    printf("请输入新密码(6位):");
    put_pw(msg.acc.password);
    if(6 != strlen(msg.acc.password))
    {
        printf("密码格式不符！
");
        sleep(2);
        return;
    }
    else
    {
        //将获得的基本信息发送到服务端
        send(sockfd,&msg,sizeof(Msg),0);
        //接收服务端传回的信息(账号id)
        recv(sockfd,&msg,sizeof(Msg),0);
        if(-1 == msg.acc.flag)
        {
            printf("修改失败！
");
            sleep(2);
        }
        else
        {
            printf("修改成功！
");
            sleep(2);
        }
    }
}

//副主界面
void ano_main(int id)
{
    while(1)
    {
        system("clear");
        func_menu();
        switch(getch())
        {
            case 49:save(id);break;
            case 50:take(id);break;
            case 51:trans(id);break;
            case 52:show(id);break;
            case 53:change(id);break;
            case 54:logout(id);return;
        }
    }
}
//加载界面
void loading(void)
{
    //system("clear");
    char buf[103] = {};
    memset(buf, ' ', sizeof(buf));
    buf[0] = '[';
    buf[101] = ']';
    buf[102] = '';
    int i = 0;
    char index[6] = "-\|/";
    while (i <= 99)
    {
        buf[i+1] = '>';
        printf("%s[%d%%][%c]
",buf,i,index[i%4]);
        fflush(stdout);//刷新缓冲区
        usleep(50000);
        i++;
    }
 
    printf("
");
}
//登录界面
void login_menu(void)
{
    Msg msg = {2};
    msg.acc.flag = -1;
    //获取账号密码
    printf("请输入账号:");
    scanf("%d",&msg.acc.account);
    //清空缓冲区
    stdin->_IO_read_ptr = stdin->_IO_read_end;
    printf("请输入密码:");
    put_pw(msg.acc.password);
    //将获得的基本信息发送到服务端
    send(sockfd,&msg,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg,sizeof(Msg),0);
    printf("%d
",msg.acc.flag);
    //根据判断结果跳转界面
    if(0 == msg.acc.flag)
    {
        if(0 == msg.acc.sp)
        {
            loading();
            ano_main(msg.acc.account);
        }
        else
        {
            printf("该账号已在其他客户端登录！
");
            sleep(2);
            return;
        }
    }
    else if(3 <= msg.acc.flag)
    {
        printf("账号已被锁定!请先前往解锁!
");
        sleep(2);
        return;
    }
    else
    {
        printf("登录失败!用户或密码错误!
");
        sleep(2);
        return;
    }    
}

//解锁界面
void unlock_menu(void)
{
    Msg msg = {3};
    printf("请输入要解锁的账号:");
    scanf("%d",&msg.acc.account);
    //清空缓冲区
    stdin->_IO_read_ptr = stdin->_IO_read_end;
    printf("请输入身份证号:");
    gets(msg.acc.idcard);
    printf("请输入新密码:");
    put_pw(msg.acc.password);
    //将获得的基本信息发送到服务端
    send(sockfd,&msg,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg,sizeof(Msg),0);
    if(-1 == msg.acc.flag)
    {    
        printf("输入信息不正确或该账户不需要解锁！
");
        sleep(2);
        return;
    }
    printf("解锁成功！
");
    printf("*****您的id是%d,新密码为%s*****
",msg.acc.account,msg.acc.password);
    sleep(2);
    return;    
}

//销户界面
void delete_menu(void)
{
    Msg msg = {4};
    //获取账号密码
    printf("请输入账号:");
    scanf("%d",&msg.acc.account);
    //清空缓冲区
    stdin->_IO_read_ptr = stdin->_IO_read_end;
    printf("请输入密码:");
    put_pw(msg.acc.password);
    //将获得的基本信息发送到服务端
    send(sockfd,&msg,sizeof(Msg),0);
    //接收服务端传回的信息(账号id)
    recv(sockfd,&msg,sizeof(Msg),0);
    if(-2 == msg.acc.flag)
    {
        printf("销户成功！
");
        sleep(2);
    }
    else
    {
        printf("销户失败！
");
        sleep(2);
    }
    return;
}
int main()
{
    // 创建socket对象
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(0 > sockfd)
    {
        perror("socket");
        return -1;
    }
    // 准备通信地址
    struct sockaddr_in addr = {};
    addr.sin_family = AF_INET;
    addr.sin_port = htons(1904);
    addr.sin_addr.s_addr = inet_addr("192.168.1.110");
    // 连接
    if(0 > connect(sockfd,(struct sockaddr*)&addr,sizeof(addr)))
    {
        perror("connect");
        return -1;
    }
    
    while(1)
    {
        system("clear");
        show_menu();
        switch(getch())
        {
            case 49:open_menu();break;
            case 50:login_menu();break;
            //case 51:unlock_menu();break;
            //case 52:delete_menu();break;
            case 53:return 0;
        }
    }
}

使用的工具函数

#include "customer.h"

//创建消息队列
int create_msg_queue(int keynum)
{
    //获取ipc键值
    key_t key = ftok("~",keynum);
    //创建消息队列
    int msgid = msgget(key,IPC_CREAT|0644);
    if(0 > msgid)
    {
        perror("msgget");
        return -1;
    }
    
    return msgid;
}

//隐藏密码输入
char* put_pw(char* pw)
{
    for(int i=0; i<20; i++)
    {
        stdin->_IO_read_ptr=stdin->_IO_read_end;
        pw[i]=getch();
        if(pw[i] == 127)
        {
            if(i>0)
            {
                i -= 2;
                printf(" ");
            }
            else
            {
                i--;
            }
        }
        else if(pw[i] == 10)
        {
            pw[i]='';
            break;
        }
        else
        {
            putchar('*');
        }
    }
    return pw;
}

定义的头文件

#ifndef CUSTOMER_H
#define CUSTOMER_H

#include <stdio.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/wait.h>
#include <getch.h>
#include <stdbool.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <semaphore.h>
#include <signal.h>

#define key1 123456
#define key2 654321
typedef struct Account
{
    int account;
    int target;
    char name[20];
    char password[10];
    char idcard[18];
    float money;
    int flag;
    int sp;
}Account;
typedef struct Msg
{
    long type;
    Account acc;
}Msg;

int create_msg_queue(int keynum);
char* put_pw(char* pw);

#endif

Makefile

 all: 2 gcc -std=gnu99 -c customer.c 3 gcc -std=gnu99 -c tools.c 4 gcc customer.o tools.o -o customer -lpthread