void test(){
User* user1 = (User*)malloc(sizeof(User));
User* user2 = (User*)malloc(sizeof(User));
User* user3 = (User*)malloc(sizeof(User));
List* list = (List*)malloc(sizeof(List));
user_init(user1);
user_init(user2);
user_init(user3);
if(user1->friends==NULL)
printf("NULL!!!\n");
printf("111\n");
strcpy(user2->ID, "123");
strcpy(user3->ID, "456");
strcpy(user1->ID, "000");
//user2->ID[0] = 2;
//user2->ID[1] = 2;
//user3->ID[0] = 3;
printf("222\n");
//user1->friends->user = user2;
//printf("333\n");
//user1->friends->next= NULL;
//printf("444\n");
list_init(list);
list->user=user2;
list->next=NULL;
printf("333\n");
user1->friends=list;
//user1->friends->next=NULL;
printf("user1's friend : %s\n", user1->friends->user->ID);
list->user=user3;
list->next=NULL;
printf("444\n");
user1->friends->next = list;
printf("555\n");
//user1->friends->next->next=NULL;
printf("666\n");
printf("friends : %s\n", user1->friends->user);
add_frinedship(user2, user1);
printf("user2's friends : %s\n", user2->friends->user->ID);
}
int
main(int argc, char *argv[])
{
if (argc != 2) {
fprintf(stderr,"usage: cutewiki <wikiname> \n");
exit(1);
}
/* Try to set different non-UTF-8 locale */
if (!setlocale(LC_ALL, "")) {
fprintf(stderr, "Error: Setting locale failed!\n");
exit(1);
}
//svr_init();
wiki_init(argv[1]);
user_init();
rcs_init();
pagelist_init(wiki->pagedir);
fprintf(stderr, "Info: CuteWiki started with configuration '%s'.\n", wiki->wikiname);
wiki_loop();
pagelist_exit();
user_exit();
wiki_exit();
svr_exit();
return 0;
}
int main (void) {
cpu_wakeup_init();
//usb_dp_pullup_en (1); //mouse should disable dp pull-up
clock_init();
dma_init();
gpio_init();
irq_init();
#if (MODULE_USB_ENABLE)
usb_init();
#endif
// rf_drv_init(0);
user_init ();
irq_enable();
while (1) {
#if(MODULE_WATCHDOG_ENABLE)
wd_clear();
#endif
main_loop ();
}
}
static void init_task_f(void *user_data)
{
timer_init();
task_set_prio(&init_task, TASK_PRIO_MIN);
user_init();
while(1) {
;
}
}
/*---------------------------------------------------------------------------------------------------------------------------------------------------
* MAIN: main routine
*---------------------------------------------------------------------------------------------------------------------------------------------------
*/
//int main(void) __attribute__((noreturn)); /* saves some Bytes but produces warning */
int
main (void)
{
static DATETIME datetime;
uart_init(); // initialize uart
log_main("Init...\n");
wcEeprom_init();
# if (DCF_PRESENT == 1)
dcf77_init (); // initialize dcf77
# endif
display_init (); // initialize display
{ // local to save stack
uint8_t i2c_errorcode;
uint8_t i2c_status;
if (! i2c_rtc_init (&i2c_errorcode, &i2c_status)) // initialize rtc
{
log_main("RTC init failed\n");
// TODO: error handling
}
}
ldr_init (); // initialize ldr
pwm_init (); // initialize pwm
irmp_init (); // initialize irmp
timer_init (); // initialize timer
user_init();
sei (); // enable interrupts
pwm_on (); // switch on pwm
//pwm_set_base_brightness_step(MAX_PWM_STEPS-1); /// @todo remove if ldr stuff is working
log_main("Init finished\n");
for (;;)
{
handle_brightness ();
handle_datetime (&datetime); // check & display new time, if necessary
handle_ir_code (); // handle ir user interaction
# if (DCF_PRESENT == 1)
if (dcf77_getDateTime (&datetime)) // handle dcf77 examination (enable_dcf77_ISR must be TRUE to enable analysis)
{
i2c_rtc_write (&datetime);
soft_seconds = datetime.ss;
user_setNewTime (&datetime);
}
# endif /** (DCF_PRESENT == 1) */
}
}
/**
* @brief Entry point to start execution at
*
* This is the main entry point where execution will start. It initializes the
* hardware and enters an infinite loop handling any upcoming events not yet
* covered.
*
* @note This function makes use of the attribute "OS_main". For details
* refer to [1].
*
* [1]: http://gcc.gnu.org/onlinedocs/gcc/Function-Attributes.html
*
* @return Should never return anything
*/
__attribute__((OS_main)) int main()
{
uart_init();
log_main("Init...\n");
preferences_init();
#if (ENABLE_DCF_SUPPORT == 1)
dcf77_init();
#endif
display_init();
datetime_init();
ldr_init();
pwm_init();
irmp_init();
timer_init();
user_init();
sei();
pwm_on();
log_main("Init finished\n");
while (1) {
brightness_handle();
datetime_handle();
handle_ir_code();
#if (ENABLE_UART_PROTOCOL == 1)
uart_protocol_handle();
#endif /* (ENABLE_UART_PROTOCOL == 1) */
#if (ENABLE_DCF_SUPPORT == 1)
datetime_t dt;
// TODO: Make sure dcf77_getDateTime() validates its result
if (dcf77_get_datetime(&dt)) {
datetime_set(&dt);
}
#endif
}
}
int attach_init() {
char *buf, buf2[1024], *t2, *t3;
int n;
srand(time(0)*2+getpid());
chdir(BBSHOME);
printf("Content-Type: application/octet-stream \n");
printf("Content-Disposition: attachment; filename=tshirt.csv \n\n", CHARSET);
n=atoi(getsenv("CONTENT_LENGTH"));
if(n>5000000) n=5000000;
buf=calloc(n+1, 1);
if(buf==0) http_fatal("memory overflow");
fread(buf, 1, n, stdin);
buf[n]=0;
t2=strtok(buf, "&");
while(t2) {
t3=strchr(t2, '=');
if(t3!=0) {
t3[0]=0;
t3++;
__unhcode(t3);
parm_add(trim(t2), t3);
}
t2=strtok(0, "&");
}
strsncpy(buf2, getsenv("QUERY_STRING"), 1024);
t2=strtok(buf2, "&");
while(t2) {
t3=strchr(t2, '=');
if(t3!=0) {
t3[0]=0;
t3++;
__unhcode(t3);
parm_add(trim(t2), t3);
}
t2=strtok(0, "&");
}
strsncpy(buf2, getsenv("HTTP_COOKIE"), 1024);
t2=strtok(buf2, ";");
while(t2) {
t3=strchr(t2, '=');
if(t3!=0) {
t3[0]=0;
t3++;
parm_add(trim(t2), t3);
}
t2=strtok(0, ";");
}
strsncpy(fromhost, getsenv("REMOTE_ADDR"), 32);
seteuid(BBSUID);
if(geteuid()!=BBSUID) http_fatal("uid error.");
shm_init();
loginok=user_init(¤tuser, &u_info);
if(u_info==0){
u_info=&guest;
}
return;
}
PyMODINIT_FUNC
init_spotify(void)
{
PyObject *m;
if (PyType_Ready(&SessionType) < 0)
return;
if (PyType_Ready(&ArtistType) < 0)
return;
if (PyType_Ready(&ArtistBrowserType) < 0)
return;
if (PyType_Ready(&LinkType) < 0)
return;
if (PyType_Ready(&PlaylistType) < 0)
return;
if (PyType_Ready(&PlaylistContainerType) < 0)
return;
if (PyType_Ready(&ResultsType) < 0)
return;
if (PyType_Ready(&ToplistBrowserType) < 0)
return;
if (PyType_Ready(&TrackType) < 0)
return;
if (PyType_Ready(&ImageType) < 0)
return;
if (PyType_Ready(&UserType) < 0)
return;
m = Py_InitModule("_spotify", module_methods);
if (m == NULL)
return;
PyObject *spotify = PyImport_ImportModule("spotify");
PyObject *d = PyModule_GetDict(spotify);
PyObject *s = PyUnicode_FromString("SpotifyError");
SpotifyError = PyDict_GetItem(d, s);
Py_INCREF(SpotifyError);
SpotifyApiVersion = Py_BuildValue("i", SPOTIFY_API_VERSION);
Py_INCREF(SpotifyApiVersion);
PyModule_AddObject(m, "api_version", SpotifyApiVersion);
album_init(m);
albumbrowser_init(m);
artist_init(m);
artistbrowser_init(m);
link_init(m);
playlist_init(m);
playlistcontainer_init(m);
session_init(m);
search_init(m);
toplistbrowser_init(m);
track_init(m);
image_init(m);
user_init(m);
}
int main() {
int sockfd, newsockfd;
socklen_t clilen;
struct sockaddr_in serv_addr, client_addr;
channel_init(&defaultChannel);
strcpy(defaultChannel.alias, DEFAULT_CHANNEL_ALIAS);
channels_init(&channels);
channels_insert(&channels, &defaultChannel);
if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
printf("ERROR opening socket");
serv_addr.sin_family = AF_INET;
serv_addr.sin_port = htons(PORT);
serv_addr.sin_addr.s_addr = INADDR_ANY;
bzero(&(serv_addr.sin_zero), 8);
if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0)
printf("ERROR on binding");
listen(sockfd, MAX_USERS);
while(1) {
clilen = sizeof(struct sockaddr_in);
if((newsockfd = accept(sockfd, (struct sockaddr *)&client_addr, (socklen_t*)&clilen)) == -1) {
fprintf(stderr, "accept() failed...\n");
return -1;
}
else {
if(defaultChannel.size == MAX_USERS) {
fprintf(stderr, "Connection full, request rejected...\n");
continue;
}
printf("Connection requested received...\n");
struct USER user;
user_init(&user);
user.socketfd = newsockfd;
strcpy(user.alias, "Anonymous");
strcpy(user.channelAlias, DEFAULT_CHANNEL_ALIAS);
channel_insert(&channels.head->channel, &user);
pthread_create(&user.threadInfo, NULL, client_handler, (void *)&user);
}
}
return 0;
}
int main(int argc, char **argv)
{
tintin_exec=argv[0];
init_locale();
user_setdriver(isatty(0)?1:0);
parse_options(argc, argv);
init_bind();
hist_num=-1;
init_parse();
strcpy(status, EMPTY_LINE);
user_init();
/* read_complete(); no tab-completion */
srand((getpid()*0x10001)^time0);
lastdraft=0;
if (ui_own_output || tty)
{
/*
Legal crap does _not_ belong here. Anyone interested in the license
can check the files accompanying KBtin, without any need of being spammed
every time. It is not GNU bc or something similar, we don't want half
a screenful of all-uppercase (cAPS kEY IS STUCK AGAIN?) text that no one
ever wants to read -- that is what docs are for.
*/
tintin_printf(0, "~2~##########################################################");
tintin_printf(0, "#~7~ ~12~K B ~3~t i n~7~ v %-25s ~2~#", VERSION);
tintin_printf(0, "# #");
tintin_printf(0, "#~7~ current developer: ~9~Adam Borowski ([email protected]~7~) ~2~#");
tintin_printf(0, "# #");
tintin_printf(0, "#~7~ based on ~12~tintin++~7~ v 2.1.9 by Peter Unold, Bill Reiss, ~2~#");
tintin_printf(0, "#~7~ David A. Wagner, Joann Ellsworth, Jeremy C. Jack, ~2~#");
tintin_printf(0, "#~7~ Ulan@GrimneMUD and Jakub Narębski ~2~#");
tintin_printf(0, "##########################################################~7~");
tintin_printf(0, "~15~#session <name> <host> <port> ~7~to connect to a remote server");
tintin_printf(0, " ~8~#ses t2t t2tmud.org 9999");
tintin_printf(0, "~15~#run <name> <command> ~7~to run a local command");
tintin_printf(0, " ~8~#run advent adventure");
tintin_printf(0, " ~8~#run sql mysql");
tintin_printf(0, "~15~#help ~7~to get the help index");
}
user_mark_greeting();
setup_signals();
#ifdef PROFILING
setup_prof();
#endif
PROF("initializing");
setup_ulimit();
init_nullses();
PROF("other");
apply_options();
tintin();
return 0;
}
void
changer_find(
void * user_data,
int (*user_init)(void *, int, int, int, int),
int (*user_slot)(void *, int, char *, char *),
char * searchlabel)
{
char *slotstr, *device = NULL, *curslotstr = NULL;
int nslots, checked, backwards, rc, done, searchable;
rc = changer_query(&nslots, &curslotstr, &backwards, &searchable);
if (rc != 0) {
/* Problem with the changer script. Bail. */
g_fprintf(stderr, _("Changer problem: %s\n"), changer_resultstr);
amfree(curslotstr);
return;
}
done = user_init(user_data, rc, nslots, backwards, searchable);
amfree(curslotstr);
if (searchlabel != NULL)
{
dbprintf(_("changer_find: looking for %s changer is searchable = %d\n"),
searchlabel, searchable);
} else {
dbprintf(_("changer_find: looking for NULL changer is searchable = %d\n"),
searchable);
}
if ((searchlabel!=NULL) && searchable && !done){
rc=changer_search(searchlabel, &curslotstr, &device);
if(rc == 0)
done = user_slot(user_data, rc, curslotstr, device);
}
slotstr = "current";
checked = 0;
while(!done && checked < nslots) {
rc = changer_loadslot(slotstr, &curslotstr, &device);
if(rc > 0)
done = user_slot(user_data, rc, curslotstr, device);
else if(!done)
done = user_slot(user_data, 0, curslotstr, device);
amfree(curslotstr);
amfree(device);
checked += 1;
slotstr = "next";
}
}
int main(void) {
AS_Init(AS_MODE_SURROUND | AS_MODE_16CH);
AS_SetDefaultSettings(AS_PCM_8BIT, 11025, AS_SURROUND); // 사운드 출력 방식 셋팅
while (1) {
if (scr_Main()) // 메인 , GAMEEXIT 일 경우 nds 종료
break;
user_init(); // 유저정보 초기화
scr_Game(); // 게임
scr_Gameover(); // 종료
}
UnLoad_Screen(); // 스크린 지우기
return 0;
}
__declspec(dllexport) int
Load(PLUGINLINK * link)
{
PROTOCOLDESCRIPTOR pd;
INITCOMMONCONTROLSEX iccs;
/* setup static data */
pluginLink = link;
/* get main thread handle */
DuplicateHandle(
GetCurrentProcess(), GetCurrentThread(),
GetCurrentProcess(), &g_hMainThread,
THREAD_SET_CONTEXT, FALSE, 0);
/* init common controls library (for the IP adress entries to work)
*/
memset(&iccs, 0, sizeof(iccs));
iccs.dwSize = sizeof(iccs);
iccs.dwICC = ICC_INTERNET_CLASSES;
InitCommonControlsEx(&iccs);
/* init vqp_link module */
vqp_init(NULL, NULL);
/* register this module with miranda */
memset(&pd, 0, sizeof(pd));
pd.cbSize = sizeof(pd);
pd.szName = VQCHAT_PROTO;
pd.type = PROTOTYPE_PROTOCOL;
CallService(MS_PROTO_REGISTERMODULE, 0, (LPARAM)&pd);
/* init our modules */
options_init();
msghandler_init();
skin_init();
user_init();
userlist_init();
chatroom_init();
/* register protocol services */
service_register_services();
service_hook_all();
return 0;
}
int main(void) {
void CreateTasks();
AS_Init(AS_MODE_SURROUND | AS_MODE_16CH);
AS_SetDefaultSettings(AS_PCM_8BIT, 11025, AS_SURROUND);
while (1) { // Infinite loop
if (scr_Main()) // 메인
break; // nds 종료
user_init(); // 유저정보 초기화
scr_Game(); // 게임
scr_Gameover(); // 종료
}
UnLoad_Screen();
return 0;
}
int server_main() {
signal(SIGINT, main_server_on_signal);
if (!pid_init()) {
wetalk_error("Server already running.\n");
}
if (!user_init()) {
wetalk_error("Could not load user database.\n");
}
if (!server_init(main_server_on_newmsg)) {
perror("server_init");
return 1;
}
server_loop();
return 0;
}
/*--------------------------------------------------------------------------
* initialize and enter main event loop
*--------------------------------------------------------------------------
*/
int main(int argc, char **argv) {
/* initialize openGL stuff
*/
printf("QRSIM (c) Embedded Systems Lab, TU Delft\n");
menu();
printf("qrsim control mode %d\n",userstate.control_mode);
init_disp(argc, argv);
/* initialize quad rotor stuff
*/
qr_init(&qrstate, Z_AT_GND);
user_init(&userstate, CONTROL_NONE);
js_init();
/* start simulation
*/
simulations = 0;
start_disp();
return 0;
}
void
changer_current(
void * user_data,
int (*user_init)(void *, int, int, int, int),
int (*user_slot)(void *, int, char *, char *))
{
char *device = NULL, *curslotstr = NULL;
int nslots, backwards, rc, done, searchable;
rc = changer_query(&nslots, &curslotstr, &backwards, &searchable);
done = user_init(user_data, rc, nslots, backwards, searchable);
amfree(curslotstr);
rc = changer_loadslot("current", &curslotstr, &device);
if(rc > 0) {
done = user_slot(user_data, rc, curslotstr, device);
} else if(!done) {
done = user_slot(user_data, 0, curslotstr, device);
}
amfree(curslotstr);
amfree(device);
}
/**
* 清理客户端连接标示
* @param i [description]
*/
void client_clean(int fd) {
/*加锁*/
if (pthread_mutex_lock(&mutex) != 0) {
perror("pthread_mutex_lock");
exit(EXIT_FAILURE);
}
//清除client_map
int ret;
char key[20];
sprintf(key, "%d", fd);
client_elem *c_el;
ret = hashmap_remove(client_map, key, (void **)&c_el);
assert(ret == HMAP_S_OK);
//如果是合法退出，则修改登录状态
if (c_el->token && strlen(c_el->token) > 0) {
_RTS_USER _rts_user = user_init();
_rts_user.id = c_el->id;
_rts_user.status = 0;
user_edit(_rts_user);
//清除client_id_map
bzero(key, 20);
sprintf(key, "%d", c_el->id);
ret = hashmap_remove(client_id_map, key, NULL);
assert(ret == HMAP_S_OK);
RTS_printf("[--退出成功--]: NAME:%s--IP:%s\n", c_el->name, inet_ntoa(c_el->addr.sin_addr));
}
free_client_elem(c_el, 0);
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
}
/**
* 和客户交互
* @param sockfd [description]
* @param i [description]
* @return [description]
*/
int client_interface(struct bufferevent *bev, client_t *client) {
/*加锁*/
if (pthread_mutex_lock(&mutex) != 0) {
perror("pthread_mutex_lock");
exit(EXIT_FAILURE);
}
char data[4096];
int nbytes;
//接收数据
char buffer[MAX_BUF + 1];
bzero(buffer, MAX_BUF + 1);
int n = 0;
int str_len = 0;
int last_len;
while ((nbytes = EVBUFFER_LENGTH(bev->input)) > 0) {
if (nbytes > 4096) nbytes = 4096;
n += evbuffer_remove(bev->input, data, nbytes);
str_len = strlen(data);
last_len = (MAX_BUF - strlen(buffer));
if (last_len >= str_len) {
strncat(buffer, data, str_len);
} else {
strncat(buffer, data, last_len);
}
}
if (n <= 0) {
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0; //客户端退出时
}
//printf("第%d个客户端 IP:%s\n", i + 1, inet_ntoa(_client[i].addr.sin_addr));
//解析客户端信息
int flag = 0;
int ret;
char key[20];
sprintf(key, "%d", client->fd);
client_elem *c_el;
ret = hashmap_get(client_map, key, (void **)&c_el);
assert(ret == HMAP_S_OK);
_RTS_TRANSPORT_DATA _rts_transport_data;
_rts_transport_data = RTS_transport_data_init();
if (RTS_transport_data_parse(buffer, &_rts_transport_data) == 0) {
client_send(client, "{\"code\":\"0001\",\"message\":\"数据格式非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
if (!_rts_transport_data.action) {
client_send(client, "{\"code\":\"0005\",\"message\":\"参数非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
if (strcmp(_rts_transport_data.action, "login") == 0) { //登录
if (strlen(_rts_transport_data.name) == 0 || strlen(_rts_transport_data.password) == 0) {
client_send(client, "{\"code\":\"0005\",\"message\":\"参数非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
//验证用户名密码
_RTS_USER *_rts_user = user_get(0, _rts_transport_data.name);
if (_rts_user->id <= 0) {
client_send(client, "{\"code\":\"0003\",\"message\":\"用户名或密码错误\"}");
CLIENT_FREE(_rts_user);
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
char *pwdhash = RTS_hash(_rts_transport_data.password, _rts_user->salt);
if (strcmp(_rts_user->password, pwdhash) != 0) {
client_send(client, "{\"code\":\"0003\",\"message\":\"用户名或密码错误\"}");
CLIENT_FREE(pwdhash);
CLIENT_FREE(_rts_user);
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
//用户名密码验证通过后，判断当前账号是否已登录
//这时应该已经拿到了用户的id
if (c_el->token && strlen(c_el->token) > 0) { //该设备已经登录了账号，不能再做登录操作
client_send(client, "{\"code\":\"1004\",\"message\":\"该设备已经登录了账号，不能再做登录操作\"}");
CLIENT_FREE(pwdhash);
CLIENT_FREE(_rts_user);
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
if (c_el->id == 0 || (c_el->token && strlen(c_el->token) == 0)) { //未登录，直接执行登录赋值操作
//登录成功，返回认证标示
char *unique = RTS_unique();
strncpy(_rts_transport_data.token, unique, MAX_TOKEN_LENGTH);
CLIENT_FREE(unique);
//重新设置client_map
ret = hashmap_remove(client_map, c_el->key, (void **)&c_el);
assert(ret == HMAP_S_OK);
client_elem *new_c_el = (client_elem *)malloc(sizeof(client_elem));
sprintf(new_c_el->key, "%d", client->fd);
new_c_el->addr = c_el->addr;
new_c_el->client = client;
new_c_el->id = _rts_user->id;
new_c_el->name = (char *)malloc(MAX_NAME_LENGTH + 1);
strncpy(new_c_el->name, _rts_transport_data.name, MAX_NAME_LENGTH);
new_c_el->token = (char *)malloc(MAX_TOKEN_LENGTH + 1);
strncpy(new_c_el->token, _rts_transport_data.token, MAX_TOKEN_LENGTH);
free_client_elem(c_el, 0);
ret = hashmap_put(client_map, new_c_el->key, new_c_el);
assert(ret == HMAP_S_OK);
//设置client_id_map
client_id_elem *cid_el = (client_id_elem *)malloc(sizeof(client_id_elem));
sprintf(cid_el->key, "%d", _rts_user->id);
cid_el->fd = client->fd;
ret = hashmap_put(client_id_map, cid_el->key, cid_el);
assert(ret == HMAP_S_OK);
RTS_printf("[++登录成功++]: NAME:%s--IP:%s\n", _rts_transport_data.name, inet_ntoa(c_el->addr.sin_addr));
bzero(buf, MAX_BUF + 1);
sprintf(buf, "{\"code\":\"0000\",\"message\":\"登录成功\",\"token\":\"%s\",\"id\":%d}", _rts_transport_data.token, _rts_user->id);
client_send(client, buf);
//修改登录成功标识
_RTS_USER _rts_user2 = user_init();
_rts_user2.id = _rts_user->id;
_rts_user2.status = 1;
user_edit(_rts_user2);
CLIENT_FREE(pwdhash);
CLIENT_FREE(_rts_user);
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
client_send(client, "{\"code\":\"1003\",\"message\":\"该用户已经在其他地方成功登录\"}");
CLIENT_FREE(pwdhash);
CLIENT_FREE(_rts_user);
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
if (strcmp(_rts_transport_data.action, "message") == 0) { //聊天
if (strlen(_rts_transport_data.token) == 0 || !_rts_transport_data.toid || !_rts_transport_data.id || wcslen(_rts_transport_data.content) == 0) {
client_send(client, "{\"code\":\"0005\",\"message\":\"参数非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
//认证信息失败，退出客户端
if (!c_el->token || (strcmp(_rts_transport_data.token, "bsh_test_$%1KP@'") != 0 && strcmp(_rts_transport_data.token, c_el->token) != 0)) {
client_send(client, "{\"code\":\"0004\",\"message\":\"token非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
//判断登录者身份
if (!c_el->id || c_el->id != _rts_transport_data.id) {
client_send(client, "{\"code\":\"0006\",\"message\":\"用户身份非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
bzero(key, 20);
sprintf(key, "%d", _rts_transport_data.toid);
client_id_elem *cid_el;
ret = hashmap_get(client_id_map, key, (void **)&cid_el);
printf("发给id:%s\n", key);
if (ret == 0) {
printf("get_toid:%s ret:%d fd:%d\n", key, ret, cid_el->fd);
}
if (ret == HMAP_E_NOTFOUND || !cid_el || !cid_el->fd) {
client_send(client, "{\"code\":\"1002\",\"message\":\"对方不在线\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
bzero(key, 20);
sprintf(key, "%d", cid_el->fd);
client_elem *to_c_el;
ret = hashmap_get(client_map, key, (void **)&to_c_el);
assert(ret == HMAP_S_OK);
if (to_c_el->id == _rts_transport_data.id) { //如果接受者是自己，则发出警告
client_send(client, "{\"code\":\"1001\",\"message\":\"不能给自己发消息\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
bzero(buf, MAX_BUF + 1);
sprintf(buf, "{\"code\":\"0000\",\"message\":\"发送成功\",\"content\":\"%ls\"}", _rts_transport_data.content);
client_send(to_c_el->client, buf);
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
if (strcmp(_rts_transport_data.action, "logout") == 0) { //logout
if (strlen(_rts_transport_data.token) == 0 || !_rts_transport_data.id) {
client_send(client, "{\"code\":\"0005\",\"message\":\"参数非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
//判断登录者身份
if (!c_el->id || c_el->id != _rts_transport_data.id) {
client_send(client, "{\"code\":\"0006\",\"message\":\"用户身份非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
//认证信息失败，退出客户端
if (c_el->token && strcmp(_rts_transport_data.token, c_el->token) != 0) {
client_send(client, "{\"code\":\"0004\",\"message\":\"token非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
client_send(client, "{\"code\":\"0000\",\"message\":\"退出成功\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
if (strcmp(_rts_transport_data.action, "register") == 0) {
if (strlen(_rts_transport_data.name) == 0 || strlen(_rts_transport_data.password) == 0) {
client_send(client, "{\"code\":\"0005\",\"message\":\"参数非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
char *salt = RTS_rand();
char *pwdhash = RTS_hash(_rts_transport_data.password, salt);
char *datetime = RTS_current_datetime();
_rts_transport_data.id = user_add(_rts_transport_data.name, pwdhash, salt, inet_ntoa(c_el->addr.sin_addr), datetime, 0);
CLIENT_FREE(salt);
CLIENT_FREE(pwdhash);
CLIENT_FREE(datetime);
if (_rts_transport_data.id == 0) {
client_send(client, "{\"code\":\"1005\",\"message\":\"注册失败,该用户名已注册\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
if (_rts_transport_data.id > 0) {
bzero(buf, MAX_BUF + 1);
sprintf(buf, "{\"code\":\"0000\",\"message\":\"注册成功\",\"id\":%d}", _rts_transport_data.id);
client_send(client, buf);
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
client_send(client, "{\"code\":\"0007\",\"message\":\"注册失败,未知错误\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 1;
}
client_send(client, "{\"code\":\"0002\",\"message\":\"动作非法\"}");
/*解锁锁*/
if (pthread_mutex_unlock(&mutex) != 0) {
perror("pthread_mutex_unlock");
exit(EXIT_FAILURE);
}
return 0;
}
/*--------------------------------------------------------------------------
* normal keys callback function
*--------------------------------------------------------------------------
*/
void keyb(unsigned char key, int x, int y) {
x = x;
y = y;
int use_body_angles = sim_mode_adjust_body_angles;
switch (key) {
/* simulator control
*/
case 03: /* ^C: exit */
exit(0);
break;
case 07: /* ^G: toggle gravity */
qrstate.sim_control.enable_gravity = ! qrstate.sim_control.enable_gravity;
if (qrstate.sim_control.enable_gravity)
printf("gravity on\n");
else
printf("gravity off\n");
break;
case 13: /* ^M: change control mode + QR reset */
qr_init(&qrstate, Z_AT_GND);
user_init(&userstate, (userstate.control_mode + 1) % 10);
printf("qrsim controller mode %d\n",userstate.control_mode);
break;
case 16: /* ^P: print statistics toggle */
sim_mode_print = ! sim_mode_print;
break;
case 18: /* ^R: reset */
qr_init(&qrstate, Z_AT_GND);
user_init(&userstate, userstate.control_mode);
break;
case 19: /* ^S: freeze simulation toggle */
qrstate.sim_control.pause = ! qrstate.sim_control.pause;
break;
case 20: /* ^T: toggle Euler angle update mode */
sim_mode_adjust_body_angles = ! sim_mode_adjust_body_angles;
if (sim_mode_adjust_body_angles)
printf("Body angles\n");
else
printf("Euler angles\n");
break;
case 22: /* ^V: toggle trace generator */
qrstate.sim_control.verbose = ! qrstate.sim_control.verbose;
break;
case 27: /* ESC: exit */
exit(0);
break;
case '0': /* override - reset x/y/x */
qrstate.x = 0;
qrstate.y = 0;
if (qrstate.z > -0.3) qrstate.z = -0.3;
if (qrstate.z < -3.3) qrstate.z = -3.3;
break;
/* view control
*/
case '1':
camloc.view = VIEW_STATIONARY;
break;
case '2':
camloc.view = VIEW_FLY_BEHIND;
break;
case '3':
camloc.view = VIEW_WALK_BEHIND;
break;
case '4':
camloc.view = VIEW_COCKPIT;
break;
case '5':
camloc.view = VIEW_NORTH_UP;
break;
case '6':
camloc.view = VIEW_TRACK_UP;
break;
case '8': /* zoom in */
camloc.tripod[0] *= 0.99;
camloc.tripod[1] *= 0.99;
camloc.tripod[2] *= 0.99;
break;
case '9': /* zoom out */
camloc.tripod[0] *= 1.01;
camloc.tripod[1] *= 1.01;
camloc.tripod[2] *= 1.01;
break;
/* quad rotor control
*/
case 'a': /* increase lift */
kb_lift += 0.5;
break;
case 'z': /* decrease lift */
kb_lift -= 0.5;
break;
case 'w': /* increase yaw */
kb_yaw += 0.2;
break;
case 'q': /* decrease yaw */
kb_yaw -= 0.2;
break;
case 'o': /* override - phi */
update_euler(&qrstate,-2/RAD2DEG,0,0,use_body_angles);
break;
case 'p': /* override - phi */
update_euler(&qrstate,+2/RAD2DEG,0,0,use_body_angles);
break;
case 'k': /* override - theta */
update_euler(&qrstate,0,-2/RAD2DEG,0,use_body_angles);
break;
case 'l': /* override - theta */
update_euler(&qrstate,0,+2/RAD2DEG,0,use_body_angles);
break;
case 'n': /* override - psi */
update_euler(&qrstate,0,0,-2/RAD2DEG,use_body_angles);
break;
case 'm': /* override - psi */
update_euler(&qrstate,0,0,+2/RAD2DEG,use_body_angles);
break;
case 'x': /* override - x axis */
qrstate.x -= 0.1;
break;
case 'c': /* override - x axis */
qrstate.x += 0.1;
break;
case 'y': /* override - y axis */
qrstate.y -= 0.1;
break;
case 'u': /* override - y axis */
qrstate.y += 0.1;
break;
default: menu();
}
}
void read_data_file(List* user_list, tweetList* word_list){
FILE* fp;
User* user;
//List* user_list;
List* temp_list;
List* root;
char userID[12];
char dayInfo[40];
char userName[20];
//int user_num=0; make it global variable
char temp[12];
char temp2[12];
char userData;
//friend.txt
char friendID[12];
char prevUserID[12];
int friend_num;
int sameuser=0;
User* user_find;
User* user_find2;
User* aaa = (User*)malloc(sizeof(User));
//int friendship_num=0; make it global variable
//tweet.txt
char tweet_temp[150];
char userID_word[12];
int tweetcmp=1;
int i=0;
//tweetList* word_list = (tweetList*)malloc(sizeof(tweetList));
tweetList* temp_word_list = (tweetList*)malloc(sizeof(tweetList));
tweetList* temp_word_list2 = (tweetList*)malloc(sizeof(tweetList));
tweetList* root_word_list = (tweetList*)malloc(sizeof(tweetList));
tweetList* temp_tweet_list = (tweetList*)malloc(sizeof(tweetList));
tweetList* temp_tweet_list2 = (tweetList*)malloc(sizeof(tweetList));
tweetList* root_tweet_list = (tweetList*)malloc(sizeof(tweetList));
User* tweet_user = (User*)malloc(sizeof(User));
//int total_tweet_num=0;// make it global variable
//user_list = (List*)malloc(sizeof(List));
user = (User*)malloc(sizeof(User));
list_init(user_list);
//user_init(user);
//user_list->user = user;
//list initialize.
root = (List*)malloc(sizeof(List));
//list_init(root);
root = user_list;
//read user.txt//
fp= fopen("user.txt", "r");
while(fgets(temp,12,fp) !=NULL){
user = (User*)malloc(sizeof(User));
user_init(user);
temp_list = (List*)malloc(sizeof(List));
//temp2[0]=temp;
//fgets(user->ID, 12, fp);
//sprintf(user->ID, "%c%s", temp, user->ID);
//strcat(temp2, user->ID);
//strcpy(user->ID, temp2);
strcpy(user->ID, temp);
fgets(user->regDay, 40, fp);
fgets(user->name, 20, fp);
//printf("ID : %s", user->ID);
//printf("Day : %s", user->regDay);
//printf("Name : %s", user->name);
if(user_list->user==NULL){
printf("user list is null!\n");
user_list->user = user;
user_list->next = NULL;
continue;
}
temp_list->user = user;
temp_list->next = user_list;
user_list = temp_list;
if(fgetc(fp) == EOF){
printf("break\n");
break;
}
//user_list->next = user_list;
//printf("aaaaaaaaID : %s\n", user_list->user->ID);
//free(user);
user_num++;
}
fclose(fp);
root = user_list;
//store root node.
///* print all user list
while(user_list->next!=NULL){
//printf("ID : %s", user_list->user->ID);
user_list = user_list->next;
}
//printf("total user num : %d\n",user_num);
printf("\n---Read List---\n");
//read friend.txt//
fp = fopen("friend.txt", "r");
while(fgets(temp, 12, fp)!=NULL){
friend_num=0;
if(strcmp(prevUserID,temp)==0){
sameuser=1;
}
else{
sameuser=0;
}
strcpy(userID, temp);
//printf("%s",userID);
user_list = root;
user_find = findUserByID(user_list, userID);
//find user, then find friend of him.
fgets(friendID, 12, fp);
user_list = root;
user_find2 = findUserByID(user_list, friendID);
add_frinedship(user_find, user_find2);
friendship_num++;
//total friendship num
user_find->friends_num++;
//each user's friends num
fgetc(fp);
if(!sameuser){
//user가 바뀌면 friend의 수를 비교해야함.
}
}
printf("friend read end!\n");
printf("friend test!\n\n");
//user_list=root;
/*
while(user_list->user->friends!=NULL){
printf("friends of %s", user_list->user->name);
printf("ID : %s", user_list->user->friends->user->ID);
printf("name : %s", user_list->user->friends->user->name);
friend_num++;
user_list->user->friends = user_list->user->friends->next;
}
printf("friends_num : %d\n\n", friend_num);
*/
//read tweet.txt//
fclose(fp);
fp = fopen("word.txt", "r");
//word_list = (tweetList*)malloc(sizeof(tweetList));
word_list->next=NULL;
word_list->count=0;
//word_list 초기화?
root_word_list = word_list;
while(fgets(userID_word, 12, fp)!=NULL){
temp_word_list = (tweetList*)malloc(sizeof(tweetList));
fgets(user->regDay,40,fp);
fgets(tweet_temp, 150, fp);
//printf("tweet_temp : %s", tweet_temp);
//트윗을 읽음.
fgetc(fp);
//temp_word_list
strcpy(temp_word_list->tweet, tweet_temp);
temp_word_list->next=NULL;
temp_word_list->count=0;
//word_list를 처음부터 끝까지 검사해서 중복되는 단어가 있으면 count만 증가.
//아니면 끝까지 검사만 해서 word_list의 마지막을 가리키기.
word_list = root_word_list;
while(word_list!=NULL){
if( (tweetcmp=strcmp(word_list->tweet, tweet_temp)==0)){
word_list->count++;
break;
}
temp_word_list2=word_list;
word_list = word_list->next;
}
//printf("tweetcmp : %d\n", tweetcmp);
//비교를 했는데 같은 값이 나오지 않았을 경우(0이 아닌 경우), word_list에 추가.
//근데 위의 while문에서, word_list는 이미 마지막을 가리키고 있다.
//따라서 word_list에 바로 temp_tweet을 넣어주기만 하면 됨.
word_list = temp_word_list2;
if(!tweetcmp){
word_list->next= temp_word_list;
word_list->next->count++;
word_list = word_list->next;
}
total_tweet_num++;
//요기까지가 word_list에 저장을 한 것.
//이제 각 사용자들별로 tweet_list를 정리해야 함.
user_list=root;
tweet_user = findUserByID(user_list, userID_word);
//printf("tweet_user ID : %s", tweet_user->ID);
//printf("tweet_user name : %s", tweet_user->name);
temp_tweet_list = (tweetList*)malloc(sizeof(tweetList));
strcpy(temp_tweet_list->tweet, tweet_temp);
temp_tweet_list->next=NULL;
root_tweet_list = tweet_user->tweetList;
//printf("tweetList : %s", tweet_user->tweetList->tweet);
//printf("temp_tweet : %s", tweet_temp);
if(tweet_user->tweetList==NULL){
//printf("tweetList is NULL!\n\n");
tweet_user->tweetList = temp_tweet_list;
tweet_user->tweetList->count=1;
continue;
}
//if문에 안걸렸으면, 즉 tweet_user->tweetList가 null이 아니면.
//while문을 돌리면서 tweetList의 마지막을 찾는다.
//그전에, 같은 단어가 있으면 count만 증가시키고 break.
tweetcmp=1;
while(tweet_user->tweetList!=NULL){
//printf("tweetList : %s", tweet_user->tweetList->tweet);
//printf("temp_tweet : %s", tweet_temp);
i++;
if( (tweetcmp = strcmp(tweet_user->tweetList->tweet, tweet_temp)) ==0){
tweet_user->tweetList->count++;
break;
}
//여기서도 if문에 안걸린 경우. 그냥 next를 확인
temp_tweet_list2 = tweet_user->tweetList;
tweet_user->tweetList = tweet_user->tweetList->next;
}
//while문의 끝까지 돌았는데 일치하는 tweet이 없었으면, 리스트의 마지막을 가리키고 있다.
//그리고 일치하지 않았기 때문에 tweetcmp는 0이 아니다.
//거기에 temp_tweet_list를 넣어주면 됨.
tweet_user->tweetList = temp_tweet_list2;
if(tweetcmp){
tweet_user->tweetList->next = temp_tweet_list;
tweet_user->tweetList->count=1;
tweet_user->tweetList = tweet_user->tweetList->next;
}
//while문의 끝까지 돌았기 때문에 tweetList를 다시 root에 설정.
tweet_user->tweetList = root_tweet_list;
//free(temp_word_list);
}
//리스트의 저장이 끝났으면 root를 다시 불러옴.
word_list = root_word_list;
printf("word_list : %s\n", word_list->tweet);
printf("count : %d\n", word_list->count);
word_list = word_list->next;
printf("word_list : %s", word_list->tweet);
printf("count : %d\n", word_list->count);
word_list = word_list->next;
printf("word_list : %s", word_list->tweet);
printf("count : %d\n", word_list->count);
word_list = word_list->next;
printf("word_list : %s", word_list->tweet);
printf("count : %d\n", word_list->count);
word_list = word_list->next;
printf("word_list : %s", word_list->tweet);
printf("count : %d\n", word_list->count);
//word_list 출력 테스트.
while(word_list->next!=NULL){
//printf("word_list : %s", word_list->tweet);
//if(word_list->count!=1)
// printf("count : %d, word : %s", word_list->count, word_list->tweet);
word_list = word_list->next;
}
//user->tweetList 출력 테스트.
user->tweetList = root_tweet_list;
printf("22\n");
printf("user : %s", user->ID);
printf("user : %s", user->name);
printf("bbb: %s", user->tweetList->next->next->next->tweet);
printf("Total users : %d\n", user_num);
printf("Total friendship record : %d\n", friendship_num);
printf("Total tweets : %d\n", total_tweet_num);
user_list =root;
}
init_system()
{
user_init();
}
void user_reset(void) {
char buffer[100];
g_handles.clear();
g_count.clear();
user_init(g_original_size, g_heap, buffer);
}
void sys_init()
{
char *p,buf[LN_buf+1];
char t;
time_t e,s;
ch.frug.called=0;
ch.frug.monitor=1;
ch.frug.esc=0;
ch.fp=NULL;
sys.rs=FALSE;
/* ch.env=NULL;*/
/* debug_start();*/
ed_init();
sion_init();
nowvar=varinit(LN_str,MAX_var);
sys.var=varinit(LN_str,MAX_var);
sys.result=varinit(LN_result,MAX_var);
strcpy(sys.home,".");
op_system();
if (sion_sr("config","system")==0)
{
exit(1);
}
sion_read(sys.var);
sion_sr("config","result");
sion_read(sys.result);
sion_close();
sys.execute=atoi(getvar("execute",sys.var))+1;
sys.logon =atoi(getvar("logon",sys.var));
sys.post =atoi(getvar("post",sys.var));
sys.t_logon=atoi(getvar("tdlogon",sys.var));
sys.t_post =atoi(getvar("tdpost",sys.var));
strjncpy(sys.netname,getvar("netname",sys.var),LN_title);
strjncpy(sys.netid,getvar("netid",sys.var),LN_netid);
strjncpy(sys.home ,getvar("home",sys.var),LN_dirs);
strjncpy(sys.temp ,getvar("temp",sys.var),LN_dirs);
pool.max=atoi(getvar("poolmax",sys.var));
copyright();
p=getvar("opentime",sys.var);
tzset(); /* 時空関数がまだ現れていないので呼ぶ必要あり */
if (p==NULL)
{
sys.opentime=DT_daytime;
sys.starttime=0;
}else
{
p=pull(buf,p);
t=atoi(buf);
p=pull(buf,p);
s=(t*60+atoi(buf))*60;
p=pull(buf,p);
t=atoi(buf);
p=pull(buf,p);
e=(t*60+atoi(buf))*60;
/*dbprintf("s:%ld,e:%ld",s,e);*/
sys.opentime=e+(e<s?DT_daytime:0)-s;
sys.starttime=DT_daytime-s;
/*dbprintf("o:%ld,s:%ld",sys.opentime,sys.starttime);*/
}
stack_init();
where_init();
user_init();
art_init();
pool_init();
sys.rs=rs_init();
/* sys_log("●システム起動");*/
}
void NetGuard_Special_Limit::loaddata()
{
struct user_data * u_data;
#ifdef userlist_use_simple
struct user_list * m_users = muser_data->get_list();
while (m_users != NULL) {
u_data = m_users->data;
#else
ip_storage_hash::iterator it;
for (it=muser_data->get_list()->begin(); it != muser_data->get_list()->end(); it++) {
u_data = (*it).second;
#endif
struct user_special_accounting_data *accouning_data = (struct user_special_accounting_data *)u_data->module_data[user_special_limit_module_number];
if (accouning_data) {
user_shutdown(u_data);
}
user_init(u_data);
#ifdef userlist_use_simple
m_users = m_users->next;
#endif
}
}
void NetGuard_Special_Limit::savedata()
{
struct user_data * u_data;
FILE *myfile;
ng_logdebug_spam("saving users to %s",db_filename.c_str());
myfile = fopen(db_filename.c_str(), "w+");
if (!myfile) return;
fwrite(ACC_SPECIAL_LIMIT_VERSION_MAGIC,strlen(ACC_SPECIAL_LIMIT_VERSION_MAGIC),1,myfile);
struct user_special_limit_data * limit_data;
int counter = 0;
#ifdef userlist_use_simple
struct user_list * m_users = muser_data->get_list();
while (m_users != NULL) {
u_data = m_users->data;
#else
ip_storage_hash::iterator it;
for (it=muser_data->get_list()->begin(); it != muser_data->get_list()->end(); it++) {
u_data = (*it).second;
#endif
limit_data = (struct user_special_limit_data *)u_data->module_data[user_special_limit_module_number];
if (limit_data) {
counter++;
}
#ifdef userlist_use_simple
m_users = m_users->next;
#endif
}
//write number of users
fwrite(&counter ,sizeof(counter),1, myfile);
#ifdef userlist_use_simple
m_users = muser_data->get_list();
while (m_users != NULL) {
u_data = m_users->data;
#else
for (it=muser_data->get_list()->begin(); it != muser_data->get_list()->end(); it++) {
u_data = (*it).second;
#endif
limit_data = (struct user_special_limit_data *)u_data->module_data[user_special_limit_module_number];
if (limit_data) {
fwrite(&u_data->saddr ,sizeof(u_data->saddr),1, myfile);
} else {
ng_logerror("skipping user %-15s on saving - no data present",inet_ntoa(*(struct in_addr *)&u_data->saddr));
}
#ifdef userlist_use_simple
m_users = m_users->next;
#endif
}
#ifdef userlist_use_simple
m_users = muser_data->get_list();
while (m_users != NULL) {
u_data = m_users->data;
#else
for (it=muser_data->get_list()->begin(); it != muser_data->get_list()->end(); it++) {
u_data = (*it).second;
#endif
limit_data = (struct user_special_limit_data *)u_data->module_data[user_special_limit_module_number];
if (limit_data) {
fwrite(limit_data,sizeof(struct user_special_limit_data),1, myfile);
ng_logdebug_spam("save user %-15s max %lld",inet_ntoa(*(struct in_addr *)&u_data->saddr),limit_data->limit);
}
#ifdef userlist_use_simple
m_users = m_users->next;
#endif
}
ng_logdebug_spam("saved %d users",counter);
fwrite(ACC_SPECIAL_LIMIT_VERSION_MAGIC,strlen(ACC_SPECIAL_LIMIT_VERSION_MAGIC),1,myfile);
fclose(myfile);
}
struct user_special_limit_data * NetGuard_Special_Limit::load_limit_data(struct user_data *u_data, char *filename, int rename_onfail){
FILE *myfile;
struct stat fileinfo;
char *tmpdata;
struct user_special_limit_data * limit_data = NULL;
int i;
off_t f_pos;
ng_logdebug_spam("loading data from %s",db_filename.c_str());
if (stat(db_filename.c_str(),&fileinfo)) {
ng_logerror("cant stat data file %s",db_filename.c_str());
return NULL;
}
myfile = fopen(db_filename.c_str(), "r");
if (!myfile) {
ng_logerror("cant open data file %s",db_filename.c_str());
return NULL;
}
//check file version
tmpdata = (char *)malloc(sizeof(unsigned char)*(strlen(ACC_SPECIAL_LIMIT_VERSION_MAGIC)+1));
tmpdata[strlen(ACC_SPECIAL_LIMIT_VERSION_MAGIC)] = 0;
int count = fread(&tmpdata[0],strlen(ACC_SPECIAL_LIMIT_VERSION_MAGIC),1,myfile);
if ((count != 1) || strcmp(tmpdata,ACC_SPECIAL_LIMIT_VERSION_MAGIC) ) {
ng_logerror("limit: cant read traffic data from %s - illegal format (%s <> %s)",db_filename.c_str(),(char *)tmpdata,ACC_SPECIAL_LIMIT_VERSION_MAGIC);
if (rename_onfail)
{
free(tmpdata);
tmpdata = (char *)malloc(sizeof(unsigned char)*(strlen(db_filename.c_str())+20));
time_t now;
time(&now); /* get the current time */
sprintf(tmpdata,"%s_%d",db_filename.c_str(),(int)now);
ng_log("renaming file to %s",tmpdata);
rename(db_filename.c_str(),tmpdata);
}
return NULL;
}
f_pos = ftell(myfile);
fseek(myfile,fileinfo.st_size-strlen(ACC_SPECIAL_LIMIT_VERSION_MAGIC),SEEK_SET);
count = fread(&tmpdata[0],strlen(ACC_SPECIAL_LIMIT_VERSION_MAGIC),1,myfile);
if ((count != 1) || strcmp(tmpdata,ACC_SPECIAL_LIMIT_VERSION_MAGIC) ) {
ng_logerror("cant read traffic data from %s - illegal (end) format (%s <> %s)",db_filename.c_str(),(char *)tmpdata,ACC_SPECIAL_LIMIT_VERSION_MAGIC);
if (rename_onfail)
{
free(tmpdata);
tmpdata = (char *)malloc(sizeof(unsigned char)*(strlen(db_filename.c_str())+20));
time_t now;
time(&now); /* get the current time */
sprintf(tmpdata,"%s_%d",db_filename.c_str(),(int)now);
ng_log("renaming file to %s",tmpdata);
rename(db_filename.c_str(),tmpdata);
}
return NULL;
}
//set to old position again
fseek(myfile,f_pos,SEEK_SET);
ng_logdebug_spam("loading %lu bytes data",fileinfo.st_size);
int counter = 0;
count = fread(&counter,sizeof(counter),1, myfile);
if (count != 1 ) return NULL;
ng_logdebug_spam("found %d users in file",counter);
u_int32_t saddr;
int found = 0;
unsigned int seek_pos = 0;
for (i=1; i<=counter ; i++ )
{
count = fread(&saddr ,sizeof(saddr),1, myfile);
if (count != 1 ) return NULL;
if (saddr == u_data->saddr)
{
found = 1;
seek_pos = i;
ng_logdebug_spam("found user %-15s on pos %d",inet_ntoa(*(struct in_addr *)&u_data->saddr),seek_pos);
}
}
if (!found) return NULL;
seek_pos = (seek_pos-1) * sizeof(struct user_special_limit_data) + ftell(myfile);
fseek(myfile,seek_pos,SEEK_SET);
limit_data = (struct user_special_limit_data *)malloc(sizeof(struct user_special_limit_data));
count = fread(limit_data,sizeof(struct user_special_limit_data),1, myfile);
if (count != 1 ) {
delete limit_data;
return NULL;
}
ng_logdebug_spam("loaded user data for %-15s (max: %lld) ",inet_ntoa(*(struct in_addr *)&u_data->saddr),limit_data->limit);
fclose(myfile);
free(tmpdata);
return limit_data;
}
int NetGuard_Special_Limit::init(NetGuard_Config *data)
{
general_acccounting = NULL;
muser_data = NULL;
ng_logdebug_spam("init");
if (!data) return -1;
int ret = NetGuard_Module::init(data); //important to get needed links
if (ret) return ret;
if (data_->GetStr("user_special_limit_filename") == "")
{
ng_logerror("need a user_special_limit_filename in config data");
return -2;
}
db_filename=data_->GetStr("user_special_limit_filename");
if (data_->GetModule("module_general_accounting") == NULL) {
ng_logerror("need general_accounting module needs to be loaded");
return -2;
}
//filter for accountable traffic
if (data_->GetStr("accounting_filter_own") == "")
{
ng_logerror("need an accounting_filter_own in config data");
return -2;
}
std::string filter_name_own=data_->GetStr("accounting_filter_own");
filter_own = NetGuard_Global_IP_Filter::Filter(filter_name_own);
if (filter_own == NULL)
{
ng_logerror("filter passed with accounting_filter_own (%s) does not exists",filter_name_own.c_str());
return -2;
}
ng_logdebug_spam("using filter (%s) for filtering accounting ips",filter_own->GetPrefixName().c_str());
//filter for internal traffic
if (data_->GetStr("accounting_filter_intern") == "")
{
ng_logerror("need an accounting_filter_intern in config data");
return -2;
}
std::string filter_name_intern=data_->GetStr("accounting_filter_intern");
filter_intern = NetGuard_Global_IP_Filter::Filter(filter_name_intern);
if (filter_intern == NULL)
{
ng_logerror("filter passed with accounting_filter_intern (%s) does not exists",filter_name_intern.c_str());
return -2;
}
ng_logdebug_spam("using filter (%s) for filtering internal traffic",filter_intern->GetPrefixName().c_str());
if (data_->GetInt("default_limit") != MININT)
{
default_limit = data_->GetInt("default_limit");
default_limit = default_limit*1024*1024;
ng_logdebug("set default_limit to %llu",default_limit);
}
if (data_->GetInt("default_max_limit") != MININT)
{
default_max_limit = data_->GetInt("default_max_limit");
default_max_limit = default_max_limit*1024*1024;
ng_logdebug("set default_max_limit to %llu",default_max_limit);
}
if (data_->GetInt("default_daily_addition") != MININT)
{
default_daily_addition = data_->GetInt("default_daily_addition");
default_daily_addition = default_daily_addition*1024*1024;
ng_logdebug("set default_daily_addition to %llu",default_daily_addition);
}
general_acccounting = (NetGuard_General_Module*)data_->GetModule("module_general_accounting");
muser_data = (User_Data_Tools*)general_acccounting->get_data(NULL);
my_dis_state = NetGuard_State_Handler::get_state(GlobalCFG::GetStr("user_special_limit.disable_state","disabled"));
if (!my_dis_state) {
ng_logerror("%s state %s unkown",__FUNCTION__,GlobalCFG::GetStr("user_special_limit.disable_state","disabled").c_str());
return -2;
}
my_fail_state = NetGuard_State_Handler::get_state(GlobalCFG::GetStr("user_special_limit.failure_state","failure"));
if (!my_fail_state) {
ng_logerror("%s state %s unkown",__FUNCTION__,GlobalCFG::GetStr("user_special_limit.failure_state","failure").c_str());
return -2;
}
my_enabled_state = NetGuard_State_Handler::get_state(GlobalCFG::GetStr("user_special_limit.enabled_state","enabled"));
if (!my_enabled_state) {
ng_logerror("%s state %s unkown",__FUNCTION__,GlobalCFG::GetStr("user_special_limit.enabled_state","enabled").c_str());
return -2;
}
loaddata();
return 0;
}
void NetGuard_Special_Limit::shutdown()
{
ng_logdebug_spam("shutdown");
if (muser_data)
{
struct user_data *u_data;
#ifdef userlist_use_simple
struct user_list *m_users = muser_data->get_list();
while (m_users != NULL) {
u_data = m_users->data;
#else
ip_storage_hash::iterator it;
for (it=muser_data->get_list()->begin(); it != muser_data->get_list()->end(); it++) {
u_data = (*it).second;
#endif
user_shutdown(u_data);
#ifdef userlist_use_simple
m_users = m_users->next;
#endif
}
}
general_acccounting = NULL;
muser_data = NULL;
}
struct user_special_limit_data *NetGuard_Special_Limit::my_user_init(struct user_data *u_data, bool doload)
{
struct user_special_limit_data * limit_data = NULL;
//try to load it from file
if (doload)
limit_data = load_limit_data(u_data,NULL,1);
if (limit_data == NULL)
{
ng_logdebug("setting new default special limits for %-15s",inet_ntoa(*(struct in_addr *)&u_data->saddr));
//we need to init a new user
//we need to init a new user
limit_data = (struct user_special_limit_data *)malloc(sizeof(struct user_special_limit_data));
//set default values
memset(limit_data,0,sizeof(struct user_special_limit_data));
limit_data->limit = (long long int)default_limit;
limit_data->daily_addition = (unsigned long long int)default_daily_addition;
limit_data->max_limit = (unsigned long long int)default_max_limit;
u_data->external.bytes = limit_data->limit;
};
u_data->module_data[user_special_limit_module_number] = limit_data;
return limit_data;
}
void NetGuard_Special_Limit::user_init(struct user_data *u_data)
{
if (!u_data) return;
ng_logdebug_spam("user_init for %-15s",inet_ntoa(*(struct in_addr *)&u_data->saddr));
my_user_init(u_data,true);
}
void NetGuard_Special_Limit::user_shutdown(struct user_data *u_data)
{
struct user_special_limit_data * limit_data = (struct user_special_limit_data *)u_data->module_data[user_special_limit_module_number];
if ( limit_data != NULL ) {
ng_logdebug_spam("free limits data for %-15s",inet_ntoa(*(struct in_addr *)&u_data->saddr));
delete limit_data;
}
u_data->module_data[user_special_limit_module_number] = NULL;
}
void NetGuard_Special_Limit::do_user_data_forgetday(int day, struct user_data *u_data){
struct user_special_limit_data *limit_data = (struct user_special_limit_data *)u_data->module_data[user_special_limit_module_number];
long long int temp = u_data->external.bytes;
if ( limit_data == NULL ) {
limit_data = my_user_init(u_data,true);
}
if((u_data->external.bytes + limit_data->daily_addition) > limit_data->max_limit)
u_data->external.bytes = limit_data->max_limit;
else
u_data->external.bytes += limit_data->daily_addition;
NetGuard_User_State* nu_state = NetGuard_State_Handler::user_state(u_data);
if (*nu_state == my_enabled_state) return;
if (*nu_state == my_fail_state) return;
if(temp == 0){
if (!nu_state->set(my_enabled_state,GlobalCFG::GetStr("user_special_limit.enable_new_traffic","automatic enabling with new traffic"))) {
ng_logerror("%s - %s - %d - ip: %s vlan: %d - could not do the state transition from %s to %s",__FUNCTION__,__FILE__,__LINE__,inet_ntoa(*(struct in_addr *)&nu_state->Getuser().saddr),nu_state->Getuser().vlan_id,nu_state->state()->GetName().c_str(),my_dis_state->GetName().c_str());
return;
}
}
}
void NetGuard_Special_Limit::user_data_forgetday(int day)
{
struct user_data *u_data;
#ifdef userlist_use_simple
struct user_list *m_users = muser_data->get_list();
while (m_users != NULL) {
u_data = m_users->data;
#else
ip_storage_hash::iterator it;
for (it=muser_data->get_list()->begin(); it != muser_data->get_list()->end(); it++) {
u_data = (*it).second;
#endif
//forget this day for external traffic
do_user_data_forgetday(day,u_data);
#ifdef userlist_use_simple
m_users = m_users->next;
#endif
}
}
void NetGuard_Special_Limit::checkmax(struct user_special_limit_data * limit_data,struct user_data *u_data) {
if(limit_data->limit > 0)
{
if(u_data->external.bytes <= 0)
{
// we have somebody over the internal over all limit
NetGuard_User_State* nu_state = NetGuard_State_Handler::user_state(u_data);
//ignore if we are already in the same state
if (*nu_state == my_dis_state) return;
if (*nu_state == my_fail_state) return;
//nu_state->params()->SetInt("external_limit_exceeded",nu_state->params()->GetInt("external_limit_exceeded",0)+1);
if (!nu_state->set(my_dis_state,GlobalCFG::GetStr("user_limit.disable_int_state_overall_msg","user have no traffic left"))) {
ng_logerror("%s - %s - %d - ip: %s vlan: %d - could not do the state transition from %s to %s",__FUNCTION__,__FILE__,__LINE__,inet_ntoa(*(struct in_addr *)&nu_state->Getuser().saddr),nu_state->Getuser().vlan_id,nu_state->state()->GetName().c_str(),my_dis_state->GetName().c_str());
return;
}
}
}
}
void NetGuard_Special_Limit::packet_in(struct user_data *u_data, int *mode, unsigned int *vlanid, struct tpacket_hdr *h, struct ether_header *eth, struct iphdr *ip, struct tcphdr *tcp, void *data)
{
in_addr_t *index_addr;
in_addr_t *index_addr2;
index_addr = 0;
index_addr2 = 0;
//we are only interested in packages that are linked to a user already
if (!u_data) return;
struct user_special_limit_data *limit_data = (struct user_special_limit_data *)u_data->module_data[user_special_limit_module_number];
if ( limit_data == NULL ) {
limit_data = my_user_init(u_data,true);
}
if (eth->ether_type == htons_ETHERTYPE_IP) {
hl_saddr = ntohl(ip->saddr);
hl_daddr = ntohl(ip->daddr);
if ((*filter_own)==&hl_saddr) index_addr = &ip->saddr;
if ((*filter_own)==&hl_daddr) index_addr2 = &ip->daddr;
} else if (eth->ether_type == htons_ETHERTYPE_ARP) {
struct ether_arp * arph;
//#if __GNUC__ >= 4
arph = (struct ether_arp *)ip;
//#else
//(void *)arph = (void *)ip;
//#endif
hl_saddr = ntohl(*(uint32_t *)&arph->arp_spa);
hl_daddr = ntohl(*(uint32_t *)&arph->arp_tpa);
if ((*filter_own)==&hl_saddr) index_addr = (uint32_t *)&arph->arp_spa;
if ((*filter_own)==&hl_daddr) index_addr2 = (uint32_t *)&arph->arp_tpa;
}
struct user_data_traffic *traffic_type;
//set if internal or external traffic
//default is internal for all non ip protocols
if (eth->ether_type == htons_ETHERTYPE_IP) {
//on ip its external per default
traffic_type = &u_data->external;
//if it match the filters its internal -> return
if ((*filter_intern) == &hl_daddr && (*filter_intern)==&hl_saddr) return;
}else
return;
signed long long int tmpval;
tmpval = traffic_type->bytes;
if((tmpval - h->tp_len) >= 0)
traffic_type->bytes = traffic_type->bytes - h->tp_len;
else
traffic_type->bytes = 0;
checkmax(limit_data,u_data);
}
void NetGuard_Special_Limit::got_input(std::vector<std::string> params, std::vector<int> intparams, std::string command)
{
if (params[0] == "help")
{
ng_logout("save - save limit data");
ng_logout("dumpip <ip> <vlan> - show details for an ip");
ng_logout("dumpip_all <ip> <vlan> - show details for an ip");
ng_logout("set_limit <ip> <vlan> <start limit MB> <daily add MB> <maximum MB> - set limit data for an ip");
//TODO: ng_logout("set_limit_reset <ip> <vlan> - set limit data for an ip to default");
ng_logout("set_current_traffic <ip> <vlan> <traffic MB> - set the current traffic volume for an ip");
}
if (params[0] == "version")
{
ng_logext(100,"%s - Version: %s build: %s from %s at %s with %s",NetGuard_NAME, NetGuard_VERSION,NetGuard_COMPILE_DATE,NetGuard_COMPILE_BY,NetGuard_COMPILE_HOST,NetGuard_COMPILER);
}
if (params[0] == "save")
{
savedata();
}
if ((params[0] == "dumpip") || (params[0] == "dumpip_all"))
{
if (params.size() != 3)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan>",params[0].c_str());
return;
}
struct in_addr m_ip;
if (!inet_aton(params[1].c_str(),&m_ip ))
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan>",params[0].c_str());
return;
}
if (intparams[2]==MININT)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan>",params[0].c_str());
return;
}
unsigned int tmpvlanid = intparams[2];
struct user_data *u_data = muser_data->get_user(&m_ip.s_addr,&tmpvlanid);
if (!u_data)
{
ng_logout_not_found("could not find user with ip: %s vlan: %d",inet_ntoa(*(struct in_addr *)&m_ip.s_addr),intparams[2]);
return;
}
#define EINHEIT 1024/1024
struct user_special_limit_data *limit_data = (struct user_special_limit_data *)u_data->module_data[user_special_limit_module_number];
if ( limit_data == NULL ) {
limit_data = my_user_init(u_data,true);
}
ng_logout("start limit \t\t: %llu MByte",limit_data->limit/EINHEIT);
ng_logout("daily addition \t\t: %llu MByte",limit_data->daily_addition/EINHEIT);
ng_logout("max limit \t\t: %llu MByte",limit_data->max_limit/EINHEIT);
ng_logout("current usage \t\t: %llu MByte",u_data->external.bytes/EINHEIT);
}
if ((params[0] == "set_limit"))
{
if (params.size() != 6)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <start limit MB> <daily add MB> <maximum MB>",params[0].c_str());
return;
}
struct in_addr m_ip;
if (!inet_aton(params[1].c_str(),&m_ip ))
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <start limit MB> <daily add MB> <maximum MB>",params[0].c_str());
return;
}
if (intparams[2]==MININT)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <start limit MB> <daily add MB> <maximum MB>",params[0].c_str());
return;
}
if (intparams[3]==MININT)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <start limit MB> <daily add MB> <maximum MB>",params[0].c_str());
return;
}
if (intparams[4]==MININT)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <start limit MB> <daily add MB> <maximum MB>",params[0].c_str());
return;
}
if (intparams[5]==MININT)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <start limit MB> <daily add MB> <maximum MB>",params[0].c_str());
return;
}
unsigned int tmpvlanid = intparams[2];
struct user_data *u_data = muser_data->get_user(&m_ip.s_addr,&tmpvlanid);
if (!u_data)
{
ng_logout_not_found("could not find user with ip: %s vlan: %d",inet_ntoa(*(struct in_addr *)&m_ip.s_addr),intparams[2]);
return;
}
struct user_special_limit_data *special_limit_data = (struct user_special_limit_data *)u_data->module_data[user_special_limit_module_number];
if ( special_limit_data == NULL ) {
special_limit_data = my_user_init(u_data,true);
}
#define EINHEIT2 1024*1024
unsigned long long int tmpval,tmpval2,tmpval3;
tmpval = (unsigned long long int)intparams[3] * (unsigned long long int)EINHEIT2;
tmpval2 = (unsigned long long int)intparams[4] * (unsigned long long int)EINHEIT2;
tmpval3 = (unsigned long long int)intparams[5] * (unsigned long long int)EINHEIT2;
ng_logout_ok("setting new special limit data");
special_limit_data->limit = tmpval;
special_limit_data->daily_addition = tmpval2;
special_limit_data->max_limit = tmpval3;
}
if ((params[0] == "set_current_traffic"))
{
if (params.size() != 4)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <traffic MB>",params[0].c_str());
return;
}
struct in_addr m_ip;
if (!inet_aton(params[1].c_str(),&m_ip ))
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <traffic MB>",params[0].c_str());
return;
}
if (intparams[2]==MININT)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <traffic MB>",params[0].c_str());
return;
}
if (intparams[3]==MININT)
{
ng_logout_ret(RET_WRONG_SYNTAX,"usage: %s <ip> <vlan> <traffic MB>",params[0].c_str());
return;
}
unsigned int tmpvlanid = intparams[2];
struct user_data *u_data = muser_data->get_user(&m_ip.s_addr,&tmpvlanid);
if (!u_data)
{
ng_logout_not_found("could not find user with ip: %s vlan: %d",inet_ntoa(*(struct in_addr *)&m_ip.s_addr),intparams[2]);
return;
}
#define EINHEIT 1024/1024
#define EINHEIT2 1024*1024
unsigned long long int tmpval;
tmpval = (unsigned long long int)intparams[3] * (unsigned long long int)EINHEIT2;
u_data->external.bytes = tmpval;
ng_logout_ok("new traffic successfully set to: %llu MB",u_data->external.bytes/EINHEIT);
}
}
void NetGuard_Special_Limit::timer_tick() {
}
void setup(void)
{
unsigned char adr, flags, d;
unsigned short i;
unsigned char *p;
_flkey = 0;
/* first disable watchdog */
watchdog_disable();
/* avoid any blocking of RS485 bus */
RS485_ENABLE = RS485_ENABLE_OFF;
/* Port and oscillator configuration */
#if defined(CPU_C8051F120)
SFRPAGE = CONFIG_PAGE;
XBR0 = 0x04; // Enable XBar, UART0 & UART1
XBR1 = 0x00;
XBR2 = 0x44;
#ifdef CLK_25MHZ
/* Select internal quartz oscillator */
SFRPAGE = LEGACY_PAGE;
FLSCL = 0x00; // set flash read time for <25 MHz
SFRPAGE = CONFIG_PAGE;
OSCICN = 0x83; // divide by 1
CLKSEL = 0x00; // select internal oscillator
#else // 98 MHz
/* Select internal quartz oscillator */
SFRPAGE = LEGACY_PAGE;
FLSCL = 0xB0; // set flash read time for 100 MHz
SFRPAGE = CONFIG_PAGE;
OSCICN = 0x83; // divide by 1
CLKSEL = 0x00; // select internal oscillator
PLL0CN |= 0x01;
PLL0DIV = 0x01;
PLL0FLT = 0x01;
PLL0MUL = 0x04;
for (i = 0 ; i < 15; i++); // Wait 5us for initialization
PLL0CN |= 0x02;
for (i = 0 ; i<50000 && ((PLL0CN & 0x10) == 0) ; i++);
CLKSEL = 0x02; // select PLL as sysclk src
#endif
#elif defined(CPU_C8051F020)
XBR0 = 0x04; // Enable UART0 & UART1
XBR1 = 0x00;
XBR2 = 0x44;
/* Select external quartz oscillator */
OSCXCN = 0x67; // Crystal mode, Power Factor 22E6
OSCICN = 0x08; // CLKSL=1 (external)
#elif defined(CPU_C8051F310) || defined(CPU_C8051F320)
XBR0 = 0x01; // Enable RX/TX
XBR1 = 0x40; // Enable crossbar
/* Select internal quartz oscillator */
OSCICN = 0x83; // IOSCEN=1, SYSCLK=24.5 MHz
CLKSEL = 0x00; // derive SYSCLK from internal source
#else
XBR0 = 0x04; // Enable RX/TX
XBR1 = 0x00;
XBR2 = 0x40; // Enable crossbar
PRT0CF = 0x01; // P0.0: TX = Push Pull
PRT1CF = 0x00; // P1
PRT2CF = 0x00; // P2 Open drain for 5V LCD
PRT3CF = 0x20; // P3.5: RS485 enable = Push Pull
/* Select external quartz oscillator */
OSCXCN = 0x67; // Crystal mode, Power Factor 22E6
OSCICN = 0x08; // CLKSL=1 (external)
#endif
#ifdef CFG_HAVE_LCD
lcd_setup();
#endif
#ifdef CFG_HAVE_EMIF
/* initialize external memory interface */
d = emif_init();
/* do memory test on cold start */
SFRPAGE = LEGACY_PAGE;
if (d > 0 && (RSTSRC & 0x02) > 0)
emif_test(d);
#endif
/* start system clock */
sysclock_init();
/* enable watchdog with default timeout */
watchdog_enable(0);
/* enable missing clock detector */
RSTSRC |= 0x04;
/* default LED mode */
for (i=0 ; i<N_LED ; i++)
led_mode(i, 1);
/* initialize all memory */
CSR = 0;
addressed = 0;
flash_param = 0;
flash_program = 0;
flash_allowed = 0;
wrong_cpu = 0;
_flkey = 0;
#ifdef CFG_HAVE_RTC
rtc_set = 0;
#endif
i_in = i_out = n_out = 0;
_cur_sub_addr = 0;
for (i=0 ; i<sizeof(in_buf) ; i++)
in_buf[i] = 0;
for (i=0 ; i<sizeof(out_buf) ; i++)
out_buf[i] = 0;
/* check if we got reset by watchdog */
#if defined(CPU_C8051F120)
SFRPAGE = LEGACY_PAGE;
#endif
WD_RESET = ((RSTSRC & 0x02) == 0 && (RSTSRC & 0x08) > 0);
/* initialize UART(s) */
uart_init(0, BD_115200);
#ifdef CFG_UART1_MSCB
uart_init(1, BD_115200);
#endif
#ifdef CFG_DYN_VARIABLES
setup_variables();
#endif
/* count variables */
for (n_variables = _var_size = 0;; n_variables++) {
_var_size += variables[n_variables].width;
if (variables[n_variables].width == 0)
break;
}
/* check if variables are in xdata and xdata is present */
if (n_variables > 0) {
p = variables[0].ud;
d = *p;
*p = 0x55;
if (*p != 0x55)
wrong_cpu = 1;
*p = 0xAA;
if (*p != 0xAA)
wrong_cpu = 1;
*p = d;
}
/* retrieve EEPROM data */
#ifdef CPU_C8051F120
SFRPAGE = LEGACY_PAGE;
#endif
if ((RSTSRC & 0x02) > 0)
flags = eeprom_retrieve(1); // vars on cold start
else
flags = eeprom_retrieve(0);
if ((flags & (1 << 0)) == 0) {
configured_addr = 0;
/* set initial values */
sys_info.node_addr = 0xFFFF;
sys_info.group_addr = 0xFFFF;
memset(sys_info.node_name, 0, sizeof(sys_info.node_name));
strncpy(sys_info.node_name, node_name, sizeof(sys_info.node_name));
} else
configured_addr = 1;
/* store SVN revision */
sys_info.svn_revision = (svn_rev_main[6]-'0')*1000+
(svn_rev_main[7]-'0')*100+
(svn_rev_main[8]-'0')*10+
(svn_rev_main[9]-'0');
if ((flags & (1 << 1)) == 0) {
/* init variables */
for (i = 0; variables[i].width; i++)
if (!(variables[i].flags & MSCBF_DATALESS)) {
/* do it for each sub-address */
for (adr = 0 ; adr < _n_sub_addr ; adr++) {
memset((char*)variables[i].ud + _var_size*adr, 0, variables[i].width);
}
}
/* call user initialization routine with initialization */
user_init(1);
/* write current variables to flash later in main routine */
configured_vars = 0;
} else {
/* call user initialization routine without initialization */
user_init(0);
configured_vars = 1;
}
/* Blink LEDs */
for (i=0 ; i<N_LED ; i++)
led_blink(i, 3, 150);
}
int main()
{
int ret;
modules_init();//模块的初始化
user_init();
/* system tag data */
store_or_retrive_taged_config();
get_mcu_version_from_psm();
LOG_DEBUG("rst cause is %x\r\n", boot_reset_cause());
#ifdef MIIO_COMMANDS
mcmd_enqueue_raw("MIIO_model_req");
mcmd_enqueue_raw("MIIO_mcu_version_req");
#endif
ret = os_mutex_create(&network_state_mutex,"network_state",OS_MUTEX_INHERIT);
if(WM_SUCCESS != ret) {
LOG_ERROR("create network_state_mutex fail");
goto main_error;
}
register_device_network_state_observer(led_network_state_observer);
/*set wifi calbration data*/
product_set_wifi_cal_data();
/* Start the application framework */
if ((ret = app_framework_start(common_event_handler)) != WM_SUCCESS) {
LOG_ERROR("Failed to start application framework.\r\n");
#if defined(CONFIG_CPU_MC200)
if (-WLAN_ERROR_FW_DNLD_FAILED == ret || -WLAN_ERROR_FW_NOT_DETECTED == ret
|| -WLAN_ERROR_FW_NOT_READY == ret) {
LOG_WARN("Wifi firmware broken, trying to recover.\r\n");
if (recover_wifi_fw() != WM_SUCCESS) {
LOG_FATAL("Recovering wifi fw failed!!!\r\n");
} else {
LOG_WARN("Recovering wifi fw success.\r\n");
pm_reboot_soc();
}
}
#endif
appln_critical_error_handler((void *) -WM_FAIL);
}
ret = is_factory_mode_enable();
LOG_DEBUG("factory mode magic is %x\r\n" ,get_factory_mode_enable());
if (0 == ret) {
// user mode
static xTaskHandle task_handle;
if (app_thread) {
xTaskCreate(app_thread, (signed char*) "app", 2000, NULL, APPLICATION_PRIORITY,
&task_handle);
xTaskCreate(app_uart_thread, (signed char*) "app_uart", 2000, NULL, APPLICATION_PRIORITY,
&task_handle);
LOG_INFO("App thread and App uart thread started.\r\n");
}
} else {
// factory mode
static xTaskHandle task_handle;
if (app_test_thread) {
xTaskCreate(app_test_thread, (signed char*) "test", 2000, NULL, 2,
&task_handle);
LOG_INFO("Test thread started.\r\n");
}
}
// Initialize power management
hp_pm_init();
//enter main loop, main loop will be used as a worker thread
main_loop_run();
//main loop return means error
main_error:
pm_reboot_soc(); //error occure on main thread , reboot
return 1;
}
lr_symbol* lr_parser::parse()
{
/* set up direct reference to tables to drive the parser */
production_tab = production_table();
action_tab = action_table();
reduce_tab = reduce_table();
/* initialize the action encapsulation object */
init_actions();
/* do user initialization */
user_init();
/* get the first token */
cur_token = scan();
/* push dummy symbol with start state to get us underway */
stack.remove_all_elements();
lr_symbol dummy_sym(0, start_state());
stack.push(&dummy_sym);
/* continue until accept or fatal error */
while (true)
{
/* Check current token for freshness. */
assert(-1 == cur_token->parse_state);
/* current state is always on the top of the stack */
/* look up action out of the current state with the current input */
int act = get_action(stack.peek()->parse_state, cur_token->sym);
/* decode the action -- > 0 encodes shift */
if (act > 0)
{
act = act - 1;
DEBUG_LOG("Shift and goto " << act);
/* shift to the encoded state by pushing it on the stack */
cur_token->parse_state = act;
stack.push(cur_token);
/* advance to the next Symbol */
cur_token = scan();
}
/* if its less than zero, then it encodes a reduce action */
else if (act < 0)
{
act = (-act) - 1;
DEBUG_LOG("Reduce by rule " << act);
/* perform the action for the reduce */
lr_symbol* lhs_sym = do_action(act);
/* check for accept indication */
if (lhs_sym == 0)
{
return stack.peek();
}
/* look up information about the production */
lhs_sym->sym = production_tab[act].lhs_sym;
short handle_size = production_tab[act].rhs_size;
/* pop the handle off the stack */
stack.npop(handle_size);
/* look up the state to go to from the one popped back to */
act = get_reduce(stack.peek()->parse_state, lhs_sym->sym);
/* shift to that state */
lhs_sym->parse_state = act;
stack.push(lhs_sym);
DEBUG_LOG(" and goto " << act);
}
/* finally if the entry is zero, we have an error */
else if (act == 0)
{
DEBUG_LOG("Error");
/* call user syntax error reporting routine */
syntax_error(cur_token);
/* try to error recover */
switch (error_recovery())
{
case ERS_FAIL:
/* if that fails give up with a fatal syntax error */
unrecovered_syntax_error(cur_token);
return 0;
case ERS_SUCCESS:
break;
case ERS_ACCEPT:
return stack.peek();
default:
assert(0);
}
}
}
}
void call_user_start(void) {
user_init();
}
