rxvt-unicode/src/rxvtdaemon.C

/*--------------------------------*-C-*---------------------------------*
 * File:        rxvtdaemon.C
 *----------------------------------------------------------------------*
 *
 * All portions of code are copyright by their respective author/s.
 * Copyright (c) 2003-2004 Marc Lehmann <pcg@goof.com>
 *
 * 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 2 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 <cstdlib>
#include <cstring>
#include <cstdio>

#include <inttypes.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/utsname.h>
#include <limits.h>

#include "rxvtdaemon.h"

// works around linux kernel bug, returns EAGAIN on a blocking socket
static ssize_t
read_ (int fd, void *buf, size_t count)
{
  int ret;

  do
    ret = read (fd, buf, count);
  while (ret < 0 && errno == EAGAIN);

  return ret;
}

char *rxvt_connection::unix_sockname ()
{
  char name[PATH_MAX];
  char *path = getenv ("RXVT_SOCKET");

  if (!path)
    {
      struct utsname u;
      uname (&u);

      path = getenv ("HOME");
      snprintf (name, PATH_MAX, "%s/.rxvt-unicode-%s",
                path ? path : "/tmp",
                u.nodename);

      path = name;
    }

  return strdup (path);
}

void rxvt_connection::send (const char *data, int len)
{
  uint8_t s[2];

  s[0] = len >> 8; s[1] = len;

  write (fd, s, 2);
  write (fd, data, len);
}

void rxvt_connection::send (const char *data)
{
  send (data, strlen (data));
}

bool rxvt_connection::recv (auto_str &data, int *len)
{
  uint8_t s[2];
  int l;

  if (read_ (fd, s, 2) != 2)
    return false;

  l = (s[0] << 8) + s[1];
  if (l > 4096)
    return false;

  if (len)
    *len = l;

  data = new char[l + 1];

  if (!data)
    return false;

  if (read_ (fd, data, l) != l)
    return false;

  data[l] = 0;

  return true;
}

void rxvt_connection::send (int data)
{
  uint8_t s[4];

  s[0] = data >> 24; s[1] = data >> 16; s[2] = data >> 8; s[3] = data;

  write (fd, s, 4);
}

bool rxvt_connection::recv (int &data)
{
  uint8_t s[4];

  if (read_ (fd, s, 4) != 4)
    return false;

  data = (((((s[0] << 8) | s[1]) << 8) | s[2]) << 8) | s[3];

  return true;
}


Revision:	1.9
Committed:	Sun Dec 25 15:30:29 2005 UTC (18 years, 5 months ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.8:	+18 -3 lines
Log Message:	* empty log message *
#	User	Rev	Content
1	pcg	1.6	/---------------------------------C----------------------------------
2			* File: rxvtdaemon.C
3			----------------------------------------------------------------------
4			*
5			* All portions of code are copyright by their respective author/s.
6			* Copyright (c) 2003-2004 Marc Lehmann <pcg@goof.com>
7			*
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			* the Free Software Foundation; either version 2 of the License, or
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.1	#include <cstdlib>
24			#include <cstring>
25	pcg	1.4	#include <cstdio>
26	pcg	1.1
27	root	1.7	#include <inttypes.h>
28	pcg	1.3	#include <unistd.h>
29	root	1.9	#include <errno.h>
30			#include <sys/types.h>
31	pcg	1.3	#include <sys/utsname.h>
32			#include <limits.h>
33
34	pcg	1.1	#include "rxvtdaemon.h"
35
36	root	1.9	// works around linux kernel bug, returns EAGAIN on a blocking socket
37			static ssize_t
38			read_ (int fd, void *buf, size_t count)
39			{
40			int ret;
41
42			do
43			ret = read (fd, buf, count);
44			while (ret < 0 && errno == EAGAIN);
45
46			return ret;
47			}
48
49	pcg	1.3	char *rxvt_connection::unix_sockname ()
50	pcg	1.1	{
51	pcg	1.3	char name[PATH_MAX];
52			char *path = getenv ("RXVT_SOCKET");
53
54			if (!path)
55			{
56			struct utsname u;
57			uname (&u);
58
59			path = getenv ("HOME");
60	root	1.8	snprintf (name, PATH_MAX, "%s/.rxvt-unicode-%s",
61	pcg	1.3	path ? path : "/tmp",
62			u.nodename);
63
64			path = name;
65			}
66
67			return strdup (path);
68	pcg	1.1	}
69
70			void rxvt_connection::send (const char *data, int len)
71			{
72			uint8_t s[2];
73
74			s[0] = len >> 8; s[1] = len;
75
76			write (fd, s, 2);
77			write (fd, data, len);
78			}
79
80			void rxvt_connection::send (const char *data)
81			{
82			send (data, strlen (data));
83			}
84
85	pcg	1.2	bool rxvt_connection::recv (auto_str &data, int *len)
86	pcg	1.1	{
87			uint8_t s[2];
88			int l;
89
90	root	1.9	if (read_ (fd, s, 2) != 2)
91	pcg	1.1	return false;
92
93			l = (s[0] << 8) + s[1];
94			if (l > 4096)
95			return false;
96
97			if (len)
98			*len = l;
99
100			data = new char[l + 1];
101
102			if (!data)
103			return false;
104
105	root	1.9	if (read_ (fd, data, l) != l)
106	pcg	1.1	return false;
107
108			data[l] = 0;
109
110			return true;
111			}
112
113			void rxvt_connection::send (int data)
114			{
115			uint8_t s[4];
116
117	pcg	1.5	s[0] = data >> 24; s[1] = data >> 16; s[2] = data >> 8; s[3] = data;
118	pcg	1.1
119			write (fd, s, 4);
120			}
121
122			bool rxvt_connection::recv (int &data)
123			{
124			uint8_t s[4];
125
126	root	1.9	if (read_ (fd, s, 4) != 4)
127	pcg	1.1	return false;
128
129			data = (((((s[0] << 8) \| s[1]) << 8) \| s[2]) << 8) \| s[3];
130
131			return true;
132			}
133
134
135