| 55 |
USER_ONLINE_LIST *p_online_current; |
USER_ONLINE_LIST *p_online_current; |
| 56 |
USER_ONLINE_LIST *p_online_new; |
USER_ONLINE_LIST *p_online_new; |
| 57 |
USER_STAT_MAP user_stat_map; |
USER_STAT_MAP user_stat_map; |
| 58 |
|
int user_login_count; |
| 59 |
}; |
}; |
| 60 |
typedef struct user_list_pool_t USER_LIST_POOL; |
typedef struct user_list_pool_t USER_LIST_POOL; |
| 61 |
|
|
| 95 |
|
|
| 96 |
static int user_list_load(MYSQL *db, USER_LIST *p_list); |
static int user_list_load(MYSQL *db, USER_LIST *p_list); |
| 97 |
static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list); |
static int user_online_list_load(MYSQL *db, USER_ONLINE_LIST *p_list); |
| 98 |
|
static int user_login_count_load(MYSQL *db); |
| 99 |
|
|
| 100 |
static int user_info_index_uid_comp(const void *ptr1, const void *ptr2) |
static int user_info_index_uid_comp(const void *ptr1, const void *ptr2) |
| 101 |
{ |
{ |
| 150 |
|
|
| 151 |
snprintf(sql, sizeof(sql), |
snprintf(sql, sizeof(sql), |
| 152 |
"SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, visit_count, " |
"SELECT user_list.UID AS UID, username, nickname, gender, gender_pub, life, exp, visit_count, " |
| 153 |
"UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(birthday), `introduction` " |
"UNIX_TIMESTAMP(signup_dt), UNIX_TIMESTAMP(last_login_dt), UNIX_TIMESTAMP(last_logout_dt), " |
| 154 |
|
"UNIX_TIMESTAMP(birthday), `introduction` " |
| 155 |
"FROM user_list INNER JOIN user_pubinfo ON user_list.UID = user_pubinfo.UID " |
"FROM user_list INNER JOIN user_pubinfo ON user_list.UID = user_pubinfo.UID " |
| 156 |
"INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID " |
"INNER JOIN user_reginfo ON user_list.UID = user_reginfo.UID " |
| 157 |
"WHERE enable ORDER BY username"); |
"WHERE enable ORDER BY username"); |
| 188 |
p_list->users[i].visit_count = (row[7] == NULL ? 0 : atoi(row[7])); |
p_list->users[i].visit_count = (row[7] == NULL ? 0 : atoi(row[7])); |
| 189 |
p_list->users[i].signup_dt = (row[8] == NULL ? 0 : atol(row[8])); |
p_list->users[i].signup_dt = (row[8] == NULL ? 0 : atol(row[8])); |
| 190 |
p_list->users[i].last_login_dt = (row[9] == NULL ? 0 : atol(row[9])); |
p_list->users[i].last_login_dt = (row[9] == NULL ? 0 : atol(row[9])); |
| 191 |
p_list->users[i].birthday = (row[10] == NULL ? 0 : atol(row[10])); |
p_list->users[i].last_logout_dt = (row[10] == NULL ? 0 : atol(row[10])); |
| 192 |
intro_len = strlen((row[11] == NULL ? "" : row[11])); |
p_list->users[i].birthday = (row[10] == NULL ? 0 : atol(row[11])); |
| 193 |
|
intro_len = strlen((row[12] == NULL ? "" : row[12])); |
| 194 |
if (intro_len >= sizeof(p_list->user_intro_buf) - 1 - intro_buf_offset) |
if (intro_len >= sizeof(p_list->user_intro_buf) - 1 - intro_buf_offset) |
| 195 |
{ |
{ |
| 196 |
log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i); |
log_error("OOM for user introduction: len=%d, i=%d\n", intro_len, i); |
| 197 |
break; |
break; |
| 198 |
} |
} |
| 199 |
memcpy(p_list->user_intro_buf + intro_buf_offset, |
memcpy(p_list->user_intro_buf + intro_buf_offset, |
| 200 |
(row[11] == NULL ? "" : row[11]), |
(row[12] == NULL ? "" : row[12]), |
| 201 |
intro_len + 1); |
intro_len + 1); |
| 202 |
p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset; |
p_list->users[i].intro = p_list->user_intro_buf + intro_buf_offset; |
| 203 |
intro_buf_offset += (intro_len + 1); |
intro_buf_offset += (intro_len + 1); |
| 248 |
int ret = 0; |
int ret = 0; |
| 249 |
int i; |
int i; |
| 250 |
int j; |
int j; |
| 251 |
|
int user_cnt; |
| 252 |
|
int guest_cnt; |
| 253 |
|
|
| 254 |
if (db == NULL || p_list == NULL) |
if (db == NULL || p_list == NULL) |
| 255 |
{ |
{ |
| 260 |
snprintf(sql, sizeof(sql), |
snprintf(sql, sizeof(sql), |
| 261 |
"SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), " |
"SELECT SID, UID, ip, current_action, UNIX_TIMESTAMP(login_tm), " |
| 262 |
"UNIX_TIMESTAMP(last_tm) FROM user_online " |
"UNIX_TIMESTAMP(last_tm) FROM user_online " |
| 263 |
"WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) AND UID <> 0 " |
"WHERE last_tm >= SUBDATE(NOW(), INTERVAL %d SECOND) " |
| 264 |
"ORDER BY last_tm DESC", |
"ORDER BY last_tm DESC", |
| 265 |
BBS_user_off_line); |
BBS_user_off_line); |
| 266 |
|
|
| 279 |
} |
} |
| 280 |
|
|
| 281 |
i = 0; |
i = 0; |
| 282 |
|
user_cnt = 0; |
| 283 |
|
guest_cnt = 0; |
| 284 |
while ((row = mysql_fetch_row(rs))) |
while ((row = mysql_fetch_row(rs))) |
| 285 |
{ |
{ |
| 286 |
|
if (atoi(row[1]) == 0) // guest |
| 287 |
|
{ |
| 288 |
|
guest_cnt++; |
| 289 |
|
continue; |
| 290 |
|
} |
| 291 |
|
else |
| 292 |
|
{ |
| 293 |
|
user_cnt++; |
| 294 |
|
} |
| 295 |
|
|
| 296 |
p_list->users[i].id = i; |
p_list->users[i].id = i; |
| 297 |
strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1); |
strncpy(p_list->users[i].session_id, row[0], sizeof(p_list->users[i].session_id) - 1); |
| 298 |
p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0'; |
p_list->users[i].session_id[sizeof(p_list->users[i].session_id) - 1] = '\0'; |
| 333 |
mysql_free_result(rs); |
mysql_free_result(rs); |
| 334 |
rs = NULL; |
rs = NULL; |
| 335 |
|
|
| 336 |
if (i > 0) |
if (user_cnt > 0) |
| 337 |
{ |
{ |
| 338 |
// Rebuild index |
// Rebuild index |
| 339 |
for (j = 0; j < i; j++) |
for (j = 0; j < user_cnt; j++) |
| 340 |
{ |
{ |
| 341 |
p_list->index_uid[j].uid = p_list->users[j].user_info.uid; |
p_list->index_uid[j].uid = p_list->users[j].user_info.uid; |
| 342 |
p_list->index_uid[j].id = j; |
p_list->index_uid[j].id = j; |
| 343 |
} |
} |
| 344 |
|
|
| 345 |
qsort(p_list->index_uid, (size_t)i, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp); |
qsort(p_list->index_uid, (size_t)user_cnt, sizeof(USER_INFO_INDEX_UID), user_info_index_uid_comp); |
| 346 |
|
|
| 347 |
#ifdef _DEBUG |
#ifdef _DEBUG |
| 348 |
log_error("Rebuild index of %d online users\n", i); |
log_error("Rebuild index of %d online users\n", user_cnt); |
| 349 |
#endif |
#endif |
| 350 |
} |
} |
| 351 |
|
|
| 352 |
p_list->user_count = i; |
p_list->user_count = user_cnt; |
| 353 |
|
p_list->guest_count = guest_cnt; |
| 354 |
|
|
| 355 |
#ifdef _DEBUG |
#ifdef _DEBUG |
| 356 |
log_error("Loaded %d online users\n", p_list->user_count); |
log_error("Loaded %d online users and %d guest users\n", p_list->user_count, p_list->guest_count); |
| 357 |
#endif |
#endif |
| 358 |
|
|
| 359 |
cleanup: |
cleanup: |
| 362 |
return ret; |
return ret; |
| 363 |
} |
} |
| 364 |
|
|
| 365 |
int user_list_pool_init(void) |
int user_login_count_load(MYSQL *db) |
| 366 |
|
{ |
| 367 |
|
MYSQL_RES *rs = NULL; |
| 368 |
|
MYSQL_ROW row; |
| 369 |
|
char sql[SQL_BUFFER_LEN]; |
| 370 |
|
|
| 371 |
|
if (db == NULL) |
| 372 |
|
{ |
| 373 |
|
log_error("NULL pointer error\n"); |
| 374 |
|
return -1; |
| 375 |
|
} |
| 376 |
|
|
| 377 |
|
snprintf(sql, sizeof(sql), |
| 378 |
|
"SELECT ID FROM user_login_log ORDER BY ID DESC LIMIT 1"); |
| 379 |
|
if (mysql_query(db, sql) != 0) |
| 380 |
|
{ |
| 381 |
|
log_error("Query user_login_log error: %s\n", mysql_error(db)); |
| 382 |
|
return -2; |
| 383 |
|
} |
| 384 |
|
if ((rs = mysql_store_result(db)) == NULL) |
| 385 |
|
{ |
| 386 |
|
log_error("Get user_login_log data failed\n"); |
| 387 |
|
return -2; |
| 388 |
|
} |
| 389 |
|
if ((row = mysql_fetch_row(rs))) |
| 390 |
|
{ |
| 391 |
|
p_user_list_pool->user_login_count = atoi(row[0]); |
| 392 |
|
} |
| 393 |
|
mysql_free_result(rs); |
| 394 |
|
|
| 395 |
|
return 0; |
| 396 |
|
} |
| 397 |
|
|
| 398 |
|
int user_list_pool_init(const char *filename) |
| 399 |
{ |
{ |
| 400 |
int shmid; |
int shmid; |
| 401 |
int semid; |
int semid; |
| 429 |
|
|
| 430 |
// Allocate shared memory |
// Allocate shared memory |
| 431 |
proj_id = (int)(time(NULL) % getpid()); |
proj_id = (int)(time(NULL) % getpid()); |
| 432 |
key = ftok(VAR_USER_LIST_SHM, proj_id); |
key = ftok(filename, proj_id); |
| 433 |
if (key == -1) |
if (key == -1) |
| 434 |
{ |
{ |
| 435 |
log_error("ftok(%s %d) error (%d)\n", VAR_USER_LIST_SHM, proj_id, errno); |
log_error("ftok(%s %d) error (%d)\n", filename, proj_id, errno); |
| 436 |
return -2; |
return -2; |
| 437 |
} |
} |
| 438 |
|
|
| 593 |
ret = -2; |
ret = -2; |
| 594 |
goto cleanup; |
goto cleanup; |
| 595 |
} |
} |
| 596 |
|
|
| 597 |
|
if (user_login_count_load(db) < 0) |
| 598 |
|
{ |
| 599 |
|
log_error("user_login_count_load() error\n"); |
| 600 |
|
ret = -2; |
| 601 |
|
goto cleanup; |
| 602 |
|
} |
| 603 |
} |
} |
| 604 |
else |
else |
| 605 |
{ |
{ |
| 913 |
return ret; |
return ret; |
| 914 |
} |
} |
| 915 |
|
|
| 916 |
|
int get_user_list_count(int *p_user_cnt) |
| 917 |
|
{ |
| 918 |
|
if (p_user_cnt == NULL) |
| 919 |
|
{ |
| 920 |
|
log_error("NULL pointer error\n"); |
| 921 |
|
return -1; |
| 922 |
|
} |
| 923 |
|
|
| 924 |
|
// acquire lock of user list |
| 925 |
|
if (user_list_rd_lock(p_user_list_pool->semid) < 0) |
| 926 |
|
{ |
| 927 |
|
log_error("user_list_rd_lock() error\n"); |
| 928 |
|
return -2; |
| 929 |
|
} |
| 930 |
|
|
| 931 |
|
*p_user_cnt = p_user_list_pool->p_current->user_count; |
| 932 |
|
|
| 933 |
|
// release lock of user list |
| 934 |
|
if (user_list_rd_unlock(p_user_list_pool->semid) < 0) |
| 935 |
|
{ |
| 936 |
|
log_error("user_list_rd_unlock() error\n"); |
| 937 |
|
return -2; |
| 938 |
|
} |
| 939 |
|
|
| 940 |
|
return 0; |
| 941 |
|
} |
| 942 |
|
|
| 943 |
|
int get_user_online_list_count(int *p_user_cnt, int *p_guest_cnt) |
| 944 |
|
{ |
| 945 |
|
if (p_user_cnt == NULL || p_guest_cnt == NULL) |
| 946 |
|
{ |
| 947 |
|
log_error("NULL pointer error\n"); |
| 948 |
|
return -1; |
| 949 |
|
} |
| 950 |
|
|
| 951 |
|
// acquire lock of user list |
| 952 |
|
if (user_list_rd_lock(p_user_list_pool->semid) < 0) |
| 953 |
|
{ |
| 954 |
|
log_error("user_list_rd_lock() error\n"); |
| 955 |
|
return -2; |
| 956 |
|
} |
| 957 |
|
|
| 958 |
|
*p_user_cnt = p_user_list_pool->p_online_current->user_count; |
| 959 |
|
*p_guest_cnt = p_user_list_pool->p_online_current->guest_count; |
| 960 |
|
|
| 961 |
|
// release lock of user list |
| 962 |
|
if (user_list_rd_unlock(p_user_list_pool->semid) < 0) |
| 963 |
|
{ |
| 964 |
|
log_error("user_list_rd_unlock() error\n"); |
| 965 |
|
return -2; |
| 966 |
|
} |
| 967 |
|
|
| 968 |
|
return 0; |
| 969 |
|
} |
| 970 |
|
|
| 971 |
|
int get_user_login_count(int *p_login_cnt) |
| 972 |
|
{ |
| 973 |
|
if (p_login_cnt == NULL) |
| 974 |
|
{ |
| 975 |
|
log_error("NULL pointer error\n"); |
| 976 |
|
return -1; |
| 977 |
|
} |
| 978 |
|
|
| 979 |
|
*p_login_cnt = p_user_list_pool->user_login_count; |
| 980 |
|
|
| 981 |
|
return 0; |
| 982 |
|
} |
| 983 |
|
|
| 984 |
int query_user_info(int32_t id, USER_INFO *p_user) |
int query_user_info(int32_t id, USER_INFO *p_user) |
| 985 |
{ |
{ |
| 986 |
int ret = 0; |
int ret = 0; |
| 1043 |
mid = (left + right) / 2; |
mid = (left + right) / 2; |
| 1044 |
if (uid < p_user_list_pool->p_current->index_uid[mid].uid) |
if (uid < p_user_list_pool->p_current->index_uid[mid].uid) |
| 1045 |
{ |
{ |
| 1046 |
right = mid; |
right = mid - 1; |
| 1047 |
} |
} |
| 1048 |
else if (uid > p_user_list_pool->p_current->index_uid[mid].uid) |
else if (uid > p_user_list_pool->p_current->index_uid[mid].uid) |
| 1049 |
{ |
{ |
| 1140 |
mid = (left + right) / 2; |
mid = (left + right) / 2; |
| 1141 |
if (uid < p_user_list_pool->p_online_current->index_uid[mid].uid) |
if (uid < p_user_list_pool->p_online_current->index_uid[mid].uid) |
| 1142 |
{ |
{ |
| 1143 |
right = mid; |
right = mid - 1; |
| 1144 |
} |
} |
| 1145 |
else if (uid > p_user_list_pool->p_online_current->index_uid[mid].uid) |
else if (uid > p_user_list_pool->p_online_current->index_uid[mid].uid) |
| 1146 |
{ |
{ |
| 1226 |
mid = (left + right) / 2; |
mid = (left + right) / 2; |
| 1227 |
if (start_uid < p_user_list_pool->p_current->index_uid[mid].uid) |
if (start_uid < p_user_list_pool->p_current->index_uid[mid].uid) |
| 1228 |
{ |
{ |
| 1229 |
right = mid; |
right = mid - 1; |
| 1230 |
} |
} |
| 1231 |
else if (start_uid > p_user_list_pool->p_current->index_uid[mid].uid) |
else if (start_uid > p_user_list_pool->p_current->index_uid[mid].uid) |
| 1232 |
{ |
{ |