Sync OpenBSD patchset 1043:

Use a lock file and flock() to serialize server start, avoids problems
when running a bunch of tmux from cron at the same time. Based on a diff
from Tim Ruehsen.
This commit is contained in:
Tiago Cunha 2012-03-18 01:28:10 +00:00
parent 9b8fc982ec
commit 40b1d64ce7
3 changed files with 46 additions and 15 deletions

View File

@ -39,6 +39,7 @@ int client_exitval;
enum msgtype client_exittype;
int client_attached;
int client_get_lock(char *);
int client_connect(char *, int);
void client_send_identify(int);
void client_send_environ(void);
@ -49,13 +50,38 @@ void client_callback(int, short, void *);
int client_dispatch_attached(void);
int client_dispatch_wait(void *);
/*
* Get server create lock. If already held then server start is happening in
* another client, so block until the lock is released and return -1 to
* retry. Ignore other errors - just continue and start the server without the
* lock.
*/
int
client_get_lock(char *lockfile)
{
int lockfd;
if ((lockfd = open(lockfile, O_WRONLY|O_CREAT, 0600)) == -1)
fatal("open failed");
if (flock(lockfd, LOCK_EX|LOCK_NB) == -1 && errno == EWOULDBLOCK) {
while (flock(lockfd, LOCK_EX) == -1 && errno == EINTR)
/* nothing */;
close(lockfd);
return (-1);
}
return (lockfd);
}
/* Connect client to server. */
int
client_connect(char *path, int start_server)
{
struct sockaddr_un sa;
size_t size;
int fd;
int fd, lockfd;
char *lockfile;
memset(&sa, 0, sizeof sa);
sa.sun_family = AF_UNIX;
@ -65,24 +91,25 @@ client_connect(char *path, int start_server)
return (-1);
}
retry:
if ((fd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
fatal("socket failed");
if (connect(fd, (struct sockaddr *) &sa, SUN_LEN(&sa)) == -1) {
if (errno != ECONNREFUSED && errno != ENOENT)
goto failed;
if (!start_server)
goto failed;
switch (errno) {
case ECONNREFUSED:
if (unlink(path) != 0)
goto failed;
/* FALLTHROUGH */
case ENOENT:
if ((fd = server_start()) == -1)
goto failed;
break;
default:
goto failed;
}
close(fd);
xasprintf(&lockfile, "%s.lock", path);
if ((lockfd = client_get_lock(lockfile)) == -1)
goto retry;
if (unlink(path) != 0 && errno != ENOENT)
return (-1);
fd = server_start(lockfd, lockfile);
xfree(lockfile);
close(lockfd);
}
setblocking(fd, 0);

View File

@ -103,7 +103,7 @@ server_create_socket(void)
/* Fork new server. */
int
server_start(void)
server_start(int lockfd, char *lockfile)
{
struct window_pane *wp;
int pair[2];
@ -162,6 +162,10 @@ server_start(void)
server_fd = server_create_socket();
server_client_create(pair[1]);
unlink(lockfile);
xfree(lockfile);
close(lockfd);
if (access(SYSTEM_CFG, R_OK) == 0)
load_cfg(SYSTEM_CFG, NULL, &cfg_causes);
else if (errno != ENOENT) {

2
tmux.h
View File

@ -1683,7 +1683,7 @@ const char *key_string_lookup_key(int);
extern struct clients clients;
extern struct clients dead_clients;
extern struct paste_stack global_buffers;
int server_start(void);
int server_start(int, char *);
void server_update_socket(void);
/* server-client.c */