Return to signal.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / bmon / src

Annotation of embedaddon/bmon/src/signal.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * signal.c           Signal Handling
                      3:  *
                      4:  * Copyright (c) 2001-2005 Thomas Graf <tgraf@suug.ch>
                      5:  *
                      6:  * Permission is hereby granted, free of charge, to any person obtaining a
                      7:  * copy of this software and associated documentation files (the "Software"),
                      8:  * to deal in the Software without restriction, including without limitation
                      9:  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
                     10:  * and/or sell copies of the Software, and to permit persons to whom the
                     11:  * Software is furnished to do so, subject to the following conditions:
                     12:  *
                     13:  * The above copyright notice and this permission notice shall be included
                     14:  * in all copies or substantial portions of the Software.
                     15:  *
                     16:  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
                     17:  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
                     18:  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
                     19:  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
                     20:  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
                     21:  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
                     22:  * DEALINGS IN THE SOFTWARE.
                     23:  */
                     24: 
                     25: #include <bmon/bmon.h>
                     26: #include <bmon/conf.h>
                     27: #include <bmon/utils.h>
                     28: 
                     29: static int sig_state = 0;
                     30: 
                     31: static RETSIGTYPE sig_usr1(int unused)
                     32: {
                     33:        sig_state = 1;
                     34:        signal(SIGUSR1, sig_usr1);
                     35: }
                     36: 
                     37: int is_signal_recvd(void)
                     38: {
                     39:        int ret = sig_state;
                     40:        sig_state = 0;
                     41:        return ret;
                     42: }
                     43: 
                     44: #if defined PRAGMA_FALLBACK
                     45: #pragma init (init_signal)
                     46: #endif
                     47: 
                     48: static void __init init_signal(void)
                     49: {
                     50:        if (signal(SIGUSR1, sig_usr1) == SIG_ERR) {
                     51:                perror("signal(SIGUSR1) failed");
                     52:                exit(1);
                     53:        }
                     54: }
                     55: 
                     56: void send_signal(const char *arg)
                     57: {
                     58:        if (*arg != '-') {
                     59:                long pid = strtol(arg, NULL, 10);
                     60:                
                     61:                if (pid != LONG_MIN && pid != LONG_MAX) {
                     62:                        if (kill((int) pid, SIGUSR1) < 0)
                     63:                                quit("kill(%d, SIGUSR1) failed: %s\n",
                     64:                                        (int) pid, strerror(errno));
                     65:                } else
                     66:                        quit("Invalid pid.\n");
                     67:        } else {
                     68:                FILE *f;
                     69:                char buf[256], hdr[256], bmon_procs[10][256];
                     70:                int cnt = 0, pid_to_kill = -1;
                     71:                int my_uid = getuid();
                     72:                char cmd[64];
                     73:                
                     74:                memset(cmd, 0, sizeof(cmd));
                     75:                snprintf(cmd, sizeof(cmd)-1, "ps -U %d -o pid,uid,args", my_uid);
                     76:                
                     77:                if (!(f = popen(cmd, "r"))) {
                     78:                        fprintf(stderr, "popen(%s: %s\n", cmd, strerror(errno));
                     79:                        quit("Your ps is probably not POSIX compatible, use kill\n");
                     80:                }
                     81: 
                     82:                fgets(hdr, sizeof(hdr), f);
                     83:                
                     84:                while(fgets(buf, sizeof(buf), f)) {
                     85:                        int pid, uid, n;
                     86:                        char args[256], cmd[256];
                     87: 
                     88:                        if (!strstr(buf, "bmon") || !strstr(buf, "-w") || strstr(buf, "-S"))
                     89:                                continue;
                     90: 
                     91:                        n = sscanf(buf, "%d %d %s %[^\n\r]s", &pid, &uid, cmd, args);
                     92:                        
                     93:                        if (n != 4)
                     94:                                continue;
                     95:                        
                     96:                        if (uid == my_uid) {
                     97:                                if (cnt <= 9) {
                     98:                                        snprintf(bmon_procs[cnt], sizeof(bmon_procs[cnt])-1,
                     99:                                                "%5d %5d %s %s", pid, uid, cmd, args);
                    100:                                        pid_to_kill = pid;
                    101:                                        cnt++;
                    102:                                } else
                    103:                                        break;
                    104:                        }
                    105:                }
                    106:                
                    107:                pclose(f);
                    108:                
                    109:                if (cnt == 1) {
                    110:                        if (pid_to_kill >= 0)
                    111:                                if (kill((int) pid_to_kill, SIGUSR1) < 0)
                    112:                                        quit("kill(%d, SIGUSR1) failed: %s\n",
                    113:                                                (int) pid_to_kill, strerror(errno));
                    114:                } else {
                    115:                        int n;
                    116:                        
                    117:                        fprintf(stderr, "%d instances of bmon are running, " \
                    118:                                "use bmon -S <pid>\n", cnt);
                    119:                        
                    120:                        printf("%s", hdr);
                    121:                        for (n=0; n < cnt; n++)
                    122:                                printf("%s\n", bmon_procs[n]);
                    123:                }
                    124:        }
                    125: }