source: neercs/trunk/src/attach.c @ 2481

Last change on this file since 2481 was 2481, checked in by Pascal Terjan, 12 years ago
  • Improve the speed of current attach (even if it will die)
  • Fix some messages
  • Property svn:eol-style set to native
File size: 5.4 KB
Line 
1#include <errno.h>
2#include <fcntl.h>
3#include <glob.h>
4#include <limits.h>
5#include <stdio.h>
6#include <stdlib.h>
7#include <unistd.h>
8
9#include <sys/socket.h>
10#include <sys/un.h>
11#include <sys/types.h>
12#include <sys/stat.h>
13
14#include "neercs.h"
15
16static char * build_socket_path(char *socket_dir)
17{
18    char *path, *dir;
19    path = (char *)malloc(PATH_MAX+1);
20    dir = socket_dir;
21    if(!dir)
22        dir = getenv("NEERCSDIR");
23    if(!dir)
24        dir = getenv("TMPDIR");
25    if(!dir)
26        dir = "/tmp";
27    if(path)
28        snprintf(path, PATH_MAX+1, "%s/neercs.%d.sock", dir, getpid());
29    return path;
30}
31
32int create_socket(struct screen_list* screen_list)
33{
34    int sock;
35    struct sockaddr_un myaddr;
36
37    sock = socket(AF_UNIX, SOCK_DGRAM, 0);
38
39    if(sock < 0)
40    {
41        perror("create_socket:socket");
42        return errno;
43    }
44
45    memset(&myaddr, 0, sizeof(struct sockaddr_un));
46
47    myaddr.sun_family = AF_UNIX;
48    screen_list->socket_path = build_socket_path(screen_list->socket_path);
49    strncpy(myaddr.sun_path, screen_list->socket_path, sizeof(myaddr.sun_path) - 1);
50    unlink(myaddr.sun_path);
51
52    if(bind(sock, (struct sockaddr *)&myaddr, sizeof(struct sockaddr_un)) < 0)
53    {
54        free(screen_list->socket_path);
55        screen_list->socket_path = NULL;
56        close(sock);
57        perror("create_socket:bind");
58        return errno;
59    }
60    fcntl(sock, F_SETFL, O_NONBLOCK);
61
62    debug("Listening on %s (%d)", screen_list->socket_path, sock);
63
64    screen_list->socket = sock;
65
66    return 0;
67}
68
69int detach(struct screen_list* screen_list, caca_display_t * dp)
70{
71    int fd;
72    int pid;
73
74    screen_list->attached = 0;
75    caca_free_display(dp);
76
77    close(0);
78    close(1);
79    close(2);
80
81    fd = open("/dev/null", O_RDWR|O_CREAT, S_IRUSR|S_IWUSR);
82    dup2(fd, 0);
83    if(fd != 0)
84        close(fd);
85    fd = open("/tmp/log.txt", O_RDWR|O_CREAT, S_IRUSR|S_IWUSR);
86    dup2(fd, 1);
87    dup2(fd, 2);
88    if (fd > 2)
89        close(fd);
90
91    if(!screen_list->socket)
92    {
93        pid = fork();
94        if (pid < 0)
95            return 1;
96        if (pid > 0)
97            exit(0);
98
99        create_socket(screen_list);
100        setsid();
101    }
102
103    return 0;
104}
105
106/* FIXME Is this really not defined anywhere ? */
107#define MIN(X,Y) ((X) < (Y) ? (X) : (Y))
108
109static int handle_command(char * cmd, struct screen_list* screen_list, cucul_canvas_t * cv, caca_display_t ** dp)
110{
111    int refresh = 0;
112    debug("Received command %s", cmd);
113    if(!strncmp("ENV ", cmd, MIN(4, strlen(cmd))))
114    {
115        putenv(strdup(cmd+4));
116    }
117    else if(!strncmp("ATTACH ", cmd, MIN(7, strlen(cmd))))
118    {
119        int fd;
120        screen_list->attached = 1;
121        close(screen_list->socket);
122        screen_list->socket= 0;
123        unlink(screen_list->socket_path);
124        screen_list->socket_path = NULL;
125        close(0);
126        fd=open(cmd+7, O_RDWR);
127        dup2(fd, 0);
128        close(1);
129        fd=open(cmd+7, O_RDWR);
130        dup2(fd, 1);
131        close(2);
132        fd=open(cmd+7, O_RDWR);
133        dup2(fd, 2);
134        *dp = caca_create_display(cv);
135        if(*dp)
136        {
137            caca_set_cursor(*dp, 1);
138            refresh = 1;
139        }
140    }
141    else
142    {
143        fprintf(stderr, "Unknown command received: %s\n", cmd);
144    }
145    return refresh;
146}
147
148int read_socket(struct screen_list* screen_list, cucul_canvas_t * cv, caca_display_t ** dp) {
149    ssize_t n;
150    char buf[4097];
151    int refresh = 0;
152
153    if(!screen_list->socket)
154        return 0;
155
156    while (screen_list->socket && (n = read(screen_list->socket, buf, sizeof(buf)-1)) > 0)
157    {
158        buf[n] = 0;
159        refresh |= handle_command(buf, screen_list, cv, dp);
160    }
161    return refresh;
162}
163
164char ** list_sockets(char *socket_dir)
165{
166    char *pattern, *dir;
167    glob_t globbuf;
168
169    globbuf.gl_pathv = NULL;
170
171    pattern = (char *)malloc(PATH_MAX+1);
172
173    dir = socket_dir;
174    if(!dir)
175        dir = getenv("NEERCSDIR");
176    if(!dir)
177        dir = getenv("TMPDIR");
178    if(!dir)
179        dir = "/tmp";
180
181    if(!pattern)
182        return globbuf.gl_pathv;
183
184    snprintf(pattern, PATH_MAX, "%s/neercs.*.sock", dir);
185    pattern[PATH_MAX] = '\0';
186
187    debug("Looking for sockets in the form %s", pattern);
188
189    glob(pattern, 0, NULL, &globbuf);
190
191    return globbuf.gl_pathv;
192}
193
194
195/* FIXME Is this really not defined anywhere ? */
196extern char **environ;
197
198int request_attach(char *socket_path)
199{
200    int sock;
201    char fdstr[PATH_MAX], lnk[PATH_MAX];
202    char buf[4096];
203    struct sockaddr_un addr;
204
205    debug("Connecting to %s", socket_path);
206
207    /* Open the socket */
208    if ((sock = socket(AF_UNIX, SOCK_DGRAM, 0)) < 0) {
209        perror("request_attach:socket");
210        return 1;
211    }
212
213    memset(&addr,0,sizeof(addr));
214    addr.sun_family = AF_UNIX;
215    strcpy(addr.sun_path,socket_path);
216    if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0)
217    {
218        printf("Failed to connect to %s: %s\n", socket_path, strerror(errno));
219        return 1;
220    }
221
222    debug("Sending environment", 0);
223
224    /* Send our environment */
225#define sendvar(var) if(getenv(var)) {snprintf(buf, 4096, "ENV %s=%s", var, getenv(var)); write(sock, buf, strlen(buf));}
226    sendvar("DISPLAY")
227    sendvar("CACA_DRIVER")
228
229    snprintf(fdstr, sizeof(fdstr), "/proc/%d/fd/0", getpid());
230    readlink(fdstr, lnk, sizeof(lnk)-1);
231    lnk[PATH_MAX-1] = '\0';
232
233    close(0);
234
235    debug("Requesting attach to %s", lnk);
236
237    /* Request attaching */
238    snprintf(buf, 4096, "ATTACH %s", lnk);
239    write(sock, buf, strlen(buf));
240    close(sock);
241
242    debug("Exiting", 0);
243    exit(0);
244}
Note: See TracBrowser for help on using the repository browser.