Move some common code into a function.

1 files changed, 19 insertions, 22 deletions

diff --git a/client.c b/client.c
index 60fef91a..afa48048 100644
--- a/client.c
+++ b/client.c
@@ -41,6 +41,7 @@ const char     *client_exitmsg;
 void		client_send_identify(int);
 void		client_send_environ(void);
 void		client_write_server(enum msgtype, void *, size_t);
+void		client_update_event(void);
 void		client_signal(int, short, void *);
 void		client_callback(int, short, void *);
 int		client_dispatch(void);
@@ -154,12 +155,24 @@ client_write_server(enum msgtype type, void *buf, size_t len)
  	imsg_compose(&client_ibuf, type, PROTOCOL_VERSION, -1, -1, buf, len);
 }
 
+void
+client_update_event(void)
+{
+	short	events;
+
+	event_del(&client_event);
+	events = EV_READ;
+	if (client_ibuf.w.queued > 0)
+		events |= EV_WRITE;
+	event_set(&client_event, client_ibuf.fd, events, client_callback, NULL);
+	event_add(&client_event, NULL);
+}
+
 __dead void
 client_main(void)
 {
 	struct event		ev_sigcont, ev_sigterm, ev_sigwinch;
 	struct sigaction	sigact;
-	short	 		events;
 
 	logfile("client");
 
@@ -197,13 +210,8 @@ client_main(void)
 	if (client_dispatch() != 0)
 		goto out;
 
-	/* Set up the client-server socket event. */
-	events = EV_READ;
-	if (client_ibuf.w.queued > 0)
-		events |= EV_WRITE;
-	event_set(&client_event, client_ibuf.fd, events, client_callback, NULL);
-	event_add(&client_event, NULL);
-	
+	/* Set the event and dispatch. */
+	client_update_event();	
 	event_dispatch();
 
 out:
@@ -217,7 +225,7 @@ out:
 }
 
 void
-client_signal(int sig, short events, unused void *data)
+client_signal(int sig, unused short events, unused void *data)
 {
 	struct sigaction	sigact;
 
@@ -240,12 +248,7 @@ client_signal(int sig, short events, unused void *data)
 		break;
 	}
 
-	event_del(&client_event);
-	events = EV_READ;
-	if (client_ibuf.w.queued > 0)
-		events |= EV_WRITE;
-	event_set(&client_event, client_ibuf.fd, events, client_callback, NULL);
-	event_add(&client_event, NULL);
+	client_update_event();
 }
 
 void
@@ -267,13 +270,7 @@ client_callback(unused int fd, short events, unused void *data)
 			goto lost_server;
 	}
 
-	event_del(&client_event);
-	events = EV_READ;
-	if (client_ibuf.w.queued > 0)
-		events |= EV_WRITE;
-	event_set(&client_event, client_ibuf.fd, events, client_callback, NULL);
-	event_add(&client_event, NULL);
-
+	client_update_event();
 	return;
 
 lost_server: