其他分享
首页 > 其他分享> > 网络版ATM项目的实现——服务端

网络版ATM项目的实现——服务端

作者:互联网

修改

 1 #include "tools.h"
 2 
 3 int main()
 4 {
 5     int queue_s2c = create_msg_queue(key1);
 6     int queue_c2s = create_msg_queue(key2);
 7     while(1)
 8     {
 9         Msg msg1 = {};
10         Msg msg2 = {};
11         printf("等待修改 %d\n",queue_c2s);
12         //接收客户端登录输入的信息
13         msgrcv(queue_c2s,&msg1,sizeof(Msg),9,0);
14         printf("激活修改\n");
15         char str[20] = {};
16         //查询文件是否存在
17         int fd = open(get_file_name(msg1.acc.account,str),O_EXCL|O_RDWR,00644);
18         printf("%d\n",fd);
19         read(fd,&msg2,sizeof(Msg));
20         if(0 != strcmp(msg1.acc.idcard,msg2.acc.idcard))
21         {
22             msg1.acc.flag = -1;
23             msg1.type = 100;
24             msgsnd(queue_s2c,&msg1,sizeof(Msg)-4,0);
25         }
26         else
27         {
28             strcpy(msg2.acc.password,msg1.acc.password);
29             lseek(fd,0,SEEK_SET);
30             write(fd,&msg2,sizeof(Msg));
31             msg2.type = 100;
32             msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
33         }
34         close(fd);
35     }
36 }

删除

 1 #include "tools.h"
 2 
 3 int main()
 4 {
 5     int queue_s2c = create_msg_queue(key1);
 6     int queue_c2s = create_msg_queue(key2);
 7     while(1)
 8     {
 9         Msg msg1 = {};
10         Msg msg2 = {};
11         printf("等待销户 %d\n",queue_c2s);
12         //接收客户端登录输入的信息
13         msgrcv(queue_c2s,&msg1,sizeof(Msg),4,0);
14         printf("激活销户\n");
15         char str[20] = {};
16         //查询文件是否存在
17         int fd = open(get_file_name(msg1.acc.account,str),O_EXCL|O_RDWR,00644);
18         printf("%d\n",fd);
19         //不存在则发送"登录失败"信息
20         if(0 > fd)
21         {
22             msg1.acc.flag = -1;
23             msg1.type = 100;
24             msgsnd(queue_s2c,&msg1,sizeof(Msg)-4,0);
25         }
26         //存在则读取文件中结构体,判断密码是否正确
27         else
28         {
29             read(fd,&msg2,sizeof(Msg));
30             close(fd);
31             //密码正确
32             if(0 == strcmp(msg1.acc.password,msg2.acc.password))
33             {
34                 unlink(get_file_name(msg1.acc.account,str));
35                 msg2.type = 100;
36                 msg2.acc.flag = -2;
37                 msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
38             }
39             //密码错误
40             else
41             {
42                 msg2.type = 100;
43                 msg2.acc.flag = -1;
44                 msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
45                 printf("错误\n");
46             }
47         }
48         
49     }
50     
51 }

登陆

 1 #include "tools.h"
 2 
 3 int main()
 4 {
 5     int queue_s2c = create_msg_queue(key1);
 6     int queue_c2s = create_msg_queue(key2);
 7     while(1)
 8     {
 9         Msg msg1 = {};
10         Msg msg2 = {};
11         printf("等待登录 %d\n",queue_c2s);
12         //接收客户端登录输入的信息
13         msgrcv(queue_c2s,&msg1,sizeof(Msg),2,0);
14         printf("激活登录\n");
15         char str[20] = {};
16         //查询文件是否存在
17         int fd = open(get_file_name(msg1.acc.account,str),O_EXCL|O_RDWR,00644);
18         printf("%d\n",fd);
19         //不存在则发送"登录失败"信息
20         if(0 > fd)
21         {
22             msg1.acc.flag = -1;
23             msg1.type = 100;
24             msgsnd(queue_s2c,&msg1,sizeof(Msg)-4,0);
25         }
26         //存在则读取文件中结构体,判断密码是否正确
27         else
28         {
29             read(fd,&msg2,sizeof(Msg));
30             //printf("%d---%s---%s---%f---\n",msg2.acc.account,msg2.acc.name,msg2.acc.password,msg2.acc.money);
31             //密码正确
32             if(3 > msg2.acc.flag && 0 == strcmp(msg1.acc.password,msg2.acc.password))
33             {
34                 
35                 if(0 == msg2.acc.sp)
36                 {
37                     msg2.acc.flag = 0;
38                     msg2.type = 100;
39                     msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
40                     msg2.acc.sp=1;
41                     lseek(fd,0,SEEK_SET);
42                     write(fd,&msg2,sizeof(Msg));
43                     printf("正确\n");
44                 }
45                 else
46                 {
47                     msg2.acc.sp=1;
48                     msg2.acc.flag = 0;
49                     msg2.type = 100;
50                     msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
51                     printf("已在其他客户端登录\n");
52                 }
53             }
54             //密码错误
55             else
56             {
57                 if(0 == msg2.acc.sp)
58                 {
59                     msg2.acc.flag++;
60                     msg2.type = 100;
61                     msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
62                     lseek(fd,0,SEEK_SET);
63                     write(fd,&msg2,sizeof(Msg));
64                     printf("flag = %d\n",msg2.acc.flag);
65                     printf("错误\n");
66                 }
67                 else
68                 {
69                     msg2.acc.flag=0;
70                     msg2.type = 100;
71                     msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
72                     printf("错误lllll\n");
73                     
74                 }
75             }
76         }
77         close(fd);
78     }
79 }

登出

 1 #include "tools.h"
 2 
 3 int main()
 4 {
 5     int queue_s2c = create_msg_queue(key1);
 6     int queue_c2s = create_msg_queue(key2);
 7     while(1)
 8     {
 9         Msg msg1 = {};
10         Msg msg2 = {};
11         printf("等待退出 %d\n",queue_c2s);
12         //接收客户端登录输入的信息
13         msgrcv(queue_c2s,&msg1,sizeof(Msg),50,0);
14         printf("激活退出\n");
15         printf("id=%d\n",msg1.acc.account);
16         char str[20] = {};
17         //打开文件
18         int fd = open(get_file_name(msg1.acc.account,str),O_EXCL|O_RDWR,00644);
19         read(fd,&msg2,sizeof(Msg));
20         msg2.acc.sp = 0;
21         lseek(fd,0,SEEK_SET);
22         write(fd,&msg2,sizeof(Msg));
23         printf("已退出 sp=%d\n",msg2.acc.sp);
24         close(fd);
25         msg2.type = 100;
26         msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
27     }
28 }

服务

  1 #include "service.h"
  2 #include "tools.h"
  3 
  4 static int queue_s2c;
  5 static int queue_c2s;
  6 pid_t pid_usr[11] = {};
  7 
  8 typedef struct sockaddr SockAddr;
  9 int fd[3] = {};
 10 sem_t sem;
 11 char ip[3][20] = {};
 12 struct sockaddr_in from_addr;
 13 
 14 void sigint(int signum)
 15 {
 16     if(0 > msgctl(queue_s2c,IPC_RMID,NULL))
 17     {
 18         perror("msgctl1");
 19         exit(0);
 20     }
 21     if(0 > msgctl(queue_c2s,IPC_RMID,NULL))
 22     {
 23         perror("msgctl2");
 24         exit(0);
 25     }
 26     for(int i=1; i<11; i++)
 27     {
 28         kill(pid_usr[i],SIGKILL);
 29     }
 30     exit(0);
 31 }
 32 
 33 void* start(void* arg)
 34 {
 35     int i = *(int*)arg;//fdp
 36     if(0 > i)
 37     {
 38         perror("accetp");
 39         return NULL;
 40     }
 41     printf("客户端:%s连接成功!\n",inet_ntoa(from_addr.sin_addr));
 42     sem_wait(&sem);
 43     int temp=0;
 44     for(int j=0; j<3; j++)
 45     {
 46         if(fd[j] == i) temp=j;
 47     }
 48     strcpy(ip[temp],inet_ntoa(from_addr.sin_addr));
 49     printf("ip[temp][] = %s\n",ip[temp]);
 50     // 收发数据
 51     while(true)
 52     {
 53         char buf[1024] = {};
 54         Msg msg = {};
 55         int ret = recv(i,&msg,sizeof(msg),0);
 56         if(0 >= ret)
 57         {
 58             printf("%s客户端退出\n",ip[temp]);
 59             fd[temp]=0;
 60             sem_post(&sem);
 61             break;
 62         }
 63         msgsnd(queue_c2s,&msg,sizeof(Msg)-4,0);
 64         msgrcv(queue_s2c,&msg,sizeof(Msg),100,0);
 65         send(i,&msg,sizeof(Msg),0);
 66     }
 67 }
 68 
 69 int main()
 70 {
 71     signal(SIGINT,sigint);
 72     
 73     queue_s2c = create_msg_queue(key1);
 74     queue_c2s = create_msg_queue(key2);
 75     
 76     
 77     Msg msg = {};
 78 
 79     pid_usr[1] = vfork();
 80     if(0 == pid_usr[1])
 81         execl("open","open",NULL);
 82         
 83     pid_usr[2] = vfork();
 84     if(0 == pid_usr[2])
 85         execl("log_in","log_in",NULL);
 86         
 87     pid_usr[3] = vfork();
 88     if(0 == pid_usr[3])
 89         execl("unlock","unlock",NULL);
 90     
 91     pid_usr[4] = vfork();
 92     if(0 == pid_usr[4])
 93         execl("delete","delete",NULL);
 94         
 95     pid_usr[5] = vfork();
 96     if(0 == pid_usr[5])
 97         execl("save","save",NULL);
 98     
 99     pid_usr[6] = vfork();
100     if(0 == pid_usr[6])
101         execl("take","take",NULL);
102         
103     pid_usr[7] = vfork();
104     if(0 == pid_usr[7])
105         execl("trans","trans",NULL);
106         
107     pid_usr[8] = vfork();
108     if(0 == pid_usr[8])
109         execl("show","show",NULL);
110         
111     pid_usr[9] = vfork();
112     if(0 == pid_usr[9])
113         execl("change","change",NULL);
114     
115     pid_usr[10] = vfork();
116     if(0 == pid_usr[10])
117         execl("log_out","log_out",NULL);    
118     
119         
120     //s_c_socket 绑定
121     
122     //signal(SIGINT,sigint);
123     sem_init(&sem,0,3);
124     // 创建socket对象
125     int sockfd = socket(AF_INET,SOCK_STREAM,0);
126     if(0 > sockfd)
127     {
128         perror("socket");
129         return -1;
130     }
131     // 准备通信地址
132     struct sockaddr_in addr = {};
133     addr.sin_family = AF_INET;
134     addr.sin_port = htons(1904);
135     addr.sin_addr.s_addr = inet_addr("192.168.1.110");
136 
137     // 绑定socket对象与通信地址
138     socklen_t len = sizeof(addr);
139     if(0 > bind(sockfd,(SockAddr*)&addr,len))
140     {
141         perror("bind");
142         return -1;
143     }
144     // 设置监听socket对象
145     listen(sockfd,10);
146     int fdp = 0;
147     //int i = 0;
148     // 等待连接
149     while(true)
150     {
151         if(fdp == 0) fdp = accept(sockfd,(SockAddr*)&from_addr,&len);
152         printf("%d\n",fdp);
153         if(0 > fdp) perror("fdp");
154         int i;
155         for(i=0; i<3; i++)
156         {
157             printf("wojinruxunhuanle\n");
158             if(0 == fd[i])
159             {
160                 fd[i] = fdp;
161                 pthread_t pid = 0;
162                 //创建子线程
163                 pthread_create(&pid,NULL,start,&fd[i]);
164                 char _buf[1024] = "欢迎进入聊天室";
165                 //send(fdp,_buf,strlen(_buf)+1,0);
166                 fdp=0;
167                 break;
168             }
169         }
170         printf("%d\n",i);
171         if(3 <= i)
172         {
173             char _buf[1024] = "聊天室满,请稍等...";
174             send(fdp,_buf,strlen(_buf)+1,0);
175             sem_wait(&sem);
176             sem_post(&sem);
177             printf("我等到了\n");
178         }
179     }
180     
181         
182     
183     
184 }

取钱

#include "tools.h"

int main()
{
    int queue_s2c = create_msg_queue(key1);
    int queue_c2s = create_msg_queue(key2);
    while(1)
    {
        //存储收到的结构体信息
        Msg msg1 = {};
        //存储文件中读取的信息
        Msg msg2 = {};
        printf("等待取款 %d\n",queue_c2s);
        //接收客户端开户输入的信息
        msgrcv(queue_c2s,&msg1,sizeof(Msg),6,0);
        printf("激活取款\n");
        char str[20] = {};
        
        int fd = open(get_file_name(msg1.acc.account,str),O_EXCL|O_RDWR,00644);
        read(fd,&msg2,sizeof(Msg));
        if(msg2.acc.money < msg1.acc.money)
        {
            msg1.acc.flag = -1;
            msg1.type = 100;
            msgsnd(queue_s2c,&msg1,sizeof(Msg)-4,0);
        }
        else
        {
            msg2.acc.money = msg2.acc.money-msg1.acc.money;
            lseek(fd,0,SEEK_SET);
            write(fd,&msg2,sizeof(Msg));
            perror("write");
            close(fd);
            //将账户结构体信息发送至客户端
            msg2.type = 100;
            msgsnd(queue_s2c,&msg2,sizeof(Msg)-4,0);
        }
    }
}

转账

 1 #include "tools.h"
 2 
 3 int main()
 4 {
 5     int queue_s2c = create_msg_queue(key1);
 6     int queue_c2s = create_msg_queue(key2);
 7     while(1)
 8     {
 9         //存储收到的结构体1信息
10         Msg msg1 = {};
11         Msg msg11 = {};
12         //存储收到的结构体2信息
13         Msg msg2 = {};
14         Msg msg22 = {};
15         printf("等待转账 %d\n",queue_c2s);
16         //接收客户端开户输入的信息
17         msgrcv(queue_c2s,&msg1,sizeof(Msg),7,0);
18         //msgrcv(queue_c2s,&msg2,sizeof(Msg),17,0);
19         printf("激活转账\n");
20         char str1[20] = {};
21         int fd1 = open(get_file_name(msg1.acc.account,str1),O_EXCL|O_RDWR,00644);
22         read(fd1,&msg11,sizeof(Msg));
23         printf("fd1:%d\n",fd1);
24         if(msg11.acc.money < msg1.acc.money)
25         {
26             msg1.acc.flag = -1;//钱不够
27             msg1.type = 100;
28             close(fd1);
29             msgsnd(queue_s2c,&msg1,sizeof(Msg)-4,0);
30         }
31         else
32         {
33             char str2[20] = {};
34             int fd2 = open(get_file_name(msg1.acc.target,str2),O_EXCL|O_RDWR,00644);
35             printf("fd2:%d\n",fd2);
36             if(0 > fd2)
37             {
38                 msg1.acc.flag = -2;//转账对象不存在
39                 msg1.type = 100;
40                 msgsnd(queue_s2c,&msg1,sizeof(Msg)-4,0);
41             }
42             else
43             {
44                 read(fd2,&msg22,sizeof(Msg));
45                 msg22.acc.money = msg22.acc.money+msg1.acc.money;
46                 lseek(fd2,0,SEEK_SET);
47                 write(fd2,&msg22,sizeof(Msg));
48                 close(fd2);
49                 msg11.acc.money = msg11.acc.money-msg1.acc.money;
50                 lseek(fd1,0,SEEK_SET);
51                 write(fd1,&msg11,sizeof(Msg));
52                 close(fd1);
53                 msg11.acc.flag = 0;//操作成功
54                 msg11.type = 100;
55                 printf("right\n");
56                 msgsnd(queue_s2c,&msg11,sizeof(Msg)-4,0);
57                 printf("%ld===\n",msg11.type);
58             }
59         }
60     }
61 }

 

标签:acc,msg1,网络版,msg2,ATM,queue,Msg,sizeof,服务端
来源: https://www.cnblogs.com/kid971220/p/10497818.html