rxvt-unicode/src/rxvtc.C

/*----------------------------------------------------------------------*
 * File:        rxvtc.C
 *----------------------------------------------------------------------*
 *
 * All portions of code are copyright by their respective author/s.
 * Copyright (c) 2003-2006 Marc Lehmann <schmorp@schmorp.de>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *----------------------------------------------------------------------*/

#include "../config.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>

#include <unistd.h>
#include <sys/socket.h>
#include <sys/un.h>

#include "rxvtdaemon.h"
#include "libptytty.h"

#define STATUS_SUCCESS           0
#define STATUS_FAILURE           1
#define STATUS_CONNECTION_FAILED 2

struct client : rxvt_connection
{
  client ();
};

client::client ()
{
  sockaddr_un sa;
  char *sockname = rxvt_connection::unix_sockname ();

  if (strlen (sockname) >= sizeof (sa.sun_path))
    {
      fputs ("socket name too long, aborting.\n", stderr);
      exit (STATUS_FAILURE);
    }

  if ((fd = socket (AF_UNIX, SOCK_STREAM, 0)) < 0)
    {
      perror ("unable to create communications socket");
      exit (STATUS_FAILURE);
    }

  sa.sun_family = AF_UNIX;
  strcpy (sa.sun_path, sockname);
  free (sockname);

  if (connect (fd, (sockaddr *)&sa, sizeof (sa)))
    {
      perror ("unable to connect to the rxvt-unicode daemon");
      exit (STATUS_CONNECTION_FAILED);
    }
}

extern char **environ;

int
main (int argc, const char *const *argv)
{
  // instead of getcwd we could opendir (".") and pass the fd for fchdir *g*
  char cwd[PATH_MAX];

  if (!getcwd (cwd, sizeof (cwd)))
    {
      perror ("unable to determine current working directory");
      exit (STATUS_FAILURE);
    }

  client c;

  {
    sigset_t ss;

    sigemptyset (&ss);
    sigaddset (&ss, SIGHUP);
    sigaddset (&ss, SIGPIPE);
    sigprocmask (SIG_BLOCK, &ss, 0);
  }

  if (argc >= 2 && !strcmp (argv[1], "-k"))
    {
      c.send ("QUIT");
      return 0;
    }

  c.send ("NEW");

  for (char **var = environ; *var; var++)
    c.send ("ENV"), c.send (*var);

  const char *base = strrchr (argv[0], '/');
  base = base ? base + 1 : argv[0];
  c.send ("ARG"), c.send (strcmp (base, RXVTNAME "c") ? base : RXVTNAME);

  c.send ("ARG"), c.send ("-cd");
  c.send ("ARG"), c.send (cwd);

  for (int i = 1; i < argc; i++)
    c.send ("ARG"), c.send (argv[i]);

  c.send ("END");

  auto_str tok;
  int cint;

  for (;;)
    if (!c.recv (tok))
      {
        fprintf (stderr, "protocol error: unexpected eof from server.\n");
        break;
      }
    else if (!strcmp (tok, "MSG") && c.recv (tok))
      fprintf (stderr, "%s", (const char *)tok);
    else if (!strcmp (tok, "GETFD") && c.recv (cint))
      {
        if (!ptytty::send_fd (c.fd, cint))
          {
            fprintf (stderr, "unable to send fd %d: ", cint); perror (0);
            exit (STATUS_FAILURE);
          }
      }
    else if (!strcmp (tok, "END"))
      {
        int success;

        if (c.recv (success))
          exit (success ? STATUS_SUCCESS : STATUS_FAILURE);
      }
    else
      {
        fprintf (stderr, "protocol error: received unsupported token '%s'.\n", (const char *)tok);
        break;
      }

  return STATUS_FAILURE;
}

Revision:	1.37
Committed:	Mon Oct 6 11:14:49 2014 UTC (9 years, 8 months ago) by sf-exg
Content type:	text/plain
Branch:	MAIN
CVS Tags:	rxvt-unicode-rel-9_26, rxvt-unicode-rel-9_25, rxvt-unicode-rel-9_22, rxvt-unicode-rel-9_21
Changes since 1.36:	+6 -0 lines
Log Message:	Add -k option to urxvtc for killing the daemon process.
#	User	Rev	Content
1	root	1.25	/----------------------------------------------------------------------
2	pcg	1.11	* File: rxvtc.C
3			----------------------------------------------------------------------
4			*
5			* All portions of code are copyright by their respective author/s.
6	root	1.33	* Copyright (c) 2003-2006 Marc Lehmann <schmorp@schmorp.de>
7	pcg	1.11	*
8			* This program is free software; you can redistribute it and/or modify
9			* it under the terms of the GNU General Public License as published by
10	root	1.36	* the Free Software Foundation; either version 3 of the License, or
11	pcg	1.11	* (at your option) any later version.
12			*
13			* This program is distributed in the hope that it will be useful,
14			* but WITHOUT ANY WARRANTY; without even the implied warranty of
15			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16			* GNU General Public License for more details.
17			*
18			* You should have received a copy of the GNU General Public License
19			* along with this program; if not, write to the Free Software
20			* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21			----------------------------------------------------------------------/
22
23	pcg	1.8	#include "../config.h"
24	sf-exg	1.35	#include <stdio.h>
25			#include <stdlib.h>
26			#include <string.h>
27			#include <signal.h>
28	pcg	1.1
29			#include <unistd.h>
30			#include <sys/socket.h>
31			#include <sys/un.h>
32
33	root	1.21	#include "rxvtdaemon.h"
34			#include "libptytty.h"
35
36	root	1.23	#define STATUS_SUCCESS 0
37			#define STATUS_FAILURE 1
38			#define STATUS_CONNECTION_FAILED 2
39
40	root	1.29	struct client : rxvt_connection
41			{
42	pcg	1.1	client ();
43			};
44
45			client::client ()
46			{
47	root	1.26	sockaddr_un sa;
48			char *sockname = rxvt_connection::unix_sockname ();
49	ayin	1.28
50	ayin	1.31	if (strlen (sockname) >= sizeof (sa.sun_path))
51	root	1.26	{
52			fputs ("socket name too long, aborting.\n", stderr);
53			exit (STATUS_FAILURE);
54			}
55	ayin	1.28
56	root	1.20	if ((fd = socket (AF_UNIX, SOCK_STREAM, 0)) < 0)
57	pcg	1.1	{
58	pcg	1.5	perror ("unable to create communications socket");
59	root	1.23	exit (STATUS_FAILURE);
60	pcg	1.1	}
61
62			sa.sun_family = AF_UNIX;
63	pcg	1.4	strcpy (sa.sun_path, sockname);
64			free (sockname);
65	pcg	1.1
66			if (connect (fd, (sockaddr *)&sa, sizeof (sa)))
67			{
68	root	1.13	perror ("unable to connect to the rxvt-unicode daemon");
69	root	1.23	exit (STATUS_CONNECTION_FAILED);
70	pcg	1.1	}
71			}
72
73	pcg	1.3	extern char **environ;
74
75	pcg	1.1	int
76	pcg	1.7	main (int argc, const char const argv)
77	pcg	1.1	{
78	root	1.30	// instead of getcwd we could opendir (".") and pass the fd for fchdir g
79			char cwd[PATH_MAX];
80
81			if (!getcwd (cwd, sizeof (cwd)))
82			{
83			perror ("unable to determine current working directory");
84			exit (STATUS_FAILURE);
85			}
86
87	pcg	1.1	client c;
88	root	1.30
89	pcg	1.3	{
90			sigset_t ss;
91
92	pcg	1.6	sigemptyset (&ss);
93	pcg	1.3	sigaddset (&ss, SIGHUP);
94	root	1.27	sigaddset (&ss, SIGPIPE);
95	pcg	1.3	sigprocmask (SIG_BLOCK, &ss, 0);
96			}
97
98	sf-exg	1.37	if (argc >= 2 && !strcmp (argv[1], "-k"))
99			{
100			c.send ("QUIT");
101			return 0;
102			}
103
104	pcg	1.1	c.send ("NEW");
105	pcg	1.3
106	root	1.16	for (char *var = environ; var; var++)
107			c.send ("ENV"), c.send (*var);
108	pcg	1.1
109	root	1.13	const char *base = strrchr (argv[0], '/');
110			base = base ? base + 1 : argv[0];
111			c.send ("ARG"), c.send (strcmp (base, RXVTNAME "c") ? base : RXVTNAME);
112
113	root	1.32	c.send ("ARG"), c.send ("-cd");
114			c.send ("ARG"), c.send (cwd);
115
116	root	1.13	for (int i = 1; i < argc; i++)
117	pcg	1.2	c.send ("ARG"), c.send (argv[i]);
118
119	pcg	1.1	c.send ("END");
120	pcg	1.8
121			auto_str tok;
122	root	1.17	int cint;
123	pcg	1.8
124			for (;;)
125			if (!c.recv (tok))
126			{
127			fprintf (stderr, "protocol error: unexpected eof from server.\n");
128			break;
129			}
130			else if (!strcmp (tok, "MSG") && c.recv (tok))
131			fprintf (stderr, "%s", (const char *)tok);
132	root	1.17	else if (!strcmp (tok, "GETFD") && c.recv (cint))
133			{
134	root	1.22	if (!ptytty::send_fd (c.fd, cint))
135	root	1.17	{
136			fprintf (stderr, "unable to send fd %d: ", cint); perror (0);
137	root	1.23	exit (STATUS_FAILURE);
138	root	1.17	}
139			}
140	pcg	1.8	else if (!strcmp (tok, "END"))
141			{
142			int success;
143	root	1.19
144	pcg	1.8	if (c.recv (success))
145	root	1.23	exit (success ? STATUS_SUCCESS : STATUS_FAILURE);
146	pcg	1.8	}
147			else
148			{
149	root	1.17	fprintf (stderr, "protocol error: received unsupported token '%s'.\n", (const char *)tok);
150	pcg	1.8	break;
151			}
152
153	root	1.23	return STATUS_FAILURE;
154	pcg	1.1	}
155