diff options
| -rw-r--r-- | segget/segget.cpp | 126 | ||||
| -rw-r--r-- | segget/segget.h | 8 |
2 files changed, 92 insertions, 42 deletions
diff --git a/segget/segget.cpp b/segget/segget.cpp index e5110a3..59c96ed 100644 --- a/segget/segget.cpp +++ b/segget/segget.cpp @@ -282,6 +282,17 @@ int download_pkgs(){ return EXIT_FAILURE; } } +void *print_message_function(void * ){ + while (true){ + ulong time_diff_msecs=time_left_from(stats.previous_time); + if (time_diff_msecs >= settings.current_speed_time_interval_msecs){ + show_progress(time_diff_msecs); + }; + stats.show_totals(); + sleep(1); + } + return 0; +} void launch_tui_thread(){ pthread_t tui_thread; @@ -331,24 +342,13 @@ void segget_exit(int sig){ exit(0); } -int main() -{ +int routine(){ try{ signal(SIGABRT,segget_exit);//If program aborts go to assigned function "segget_exit". signal(SIGTERM,segget_exit);//If program terminates go to assigned function "segget_exit". signal(SIGINT,segget_exit);//If program terminates go to assigned function "segget_exit". prev_time=time((time_t *)NULL); try{ - //init curses - initscr(); - curs_set(0); - refresh(); - } - catch(...) - { - //error while init curses - } - try{ //load settings settings.init(); } @@ -378,53 +378,95 @@ int main() } try{ load_pkgs(); - } - catch(...){ + }catch(...){ //error while loading pkgs } try{ //show_pkgs(); stats.show_totals(); - } - catch(...){ + }catch(...){ //error while showing stats } try{ download_pkgs(); - } - catch(...){ + }catch(...){ //error while downloading_pkgs } - getch(); - } - catch(...){ - //error during init and downloading process - } - try{ - endwin(); - } - catch(...) + try{ + close(ui_server.server_sockfd); + }catch(...){ + //error while clossing server_sockfd + } + return 0; + }catch(...) { - //error while ending curses + perror("Error in segget.cpp: routine()"); } + return 1; +} + +void start_daemon_mode(){ try{ - close(ui_server.server_sockfd); + debug("start_daemon_mode()"); + switch (fork()) + { + case 0: break; //child + case -1: error_log("Error in segget.cpp: start_daemon_mode() during first fork()"); //fork error + default: exit(0); //parent + } + struct rlimit _ResourceLimit; + if (getrlimit(RLIMIT_NOFILE, &_ResourceLimit)!=0) error_log("Error in segget.cpp: start_daemon_mode() during getrlimit()"); + if (_ResourceLimit.rlim_max==0) throw("Error in segget.cpp: start_daemon_mode() - no open file descriptors"); + for (uint i=0;i<_ResourceLimit.rlim_max;i++) close(i); + if (setsid()==-1) error_log_no_msg("Error in segget.cpp: start_daemon_mode() during setsid()"); + switch (fork()){ + case 0: break; // child + case -1: error_log_no_msg("Error in segget.cpp: start_daemon_mode() during second fork()"); + default: exit(0); // parent + } + umask(0); + // prevent unmount problems - switch to root +// chdir("/"); + // create descriptors in case someone will use them + int fileDesc = open("/dev/null", O_RDWR); //stdin + dup(fileDesc); //stdout + dup(fileDesc); //stderr + log_no_msg("Starting daemon routine"); + routine(); + log_no_msg("Exited daemon routine"); + }catch (...){ + error_log_no_msg("Error in segget.cpp: start_daemon_mode()"); + exit(1); } - catch(...) - { - //error while clossing server_sockfd +} + +int init_curses(){ + try{ + initscr(); + curs_set(0); + refresh(); + return 0; + }catch(...){ + perror("Error in segget.cpp: init_curses()"); } - return 0; + return 1; } -void *print_message_function(void * ){ - while (true){ - ulong time_diff_msecs=time_left_from(stats.previous_time); - if (time_diff_msecs >= settings.current_speed_time_interval_msecs){ - show_progress(time_diff_msecs); - }; - stats.show_totals(); - sleep(1); +int main() +{ + try{ + bool daemon_mode_flag=true; + if (daemon_mode_flag){ + start_daemon_mode(); + }else{ + init_curses(); + routine(); + // exit curses + endwin(); + } + exit (0); + }catch(...){ + perror("Error in segget.cpp main()"); } - return 0; + exit(1); }
\ No newline at end of file diff --git a/segget/segget.h b/segget/segget.h index 2a2b76f..c03e6ea 100644 --- a/segget/segget.h +++ b/segget/segget.h @@ -35,6 +35,11 @@ #include <json/json.h> #include <ncurses.h> #include <pthread.h> +#include <sys/time.h> +#include <sys/resource.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> #include "checksum.h" #include "config.h" #include "distfile.h" @@ -55,6 +60,9 @@ using namespace std; CURLM *cm; +int routine(); +void start_daemon_mode(); +int init_curses(); int load_pkgs(); void show_pkgs(); int pkg_choose_segment(Tpkg * cur_pkg, uint connection_num); |