/[LeafOK_CVS]/lbbs/src/login.c
ViewVC logotype

Diff of /lbbs/src/login.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1.2 by sysadm, Thu Oct 21 17:28:46 2004 UTC Revision 1.15 by sysadm, Sun May 4 14:54:55 2025 UTC
# Line 1  Line 1 
1  /***************************************************************************  /***************************************************************************
2                            main.c  -  description                                                    login.c  -  description
3                               -------------------                                                           -------------------
4      begin                : Mon Oct 11 2004          begin                : Mon Oct 20 2004
5      copyright            : (C) 2004 by Leaflet          copyright            : (C) 2004 by Leaflet
6      email                : leaflet@leafok.com          email                : leaflet@leafok.com
7   ***************************************************************************/   ***************************************************************************/
8    
9  /***************************************************************************  /***************************************************************************
# Line 15  Line 15 
15   *                                                                         *   *                                                                         *
16   ***************************************************************************/   ***************************************************************************/
17    
18    #include "login.h"
19    #include "register.h"
20  #include "bbs.h"  #include "bbs.h"
21    #include "user_priv.h"
22    #include "reg_ex.h"
23  #include "common.h"  #include "common.h"
24    #include "log.h"
25    #include "io.h"
26    #include "screen.h"
27    #include "database.h"
28    #include <string.h>
29    #include <mysql.h>
30    #include <regex.h>
31    #include <unistd.h>
32    
33  void  void login_fail()
 login_fail ()  
34  {  {
35    char temp[256];          display_file(DATA_LOGIN_ERROR);
36    
37    strcpy (temp, app_home_dir);          sleep(1);
   strcat (temp, "data/login_error.dat");  
   display_file (temp);  
38  }  }
39    
40  int  int bbs_login()
 bbs_login ()  
41  {  {
42    char username[20], password[20];          char username[BBS_username_max_len + 1];
43    int count, ok;          char password[BBS_password_max_len + 1];
44            int count, ok;
45    
46            // Input username
47            count = 0;
48            ok = 0;
49            while (!ok)
50            {
51                    prints("\033[1;33m请输入帐号\033[m(试用请输入 `\033[1;36mguest\033[m', "
52                               "注册请输入`\033[1;31mnew\033[m'): ");
53                    iflush();
54    
55                    str_input(username, BBS_username_max_len, DOECHO);
56                    count++;
57    
58                    if (strcmp(username, "guest") == 0)
59                    {
60                            MYSQL *db = db_open();
61                            if (db == NULL)
62                            {
63                                    return -1;
64                            }
65    
66                            load_guest_info(db);
67    
68                            mysql_close(db);
69    
70                            return 0;
71                    }
72    
73                    if (strcmp(username, "new") == 0)
74                    {
75                            if (user_register() == 0)
76                                    return 0;
77                            else
78                                    return -2;
79                    }
80    
81                    if (strlen(username) > 0)
82                    {
83                            // Input password
84                            prints("\033[1;37m请输入密码\033[m: ");
85                            iflush();
86    
87                            str_input(password, BBS_password_max_len, NOECHO);
88    
89                            MYSQL *db = db_open();
90                            if (db == NULL)
91                            {
92                                    return -1;
93                            }
94    
95                            ok = (check_user(db, username, password) == 0);
96    
97                            mysql_close(db);
98                    }
99                    if (count >= 3 && !ok)
100                    {
101                            login_fail();
102                            return -1;
103                    }
104            }
105    
106            return 0;
107    }
108    
109    int check_user(MYSQL *db, char *username, char *password)
110    {
111            MYSQL_RES *rs;
112            MYSQL_ROW row;
113            char sql[SQL_BUFFER_LEN];
114            int ret;
115            long BBS_uid = 0;
116    
117            // Verify format
118            if (ireg("^[A-Za-z][A-Za-z0-9]{2,11}$", username, 0, NULL) != 0 ||
119                    ireg("^[A-Za-z0-9]{5,12}$", password, 0, NULL) != 0)
120            {
121                    prints("\033[1;31m用户名或密码格式错误...\033[m\r\n");
122                    iflush();
123                    return 1;
124            }
125    
126            // Begin transaction
127            if (mysql_query(db, "SET autocommit=0") != 0)
128            {
129                    log_error("SET autocommit=0 failed\n");
130                    return -1;
131            }
132    
133            if (mysql_query(db, "BEGIN") != 0)
134            {
135                    log_error("Begin transaction failed\n");
136                    return -1;
137            }
138    
139            sprintf(sql,
140                            "SELECT UID, username, p_login FROM user_list "
141                            "WHERE username = '%s' AND password = SHA2('%s', 256) AND enable",
142                            username, password);
143            if (mysql_query(db, sql) != 0)
144            {
145                    log_error("Query user_list failed\n");
146                    return -1;
147            }
148            if ((rs = mysql_store_result(db)) == NULL)
149            {
150                    log_error("Get user_list data failed\n");
151                    return -1;
152            }
153            if ((row = mysql_fetch_row(rs)))
154            {
155                    BBS_uid = atol(row[0]);
156                    strcpy(BBS_username, row[1]);
157                    int p_login = atoi(row[2]);
158    
159                    mysql_free_result(rs);
160    
161                    // Add user login log
162                    sprintf(sql,
163                                    "INSERT INTO user_login_log(UID, login_dt, login_ip) "
164                                    "VALUES(%ld, NOW(), '%s')",
165                                    BBS_uid, hostaddr_client);
166                    if (mysql_query(db, sql) != 0)
167                    {
168                            log_error("Insert into user_login_log failed\n");
169                            return -1;
170                    }
171    
172                    // Commit transaction
173                    if (mysql_query(db, "COMMIT") != 0)
174                    {
175                            log_error("Commit transaction failed\n");
176                            return -1;
177                    }
178    
179                    if (p_login == 0)
180                    {
181                            mysql_free_result(rs);
182    
183                            prints("\033[1;31m您目前无权登陆...\033[m\r\n");
184                            iflush();
185                            return 1;
186                    }
187            }
188            else
189            {
190                    mysql_free_result(rs);
191    
192                    sprintf(sql,
193                                    "INSERT INTO user_err_login_log(username, password, login_dt, login_ip) "
194                                    "VALUES('%s', '%s', NOW(), '%s')",
195                                    username, password, hostaddr_client);
196                    if (mysql_query(db, sql) != 0)
197                    {
198                            log_error("Insert into user_err_login_log failed\n");
199                            return -1;
200                    }
201    
202                    // Commit transaction
203                    if (mysql_query(db, "COMMIT") != 0)
204                    {
205                            log_error("Commit transaction failed\n");
206                            return -1;
207                    }
208    
209                    prints("\033[1;31m错误的用户名或密码...\033[m\r\n");
210                    iflush();
211                    return 1;
212            }
213    
214            // Set AUTOCOMMIT = 1
215            if (mysql_query(db, "SET autocommit=1") != 0)
216            {
217                    log_error("SET autocommit=1 failed\n");
218                    return -1;
219            }
220    
221            ret = load_user_info(db, BBS_uid);
222    
223            switch (ret)
224            {
225            case 0: // Login successfully
226                    break;
227            case -1: // Load data error
228                    prints("\033[1;31m读取用户数据错误...\033[m\r\n");
229                    iflush();
230                    return -1;
231            case -2: // Unused
232                    prints("\033[1;31m请通过Web登录更新用户许可协议...\033[m\r\n");
233                    iflush();
234                    return 1;
235            case -3: // Dead
236                    prints("\033[1;31m很遗憾,您已经永远离开了我们的世界!\033[m\r\n");
237                    iflush();
238                    return 1;
239            default:
240                    return -2;
241            }
242    
243            sprintf(sql,
244                            "UPDATE user_pubinfo SET visit_count = visit_count + 1, "
245                            "last_login_dt = NOW() WHERE UID = %ld",
246                            BBS_uid);
247            if (mysql_query(db, sql) != 0)
248            {
249                    log_error("Update user_pubinfo failed\n");
250                    return -1;
251            }
252    
253            if (user_online_add(db) != 0)
254            {
255                    return -1;
256            }
257    
258    //Input username          BBS_last_access_tm = BBS_login_tm = time(0);
   count = 0;  
   ok = 0;  
   while (!ok)  
     {  
       printf  
         ("\033[1;33m请输入帐号\033[m(试用请输入 `\033[1;36mguest\033[m', "  
          "注册请输入`\033[1;31mnew\033[m'): ");  
       fflush (stdout);  
259    
260        str_input (username, 19, 0);          return 0;
261        count++;  }
262    
263    int load_user_info(MYSQL *db, long int BBS_uid)
264    {
265            MYSQL_RES *rs;
266            MYSQL_ROW row;
267            char sql[SQL_BUFFER_LEN];
268            int life;
269            time_t last_login_dt;
270    
271            sprintf(sql,
272                            "SELECT life, UNIX_TIMESTAMP(last_login_dt) "
273                            "FROM user_pubinfo WHERE UID = %ld",
274                            BBS_uid);
275            if (mysql_query(db, sql) != 0)
276            {
277                    log_error("Query user_pubinfo failed\n");
278                    return -1;
279            }
280            if ((rs = mysql_store_result(db)) == NULL)
281            {
282                    log_error("Get user_pubinfo data failed\n");
283                    return -1;
284            }
285            if ((row = mysql_fetch_row(rs)))
286            {
287                    life = atoi(row[0]);
288                    last_login_dt = (time_t)atol(row[1]);
289            }
290            else
291            {
292                    mysql_free_result(rs);
293                    return -1; // Data not found
294            }
295            mysql_free_result(rs);
296    
297        if (strcmp (username, "guest") == 0)          if (life != 333 && life != 365 && life != 666 && life != 999 && // Not immortal
298          return 1;                  time(0) - last_login_dt > 60 * 60 * 24 * life)
299            {
300                    return -3; // Dead
301            }
302    
303        if (strlen (username) > 0)          if (load_priv(db, &BBS_priv, BBS_uid) != 0)
304          {          {
305            //Input password                  return -1;
306            printf ("\033[1;37m请输入密码\033[m: ");          }
           fflush (stdout);  
307    
308            str_input (password, 19, 0);          return 0;
309    }
310    
311            if (strlen (password) > 0)  int load_guest_info(MYSQL *db)
312              {  {
313                ok = check_user (username, password);          strcpy(BBS_username, "guest");
             }  
314    
315            if (!ok)          if (load_priv(db, &BBS_priv, 0) != 0)
316              {          {
317                printf ("\033[1;31m错误的用户名或密码...\r\n");                  return -1;
               fflush (stdout);  
             }  
318          }          }
319        if (count >= 3)  
320            if (user_online_add(db) != 0)
321          {          {
322            login_fail ();                  return -1;
           return -1;  
323          }          }
     }  
324    
325    return 0;          BBS_last_access_tm = BBS_login_tm = time(0);
326    
327            return 0;
328  }  }
329    
330  int  int user_online_add(MYSQL *db)
 check_user(char *username, char *password)  
331  {  {
332    return 1;          char sql[SQL_BUFFER_LEN];
333    
334            if (user_online_del(db) != 0)
335            {
336                    return -1;
337            }
338    
339            sprintf(sql,
340                            "INSERT INTO user_online(SID, UID, ip, login_tm, last_tm) "
341                            "VALUES('Telnet_Process_%d', %ld, '%s', NOW(), NOW())",
342                            getpid(), BBS_priv.uid, hostaddr_client);
343            if (mysql_query(db, sql) != 0)
344            {
345                    log_error("Add user_online failed\n");
346                    return -1;
347            }
348    
349            return 0;
350    }
351    
352    int user_online_del(MYSQL *db)
353    {
354            char sql[SQL_BUFFER_LEN];
355    
356            sprintf(sql,
357                            "DELETE FROM user_online WHERE SID = 'Telnet_Process_%d'",
358                            getpid());
359            if (mysql_query(db, sql) != 0)
360            {
361                    log_error("Delete user_online failed\n");
362                    return -1;
363            }
364    
365            return 0;
366  }  }


Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters

webmaster@leafok.com
ViewVC Help
Powered by ViewVC 1.3.0-beta1