Rearrange code slightly to avoid util.c depending on main.c.
[rsync/rsync.git] / lib / compat.c
1 /* 
2    Copyright (C) Andrew Tridgell 1998
3    
4    This program is free software; you can redistribute it and/or modify
5    it under the terms of the GNU General Public License as published by
6    the Free Software Foundation; either version 2 of the License, or
7    (at your option) any later version.
8    
9    This program is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12    GNU General Public License for more details.
13    
14    You should have received a copy of the GNU General Public License
15    along with this program; if not, write to the Free Software
16    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17 */
18
19 /*
20   compatibility functions - replacing functions for platforms that don't
21   have them.
22
23   */
24 #include "rsync.h"
25
26
27 #ifndef HAVE_STRDUP
28  char *strdup(char *s)
29 {
30   int l = strlen(s) + 1;
31   char *ret = (char *)malloc(l);
32   if (ret)
33     strcpy(ret,s);
34   return ret;
35 }
36 #endif
37
38 #ifndef HAVE_GETCWD
39  char *getcwd(char *buf, int size)
40 {
41         return getwd(buf);
42 }
43 #endif
44
45
46 #ifndef HAVE_WAITPID
47  pid_t waitpid(pid_t pid, int *statptr, int options)
48 {
49 #ifdef HAVE_WAIT4
50         return wait4(pid, statptr, options, NULL);
51 #else
52         /* If wait4 is also not available, try wait3 for SVR3 variants */
53         /* Less ideal because can't actually request a specific pid */
54         /* At least the WNOHANG option is supported */
55         /* Code borrowed from apache fragment written by dwd@bell-labs.com */
56         int tmp_pid, dummystat;;
57         if (kill(pid, 0) == -1) {
58                 errno = ECHILD;
59                 return -1;
60         }
61         if (statptr == NULL)
62                 statptr = &dummystat;
63         while (((tmp_pid = wait3(statptr, options, 0)) != pid) &&
64                     (tmp_pid != -1) && (tmp_pid != 0) && (pid != -1))
65             ;
66         return tmp_pid;
67 #endif
68 }
69 #endif
70
71
72 #ifndef HAVE_MEMMOVE
73  void *memmove(void *dest, const void *src, size_t n)
74 {
75         bcopy((char *) src, (char *) dest, n);
76         return dest;
77 }
78 #endif
79
80 #ifndef HAVE_STRPBRK
81 /* Find the first ocurrence in S of any character in ACCEPT.  
82    derived from glibc 
83 */
84  char *strpbrk(const char *s, const char *accept)
85 {
86         while (*s != '\0')  {
87                 const char *a = accept;
88                 while (*a != '\0') {
89                         if (*a++ == *s) return (char *)s;
90                 }
91                 ++s;
92         }
93
94         return NULL;
95 }
96 #endif
97
98
99 #ifndef HAVE_STRLCPY
100 /* Like strncpy but does not 0 fill the buffer and always null 
101  * terminates. bufsize is the size of the destination buffer.
102  *
103  * Returns the index of the terminating byte. */
104  size_t strlcpy(char *d, const char *s, size_t bufsize)
105 {
106         size_t len = strlen(s);
107         size_t ret = len;
108         if (bufsize <= 0) return 0;
109         if (len >= bufsize) len = bufsize-1;
110         memcpy(d, s, len);
111         d[len] = 0;
112         return ret;
113 }
114 #endif
115
116 #ifndef HAVE_STRLCAT
117 /* like strncat but does not 0 fill the buffer and always null 
118    terminates. bufsize is the length of the buffer, which should
119    be one more than the maximum resulting string length */
120  size_t strlcat(char *d, const char *s, size_t bufsize)
121 {
122         size_t len1 = strlen(d);
123         size_t len2 = strlen(s);
124         size_t ret = len1 + len2;
125
126         if (len1+len2 >= bufsize) {
127                 len2 = bufsize - (len1+1);
128         }
129         if (len2 > 0) {
130                 memcpy(d+len1, s, len2);
131                 d[len1+len2] = 0;
132         }
133         return ret;
134 }
135 #endif
136
137 #ifdef REPLACE_INET_NTOA
138  char *rep_inet_ntoa(struct in_addr ip)
139 {
140         unsigned char *p = (unsigned char *)&ip.s_addr;
141         static char buf[18];
142 #if WORDS_BIGENDIAN
143         snprintf(buf, 18, "%d.%d.%d.%d", 
144                  (int)p[0], (int)p[1], (int)p[2], (int)p[3]);
145 #else
146         snprintf(buf, 18, "%d.%d.%d.%d", 
147                  (int)p[3], (int)p[2], (int)p[1], (int)p[0]);
148 #endif
149         return buf;
150 }
151 #endif
152
153 #ifdef REPLACE_INET_ATON
154  int inet_aton(const char *cp, struct in_addr *inp)
155 {
156         unsigned int a1, a2, a3, a4;
157         unsigned long ret;
158
159         if (strcmp(cp, "255.255.255.255") == 0) {
160                 inp->s_addr = (unsigned) -1;
161                 return 0;
162         }
163
164         if (sscanf(cp, "%u.%u.%u.%u", &a1, &a2, &a3, &a4) != 4 ||
165             a1 > 255 || a2 > 255 || a3 > 255 || a4 > 255) {
166                 return 0;
167         }
168
169         ret = (a1 << 24) | (a2 << 16) | (a3 << 8) | a4;
170
171         inp->s_addr = htonl(ret);
172         
173         if (inp->s_addr == (unsigned) -1) {
174                 return 0;
175         }
176         return 1;
177 }
178 #endif
179
180 /* some systems don't take the 2nd argument */
181 int sys_gettimeofday(struct timeval *tv)
182 {
183 #if HAVE_GETTIMEOFDAY_TZ
184         return gettimeofday(tv, NULL);
185 #else
186         return gettimeofday(tv);
187 #endif
188 }