Changeset 2488 for neercs/trunk/src/main.c

Timestamp:

06/25/08 00:33:40 (5 years ago)

Author:

pterjan

Message:

• Have -r/-R/-S like screen

File:

• neercs/trunk/src/main.c (modified) (9 diffs)

neercs/trunk/src/main.c

@@ -68 +68 @@
     printf("Example : %s zsh top \n\n", argv[0]);
     printf("Options :\n");
-    printf("\t--config\t\t-c <file>\t\tuse given config file\n");
-    printf("\t--pid\t\t-P <pid>\t\tattach <pid>\n");
-    printf("\t\t\t-R\t\t\tre-attach another neercs\n");
+    printf("\t--config\t-c <file>\t\tuse given config file\n");
+    printf("\t--pid\t\t-P <pid>\t\tgrab <pid>\n");
+    printf("\t\t\t-r [session]\t\treattach to a detached neercs\n");
+    printf("\t\t\t-R [session]\t\treattach if possible, otherwise start a new session\n");
+    printf("\t\t\t-S <name>\t\tname this session <name> instead of <pid>\n");
     printf("\t--version\t-v \t\t\tdisplay version and exit\n");
     printf("\t--help\t\t-h \t\t\tthis help\n");
@@ -82 +84 @@
     struct recurrent_list *recurrent_list = NULL;
     struct passwd *user_info;
-    char *default_shell = NULL, *user_path = NULL;
+    char *default_shell = NULL, *user_path = NULL, *session_name = NULL;
     int i, w, h, args, s=0;
     int eof = 0, refresh = 1, command = 0;
@@ -88 +90 @@
     int lock_offset = 0;
     int mainret = 0;
-    int attach = 0;
+    int attach = 0, forceattach = 0;
     int *to_grab = NULL;
     int nb_to_grab = 0;
@@ -142 +144 @@
     screen_list->socket_dir = NULL;
     screen_list->socket_path = NULL;
+    screen_list->session_name = NULL;
     memset(screen_list->lockmsg, 0, 1024);
     memset(screen_list->lockpass, 0, 1024);
@@ -168 +171 @@
             };
 #if defined USE_GRAB
-        int c = mygetopt(argc, argv, "c:RP:hv", long_options, &option_index);
+        int c = mygetopt(argc, argv, "c:S:R::r::P:hv", long_options, &option_index);
 #else
-        int c = mygetopt(argc, argv, "c:Rhv", long_options, &option_index);
+        int c = mygetopt(argc, argv, "c:S:R::r::hv", long_options, &option_index);
 #endif
         if(c == -1)
@@ -181 +184 @@
                 free(user_path);
             user_path = strdup(myoptarg);
+            s+=2;
+            break;
+        case 'S':
+            if(screen_list->session_name)
+                free(screen_list->session_name);
+            screen_list->session_name = strdup(myoptarg);
             s+=2;
             break;
@@ -205 +214 @@
             s+=2;
             break;
+        case 'r':
+            forceattach = 1;
         case 'R':
+            if(attach)
+            {
+                fprintf(stderr, "Attaching can only be requested once\n");
+                return -1;
+            }
+            if(myoptarg)
+            {
+                session_name = strdup(myoptarg);
+                s+=1;
+            }
             attach = 1;
             s+=1;
@@ -242 +263 @@
             return -1;
         }
-        sockets = list_sockets(screen_list->socket_dir);
+        sockets = list_sockets(screen_list->socket_dir, session_name);
+        if(!screen_list->session_name)
+            screen_list->session_name = session_name;
+        else
+            free(session_name);
         if(sockets && sockets[0])
         {
@@ -255 +280 @@
             fprintf(stderr, "No socket found!\n");
         }
-        return -1;
+        if(forceattach)
+            return -1;
+    }
+
+    /* Build default session name */
+    if(!screen_list->session_name)
+    {
+        char mypid[32]; /* FIXME Compute the length of PID_MAX ? */
+        snprintf(mypid, 31, "%d", getpid());
+        mypid[31]= '\0';
+        screen_list->session_name = strdup(mypid);
+        if(!screen_list->session_name)
+        {
+            fprintf(stderr, "Can't allocate memory at %s:%d\n", __FUNCTION__, __LINE__);
+            return -1;
+        }
     }