--- rxvt-unicode/src/rxvtc.C 2006/01/22 16:36:40 1.22 +++ rxvt-unicode/src/rxvtc.C 2006/08/09 22:01:07 1.26 @@ -1,9 +1,9 @@ -/*--------------------------------*-C-*---------------------------------* +/*----------------------------------------------------------------------* * File: rxvtc.C *----------------------------------------------------------------------* * * All portions of code are copyright by their respective author/s. - * Copyright (c) 2003-2004 Marc Lehmann + * Copyright (c) 2003-2006 Marc Lehmann * * 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 @@ -35,20 +35,31 @@ #include "rxvt.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 (EXIT_FAILURE); + exit (STATUS_FAILURE); } - char *sockname = rxvt_connection::unix_sockname (); - sockaddr_un sa; sa.sun_family = AF_UNIX; strcpy (sa.sun_path, sockname); free (sockname); @@ -56,7 +67,7 @@ if (connect (fd, (sockaddr *)&sa, sizeof (sa))) { perror ("unable to connect to the rxvt-unicode daemon"); - exit (EXIT_FAILURE); + exit (STATUS_CONNECTION_FAILED); } } @@ -82,7 +93,7 @@ if (!getcwd (cwd, sizeof (cwd))) { perror ("unable to determine current working directory"); - exit (EXIT_FAILURE); + exit (STATUS_FAILURE); } c.send ("CWD"), c.send (cwd); @@ -115,7 +126,7 @@ if (!ptytty::send_fd (c.fd, cint)) { fprintf (stderr, "unable to send fd %d: ", cint); perror (0); - exit (EXIT_FAILURE); + exit (STATUS_FAILURE); } } else if (!strcmp (tok, "END")) @@ -123,7 +134,7 @@ int success; if (c.recv (success)) - exit (success ? EXIT_SUCCESS : EXIT_FAILURE); + exit (success ? STATUS_SUCCESS : STATUS_FAILURE); } else { @@ -131,6 +142,6 @@ break; } - return EXIT_FAILURE; + return STATUS_FAILURE; }