git-completion.bash: pass $__git_subcommand_idx from __git_main()
[git] / unix-socket.c
1 #include "cache.h"
2 #include "unix-socket.h"
3
4 static int unix_stream_socket(void)
5 {
6         int fd = socket(AF_UNIX, SOCK_STREAM, 0);
7         if (fd < 0)
8                 die_errno("unable to create socket");
9         return fd;
10 }
11
12 static int chdir_len(const char *orig, int len)
13 {
14         char *path = xmemdupz(orig, len);
15         int r = chdir(path);
16         free(path);
17         return r;
18 }
19
20 struct unix_sockaddr_context {
21         char *orig_dir;
22 };
23
24 static void unix_sockaddr_cleanup(struct unix_sockaddr_context *ctx)
25 {
26         if (!ctx->orig_dir)
27                 return;
28         /*
29          * If we fail, we can't just return an error, since we have
30          * moved the cwd of the whole process, which could confuse calling
31          * code.  We are better off to just die.
32          */
33         if (chdir(ctx->orig_dir) < 0)
34                 die("unable to restore original working directory");
35         free(ctx->orig_dir);
36 }
37
38 static int unix_sockaddr_init(struct sockaddr_un *sa, const char *path,
39                               struct unix_sockaddr_context *ctx)
40 {
41         int size = strlen(path) + 1;
42
43         ctx->orig_dir = NULL;
44         if (size > sizeof(sa->sun_path)) {
45                 const char *slash = find_last_dir_sep(path);
46                 const char *dir;
47                 struct strbuf cwd = STRBUF_INIT;
48
49                 if (!slash) {
50                         errno = ENAMETOOLONG;
51                         return -1;
52                 }
53
54                 dir = path;
55                 path = slash + 1;
56                 size = strlen(path) + 1;
57                 if (size > sizeof(sa->sun_path)) {
58                         errno = ENAMETOOLONG;
59                         return -1;
60                 }
61                 if (strbuf_getcwd(&cwd))
62                         return -1;
63                 ctx->orig_dir = strbuf_detach(&cwd, NULL);
64                 if (chdir_len(dir, slash - dir) < 0)
65                         return -1;
66         }
67
68         memset(sa, 0, sizeof(*sa));
69         sa->sun_family = AF_UNIX;
70         memcpy(sa->sun_path, path, size);
71         return 0;
72 }
73
74 int unix_stream_connect(const char *path)
75 {
76         int fd, saved_errno;
77         struct sockaddr_un sa;
78         struct unix_sockaddr_context ctx;
79
80         if (unix_sockaddr_init(&sa, path, &ctx) < 0)
81                 return -1;
82         fd = unix_stream_socket();
83         if (connect(fd, (struct sockaddr *)&sa, sizeof(sa)) < 0)
84                 goto fail;
85         unix_sockaddr_cleanup(&ctx);
86         return fd;
87
88 fail:
89         saved_errno = errno;
90         unix_sockaddr_cleanup(&ctx);
91         close(fd);
92         errno = saved_errno;
93         return -1;
94 }
95
96 int unix_stream_listen(const char *path)
97 {
98         int fd, saved_errno;
99         struct sockaddr_un sa;
100         struct unix_sockaddr_context ctx;
101
102         unlink(path);
103
104         if (unix_sockaddr_init(&sa, path, &ctx) < 0)
105                 return -1;
106         fd = unix_stream_socket();
107
108         if (bind(fd, (struct sockaddr *)&sa, sizeof(sa)) < 0)
109                 goto fail;
110
111         if (listen(fd, 5) < 0)
112                 goto fail;
113
114         unix_sockaddr_cleanup(&ctx);
115         return fd;
116
117 fail:
118         saved_errno = errno;
119         unix_sockaddr_cleanup(&ctx);
120         close(fd);
121         errno = saved_errno;
122         return -1;
123 }