njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 1 | |
| 2 | /*--------------------------------------------------------------------*/ |
| 3 | /*--- File- and socket-related libc stuff. m_libcfile.c ---*/ |
| 4 | /*--------------------------------------------------------------------*/ |
| 5 | |
| 6 | /* |
| 7 | This file is part of Valgrind, a dynamic binary instrumentation |
| 8 | framework. |
| 9 | |
sewardj | e4b0bf0 | 2006-06-05 23:21:15 +0000 | [diff] [blame] | 10 | Copyright (C) 2000-2006 Julian Seward |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 11 | jseward@acm.org |
| 12 | |
| 13 | This program is free software; you can redistribute it and/or |
| 14 | modify it under the terms of the GNU General Public License as |
| 15 | published by the Free Software Foundation; either version 2 of the |
| 16 | License, or (at your option) any later version. |
| 17 | |
| 18 | This program is distributed in the hope that it will be useful, but |
| 19 | WITHOUT ANY WARRANTY; without even the implied warranty of |
| 20 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 21 | General Public License for more details. |
| 22 | |
| 23 | You should have received a copy of the GNU General Public License |
| 24 | along with this program; if not, write to the Free Software |
| 25 | Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA |
| 26 | 02111-1307, USA. |
| 27 | |
| 28 | The GNU General Public License is contained in the file COPYING. |
| 29 | */ |
| 30 | |
njn | c7561b9 | 2005-06-19 01:24:32 +0000 | [diff] [blame] | 31 | #include "pub_core_basics.h" |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 32 | #include "pub_core_libcbase.h" |
| 33 | #include "pub_core_libcassert.h" |
| 34 | #include "pub_core_libcfile.h" |
sewardj | fdf91b4 | 2005-09-28 00:53:09 +0000 | [diff] [blame] | 35 | #include "pub_core_libcprint.h" // VG_(sprintf) |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 36 | #include "pub_core_libcproc.h" // VG_(getpid), VG_(getppid) |
sewardj | fdf91b4 | 2005-09-28 00:53:09 +0000 | [diff] [blame] | 37 | #include "pub_core_clientstate.h" // VG_(fd_hard_limit) |
njn | 9abd608 | 2005-06-17 21:31:45 +0000 | [diff] [blame] | 38 | #include "pub_core_syscall.h" |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 39 | #include "vki_unistd.h" |
| 40 | |
| 41 | /* --------------------------------------------------------------------- |
| 42 | File stuff |
| 43 | ------------------------------------------------------------------ */ |
| 44 | |
| 45 | static inline Bool fd_exists(Int fd) |
| 46 | { |
| 47 | struct vki_stat st; |
| 48 | |
| 49 | return VG_(fstat)(fd, &st) == 0; |
| 50 | } |
| 51 | |
| 52 | /* Move an fd into the Valgrind-safe range */ |
| 53 | Int VG_(safe_fd)(Int oldfd) |
| 54 | { |
| 55 | Int newfd; |
| 56 | |
| 57 | vg_assert(VG_(fd_hard_limit) != -1); |
| 58 | |
| 59 | newfd = VG_(fcntl)(oldfd, VKI_F_DUPFD, VG_(fd_hard_limit)); |
| 60 | if (newfd != -1) |
| 61 | VG_(close)(oldfd); |
| 62 | |
| 63 | VG_(fcntl)(newfd, VKI_F_SETFD, VKI_FD_CLOEXEC); |
| 64 | |
| 65 | vg_assert(newfd >= VG_(fd_hard_limit)); |
| 66 | return newfd; |
| 67 | } |
| 68 | |
njn | ae7359b | 2005-06-19 21:10:42 +0000 | [diff] [blame] | 69 | /* Given a file descriptor, attempt to deduce its filename. To do |
| 70 | this, we use /proc/self/fd/<FD>. If this doesn't point to a file, |
njn | f845f8f | 2005-06-23 02:26:47 +0000 | [diff] [blame] | 71 | or if it doesn't exist, we return False. */ |
| 72 | Bool VG_(resolve_filename) ( Int fd, HChar* buf, Int n_buf ) |
njn | ae7359b | 2005-06-19 21:10:42 +0000 | [diff] [blame] | 73 | { |
| 74 | HChar tmp[64]; |
| 75 | |
| 76 | VG_(sprintf)(tmp, "/proc/self/fd/%d", fd); |
njn | f845f8f | 2005-06-23 02:26:47 +0000 | [diff] [blame] | 77 | VG_(memset)(buf, 0, n_buf); |
njn | ae7359b | 2005-06-19 21:10:42 +0000 | [diff] [blame] | 78 | |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 79 | if (VG_(readlink)(tmp, buf, n_buf) > 0 && buf[0] == '/') |
njn | f845f8f | 2005-06-23 02:26:47 +0000 | [diff] [blame] | 80 | return True; |
| 81 | else |
| 82 | return False; |
njn | ae7359b | 2005-06-19 21:10:42 +0000 | [diff] [blame] | 83 | } |
| 84 | |
sewardj | 9264559 | 2005-07-23 09:18:34 +0000 | [diff] [blame] | 85 | SysRes VG_(open) ( const Char* pathname, Int flags, Int mode ) |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 86 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 87 | SysRes res = VG_(do_syscall3)(__NR_open, (UWord)pathname, flags, mode); |
sewardj | 9264559 | 2005-07-23 09:18:34 +0000 | [diff] [blame] | 88 | return res; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 89 | } |
| 90 | |
| 91 | void VG_(close) ( Int fd ) |
| 92 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 93 | (void)VG_(do_syscall1)(__NR_close, fd); |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 94 | } |
| 95 | |
| 96 | Int VG_(read) ( Int fd, void* buf, Int count) |
| 97 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 98 | SysRes res = VG_(do_syscall3)(__NR_read, fd, (UWord)buf, count); |
| 99 | return res.isError ? -1 : res.val; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 100 | } |
| 101 | |
| 102 | Int VG_(write) ( Int fd, const void* buf, Int count) |
| 103 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 104 | SysRes res = VG_(do_syscall3)(__NR_write, fd, (UWord)buf, count); |
| 105 | return res.isError ? -1 : res.val; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 106 | } |
| 107 | |
| 108 | Int VG_(pipe) ( Int fd[2] ) |
| 109 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 110 | SysRes res = VG_(do_syscall1)(__NR_pipe, (UWord)fd); |
| 111 | return res.isError ? -1 : 0; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 112 | } |
| 113 | |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 114 | OffT VG_(lseek) ( Int fd, OffT offset, Int whence ) |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 115 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 116 | SysRes res = VG_(do_syscall3)(__NR_lseek, fd, offset, whence); |
tom | f4c2310 | 2005-10-31 17:05:21 +0000 | [diff] [blame] | 117 | return res.isError ? (-1) : res.val; |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 118 | /* if you change the error-reporting conventions of this, also |
| 119 | change VG_(pread) and all other usage points. */ |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 120 | } |
| 121 | |
tom | 6c93c4f | 2005-08-05 07:46:32 +0000 | [diff] [blame] | 122 | SysRes VG_(stat) ( Char* file_name, struct vki_stat* buf ) |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 123 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 124 | SysRes res = VG_(do_syscall2)(__NR_stat, (UWord)file_name, (UWord)buf); |
tom | 6c93c4f | 2005-08-05 07:46:32 +0000 | [diff] [blame] | 125 | return res; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 126 | } |
| 127 | |
| 128 | Int VG_(fstat) ( Int fd, struct vki_stat* buf ) |
| 129 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 130 | SysRes res = VG_(do_syscall2)(__NR_fstat, fd, (UWord)buf); |
| 131 | return res.isError ? (-1) : 0; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 132 | } |
| 133 | |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 134 | Int VG_(fsize) ( Int fd ) |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 135 | { |
tom | 574b893 | 2006-07-05 17:47:46 +0000 | [diff] [blame^] | 136 | #ifdef __NR_fstat64 |
| 137 | struct vki_stat64 buf; |
| 138 | SysRes res = VG_(do_syscall2)(__NR_fstat64, fd, (UWord)&buf); |
| 139 | #else |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 140 | struct vki_stat buf; |
| 141 | SysRes res = VG_(do_syscall2)(__NR_fstat, fd, (UWord)&buf); |
tom | 574b893 | 2006-07-05 17:47:46 +0000 | [diff] [blame^] | 142 | #endif |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 143 | return res.isError ? (-1) : buf.st_size; |
| 144 | } |
| 145 | |
njn | 7375061 | 2005-10-14 03:11:30 +0000 | [diff] [blame] | 146 | Bool VG_(is_dir) ( HChar* f ) |
| 147 | { |
| 148 | struct vki_stat buf; |
| 149 | SysRes res = VG_(do_syscall2)(__NR_stat, (UWord)f, (UWord)&buf); |
| 150 | return res.isError ? False |
| 151 | : VKI_S_ISDIR(buf.st_mode) ? True : False; |
| 152 | } |
| 153 | |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 154 | SysRes VG_(dup) ( Int oldfd ) |
| 155 | { |
| 156 | return VG_(do_syscall1)(__NR_dup, oldfd); |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 157 | } |
| 158 | |
njn | 327fe8a | 2005-06-12 02:49:35 +0000 | [diff] [blame] | 159 | /* Returns -1 on error. */ |
| 160 | Int VG_(fcntl) ( Int fd, Int cmd, Int arg ) |
| 161 | { |
| 162 | SysRes res = VG_(do_syscall3)(__NR_fcntl, fd, cmd, arg); |
| 163 | return res.isError ? -1 : res.val; |
| 164 | } |
| 165 | |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 166 | Int VG_(rename) ( Char* old_name, Char* new_name ) |
| 167 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 168 | SysRes res = VG_(do_syscall2)(__NR_rename, (UWord)old_name, (UWord)new_name); |
| 169 | return res.isError ? (-1) : 0; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 170 | } |
| 171 | |
| 172 | Int VG_(unlink) ( Char* file_name ) |
| 173 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 174 | SysRes res = VG_(do_syscall1)(__NR_unlink, (UWord)file_name); |
| 175 | return res.isError ? (-1) : 0; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 176 | } |
| 177 | |
| 178 | /* Nb: we do not allow the Linux extension which malloc()s memory for the |
| 179 | buffer if buf==NULL, because we don't want Linux calling malloc() */ |
njn | 57ca7ab | 2005-06-21 23:44:58 +0000 | [diff] [blame] | 180 | Bool VG_(getcwd) ( Char* buf, SizeT size ) |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 181 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 182 | SysRes res; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 183 | vg_assert(buf != NULL); |
| 184 | res = VG_(do_syscall2)(__NR_getcwd, (UWord)buf, size); |
njn | 57ca7ab | 2005-06-21 23:44:58 +0000 | [diff] [blame] | 185 | return res.isError ? False : True; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 186 | } |
| 187 | |
| 188 | Int VG_(readlink) (Char* path, Char* buf, UInt bufsiz) |
| 189 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 190 | SysRes res; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 191 | /* res = readlink( path, buf, bufsiz ); */ |
| 192 | res = VG_(do_syscall3)(__NR_readlink, (UWord)path, (UWord)buf, bufsiz); |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 193 | return res.isError ? -1 : res.val; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 194 | } |
| 195 | |
| 196 | Int VG_(getdents) (UInt fd, struct vki_dirent *dirp, UInt count) |
| 197 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 198 | SysRes res; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 199 | /* res = getdents( fd, dirp, count ); */ |
| 200 | res = VG_(do_syscall3)(__NR_getdents, fd, (UWord)dirp, count); |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 201 | return res.isError ? -1 : res.val; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 202 | } |
| 203 | |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 204 | /* Check accessibility of a file. Returns zero for access granted, |
| 205 | nonzero otherwise. */ |
| 206 | Int VG_(access) ( HChar* path, Bool irusr, Bool iwusr, Bool ixusr ) |
| 207 | { |
| 208 | #if defined(VGO_linux) |
| 209 | /* Very annoyingly, I cannot find any definition for R_OK et al in |
| 210 | the kernel interfaces. Therefore I reluctantly resort to |
| 211 | hardwiring in these magic numbers that I determined by |
| 212 | experimentation. */ |
| 213 | UWord w = (irusr ? 4/*R_OK*/ : 0) |
| 214 | | (iwusr ? 2/*W_OK*/ : 0) |
| 215 | | (ixusr ? 1/*X_OK*/ : 0); |
| 216 | SysRes res = VG_(do_syscall2)(__NR_access, (UWord)path, w); |
| 217 | return res.isError ? 1 : res.val; |
| 218 | #else |
| 219 | # error "Don't know how to do VG_(access) on this OS" |
| 220 | #endif |
| 221 | } |
| 222 | |
njn | 7375061 | 2005-10-14 03:11:30 +0000 | [diff] [blame] | 223 | /* |
| 224 | Emulate the normal Unix permissions checking algorithm. |
| 225 | |
| 226 | If owner matches, then use the owner permissions, else |
| 227 | if group matches, then use the group permissions, else |
| 228 | use other permissions. |
| 229 | |
| 230 | Note that we can't deal with SUID/SGID, so we refuse to run them |
| 231 | (otherwise the executable may misbehave if it doesn't have the |
| 232 | permissions it thinks it does). |
| 233 | */ |
| 234 | /* returns: 0 = success, non-0 is failure */ |
| 235 | Int VG_(check_executable)(HChar* f) |
| 236 | { |
| 237 | struct vki_stat st; |
| 238 | SysRes res; |
| 239 | |
| 240 | res = VG_(stat)(f, &st); |
| 241 | if (res.isError) { |
| 242 | return res.val; |
| 243 | } |
| 244 | |
| 245 | if (st.st_mode & (VKI_S_ISUID | VKI_S_ISGID)) { |
| 246 | //VG_(printf)("Can't execute suid/sgid executable %s\n", exe); |
| 247 | return VKI_EACCES; |
| 248 | } |
| 249 | |
| 250 | if (VG_(geteuid)() == st.st_uid) { |
| 251 | if (!(st.st_mode & VKI_S_IXUSR)) |
| 252 | return VKI_EACCES; |
| 253 | } else { |
| 254 | int grpmatch = 0; |
| 255 | |
| 256 | if (VG_(getegid)() == st.st_gid) |
| 257 | grpmatch = 1; |
| 258 | else { |
| 259 | UInt groups[32]; |
| 260 | Int ngrp = VG_(getgroups)(32, groups); |
| 261 | Int i; |
| 262 | /* ngrp will be -1 if VG_(getgroups) failed. */ |
| 263 | for (i = 0; i < ngrp; i++) { |
| 264 | if (groups[i] == st.st_gid) { |
| 265 | grpmatch = 1; |
| 266 | break; |
| 267 | } |
| 268 | } |
| 269 | } |
| 270 | |
| 271 | if (grpmatch) { |
| 272 | if (!(st.st_mode & VKI_S_IXGRP)) { |
| 273 | return VKI_EACCES; |
| 274 | } |
| 275 | } else if (!(st.st_mode & VKI_S_IXOTH)) { |
| 276 | return VKI_EACCES; |
| 277 | } |
| 278 | } |
| 279 | |
| 280 | return 0; |
| 281 | } |
| 282 | |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 283 | SysRes VG_(pread) ( Int fd, void* buf, Int count, Int offset ) |
| 284 | { |
| 285 | OffT off = VG_(lseek)( fd, (OffT)offset, VKI_SEEK_SET); |
tom | f4c2310 | 2005-10-31 17:05:21 +0000 | [diff] [blame] | 286 | if (off < 0) |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 287 | return VG_(mk_SysRes_Error)( VKI_EINVAL ); |
tom | 60a4b0b | 2005-10-12 10:45:27 +0000 | [diff] [blame] | 288 | return VG_(do_syscall3)(__NR_read, fd, (UWord)buf, count ); |
sewardj | 45f4e7c | 2005-09-27 19:20:21 +0000 | [diff] [blame] | 289 | } |
| 290 | |
| 291 | /* Create and open (-rw------) a tmp file name incorporating said arg. |
| 292 | Returns -1 on failure, else the fd of the file. If fullname is |
| 293 | non-NULL, the file's name is written into it. The number of bytes |
| 294 | written is guaranteed not to exceed 64+strlen(part_of_name). */ |
| 295 | |
| 296 | Int VG_(mkstemp) ( HChar* part_of_name, /*OUT*/HChar* fullname ) |
| 297 | { |
| 298 | HChar buf[200]; |
| 299 | Int n, tries, fd; |
| 300 | UInt seed; |
| 301 | SysRes sres; |
| 302 | |
| 303 | vg_assert(part_of_name); |
| 304 | n = VG_(strlen)(part_of_name); |
| 305 | vg_assert(n > 0 && n < 100); |
| 306 | |
| 307 | seed = (VG_(getpid)() << 9) ^ VG_(getppid)(); |
| 308 | |
| 309 | tries = 0; |
| 310 | while (True) { |
| 311 | if (tries > 10) |
| 312 | return -1; |
| 313 | VG_(sprintf)( buf, "/tmp/valgrind_%s_%08x", |
| 314 | part_of_name, VG_(random)( &seed )); |
| 315 | if (0) |
| 316 | VG_(printf)("VG_(mkstemp): trying: %s\n", buf); |
| 317 | |
| 318 | sres = VG_(open)(buf, |
| 319 | VKI_O_CREAT|VKI_O_RDWR|VKI_O_EXCL|VKI_O_TRUNC, |
| 320 | VKI_S_IRUSR|VKI_S_IWUSR); |
| 321 | if (sres.isError) |
| 322 | continue; |
| 323 | /* VG_(safe_fd) doesn't return if it fails. */ |
| 324 | fd = VG_(safe_fd)( sres.val ); |
| 325 | if (fullname) |
| 326 | VG_(strcpy)( fullname, buf ); |
| 327 | return fd; |
| 328 | } |
| 329 | /* NOTREACHED */ |
| 330 | } |
| 331 | |
| 332 | |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 333 | /* --------------------------------------------------------------------- |
| 334 | Socket-related stuff. This is very Linux-kernel specific. |
| 335 | ------------------------------------------------------------------ */ |
| 336 | |
| 337 | static |
| 338 | Int parse_inet_addr_and_port ( UChar* str, UInt* ip_addr, UShort* port ); |
| 339 | |
| 340 | static |
| 341 | Int my_socket ( Int domain, Int type, Int protocol ); |
| 342 | |
| 343 | static |
| 344 | Int my_connect ( Int sockfd, struct vki_sockaddr_in* serv_addr, |
| 345 | Int addrlen ); |
| 346 | |
tom | a4a4f41 | 2005-11-17 12:01:56 +0000 | [diff] [blame] | 347 | UInt VG_(htonl) ( UInt x ) |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 348 | { |
sewardj | a1c4bfb | 2005-10-18 02:15:39 +0000 | [diff] [blame] | 349 | # if defined(VG_BIGENDIAN) |
| 350 | return x; |
| 351 | # else |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 352 | return |
| 353 | (((x >> 24) & 0xFF) << 0) | (((x >> 16) & 0xFF) << 8) |
| 354 | | (((x >> 8) & 0xFF) << 16) | (((x >> 0) & 0xFF) << 24); |
sewardj | a1c4bfb | 2005-10-18 02:15:39 +0000 | [diff] [blame] | 355 | # endif |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 356 | } |
| 357 | |
tom | a4a4f41 | 2005-11-17 12:01:56 +0000 | [diff] [blame] | 358 | UInt VG_(ntohl) ( UInt x ) |
| 359 | { |
| 360 | # if defined(VG_BIGENDIAN) |
| 361 | return x; |
| 362 | # else |
| 363 | return |
| 364 | (((x >> 24) & 0xFF) << 0) | (((x >> 16) & 0xFF) << 8) |
| 365 | | (((x >> 8) & 0xFF) << 16) | (((x >> 0) & 0xFF) << 24); |
| 366 | # endif |
| 367 | } |
| 368 | |
| 369 | UShort VG_(htons) ( UShort x ) |
| 370 | { |
| 371 | # if defined(VG_BIGENDIAN) |
| 372 | return x; |
| 373 | # else |
| 374 | return |
| 375 | (((x >> 8) & 0xFF) << 0) | (((x >> 0) & 0xFF) << 8); |
| 376 | # endif |
| 377 | } |
| 378 | |
| 379 | UShort VG_(ntohs) ( UShort x ) |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 380 | { |
sewardj | a1c4bfb | 2005-10-18 02:15:39 +0000 | [diff] [blame] | 381 | # if defined(VG_BIGENDIAN) |
| 382 | return x; |
| 383 | # else |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 384 | return |
| 385 | (((x >> 8) & 0xFF) << 0) | (((x >> 0) & 0xFF) << 8); |
sewardj | a1c4bfb | 2005-10-18 02:15:39 +0000 | [diff] [blame] | 386 | # endif |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 387 | } |
| 388 | |
| 389 | |
| 390 | /* The main function. |
| 391 | |
| 392 | Supplied string contains either an ip address "192.168.0.1" or |
| 393 | an ip address and port pair, "192.168.0.1:1500". Parse these, |
| 394 | and return: |
| 395 | -1 if there is a parse error |
| 396 | -2 if no parse error, but specified host:port cannot be opened |
| 397 | the relevant file (socket) descriptor, otherwise. |
| 398 | is used. |
| 399 | */ |
| 400 | Int VG_(connect_via_socket)( UChar* str ) |
| 401 | { |
| 402 | Int sd, res; |
| 403 | struct vki_sockaddr_in servAddr; |
| 404 | UInt ip = 0; |
| 405 | UShort port = VG_CLO_DEFAULT_LOGPORT; |
| 406 | Bool ok = parse_inet_addr_and_port(str, &ip, &port); |
| 407 | if (!ok) |
| 408 | return -1; |
| 409 | |
| 410 | //if (0) |
| 411 | // VG_(printf)("ip = %d.%d.%d.%d, port %d\n", |
| 412 | // (ip >> 24) & 0xFF, (ip >> 16) & 0xFF, |
| 413 | // (ip >> 8) & 0xFF, ip & 0xFF, |
| 414 | // (UInt)port ); |
| 415 | |
| 416 | servAddr.sin_family = VKI_AF_INET; |
tom | a4a4f41 | 2005-11-17 12:01:56 +0000 | [diff] [blame] | 417 | servAddr.sin_addr.s_addr = VG_(htonl)(ip); |
| 418 | servAddr.sin_port = VG_(htons)(port); |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 419 | |
| 420 | /* create socket */ |
| 421 | sd = my_socket(VKI_AF_INET, VKI_SOCK_STREAM, 0 /* IPPROTO_IP ? */); |
| 422 | if (sd < 0) { |
sewardj | a1c4bfb | 2005-10-18 02:15:39 +0000 | [diff] [blame] | 423 | /* this shouldn't happen ... nevertheless */ |
| 424 | return -2; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 425 | } |
sewardj | a1c4bfb | 2005-10-18 02:15:39 +0000 | [diff] [blame] | 426 | |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 427 | /* connect to server */ |
| 428 | res = my_connect(sd, (struct vki_sockaddr_in *) &servAddr, |
| 429 | sizeof(servAddr)); |
| 430 | if (res < 0) { |
sewardj | a1c4bfb | 2005-10-18 02:15:39 +0000 | [diff] [blame] | 431 | /* connection failed */ |
| 432 | return -2; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 433 | } |
| 434 | |
| 435 | return sd; |
| 436 | } |
| 437 | |
| 438 | |
| 439 | /* Let d = one or more digits. Accept either: |
| 440 | d.d.d.d or d.d.d.d:d |
| 441 | */ |
| 442 | Int parse_inet_addr_and_port ( UChar* str, UInt* ip_addr, UShort* port ) |
| 443 | { |
| 444 | # define GET_CH ((*str) ? (*str++) : 0) |
| 445 | UInt ipa, i, j, c, any; |
| 446 | ipa = 0; |
| 447 | for (i = 0; i < 4; i++) { |
| 448 | j = 0; |
| 449 | any = 0; |
| 450 | while (1) { |
| 451 | c = GET_CH; |
| 452 | if (c < '0' || c > '9') break; |
| 453 | j = 10 * j + (int)(c - '0'); |
| 454 | any = 1; |
| 455 | } |
| 456 | if (any == 0 || j > 255) goto syntaxerr; |
| 457 | ipa = (ipa << 8) + j; |
| 458 | if (i <= 2 && c != '.') goto syntaxerr; |
| 459 | } |
| 460 | if (c == 0 || c == ':') |
| 461 | *ip_addr = ipa; |
| 462 | if (c == 0) goto ok; |
| 463 | if (c != ':') goto syntaxerr; |
| 464 | j = 0; |
| 465 | any = 0; |
| 466 | while (1) { |
| 467 | c = GET_CH; |
| 468 | if (c < '0' || c > '9') break; |
| 469 | j = j * 10 + (int)(c - '0'); |
| 470 | any = 1; |
| 471 | if (j > 65535) goto syntaxerr; |
| 472 | } |
| 473 | if (any == 0 || c != 0) goto syntaxerr; |
| 474 | if (j < 1024) goto syntaxerr; |
| 475 | *port = (UShort)j; |
| 476 | ok: |
| 477 | return 1; |
| 478 | syntaxerr: |
| 479 | return 0; |
| 480 | # undef GET_CH |
| 481 | } |
| 482 | |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 483 | static |
| 484 | Int my_socket ( Int domain, Int type, Int protocol ) |
| 485 | { |
sewardj | 2c48c7b | 2005-11-29 13:05:56 +0000 | [diff] [blame] | 486 | #if defined(VGP_x86_linux) || defined(VGP_ppc32_linux) || defined(VGP_ppc64_linux) |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 487 | SysRes res; |
| 488 | UWord args[3]; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 489 | args[0] = domain; |
| 490 | args[1] = type; |
| 491 | args[2] = protocol; |
| 492 | res = VG_(do_syscall2)(__NR_socketcall, VKI_SYS_SOCKET, (UWord)&args); |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 493 | return res.isError ? -1 : res.val; |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 494 | |
| 495 | #elif defined(VGP_amd64_linux) |
sewardj | 1e86656 | 2005-09-28 19:58:58 +0000 | [diff] [blame] | 496 | SysRes res; |
| 497 | res = VG_(do_syscall3)(__NR_socket, domain, type, protocol ); |
| 498 | return res.isError ? -1 : res.val; |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 499 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 500 | #else |
| 501 | # error Unknown arch |
| 502 | #endif |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 503 | } |
| 504 | |
| 505 | static |
| 506 | Int my_connect ( Int sockfd, struct vki_sockaddr_in* serv_addr, |
| 507 | Int addrlen ) |
| 508 | { |
sewardj | 2c48c7b | 2005-11-29 13:05:56 +0000 | [diff] [blame] | 509 | #if defined(VGP_x86_linux) || defined(VGP_ppc32_linux) || defined(VGP_ppc64_linux) |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 510 | SysRes res; |
| 511 | UWord args[3]; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 512 | args[0] = sockfd; |
| 513 | args[1] = (UWord)serv_addr; |
| 514 | args[2] = addrlen; |
| 515 | res = VG_(do_syscall2)(__NR_socketcall, VKI_SYS_CONNECT, (UWord)&args); |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 516 | return res.isError ? -1 : res.val; |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 517 | |
| 518 | #elif defined(VGP_amd64_linux) |
sewardj | 1e86656 | 2005-09-28 19:58:58 +0000 | [diff] [blame] | 519 | SysRes res; |
| 520 | res = VG_(do_syscall3)(__NR_connect, sockfd, (UWord)serv_addr, addrlen); |
| 521 | return res.isError ? -1 : res.val; |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 522 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 523 | #else |
| 524 | # error Unknown arch |
| 525 | #endif |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 526 | } |
| 527 | |
| 528 | Int VG_(write_socket)( Int sd, void *msg, Int count ) |
| 529 | { |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 530 | /* This is actually send(). */ |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 531 | /* Requests not to send SIGPIPE on errors on stream oriented |
| 532 | sockets when the other end breaks the connection. The EPIPE |
| 533 | error is still returned. */ |
| 534 | Int flags = VKI_MSG_NOSIGNAL; |
| 535 | |
sewardj | 2c48c7b | 2005-11-29 13:05:56 +0000 | [diff] [blame] | 536 | #if defined(VGP_x86_linux) || defined(VGP_ppc32_linux) || defined(VGP_ppc64_linux) |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 537 | SysRes res; |
| 538 | UWord args[4]; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 539 | args[0] = sd; |
| 540 | args[1] = (UWord)msg; |
| 541 | args[2] = count; |
| 542 | args[3] = flags; |
| 543 | res = VG_(do_syscall2)(__NR_socketcall, VKI_SYS_SEND, (UWord)&args); |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 544 | return res.isError ? -1 : res.val; |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 545 | |
| 546 | #elif defined(VGP_amd64_linux) |
sewardj | 1e86656 | 2005-09-28 19:58:58 +0000 | [diff] [blame] | 547 | SysRes res; |
| 548 | res = VG_(do_syscall6)(__NR_sendto, sd, (UWord)msg, count, flags, 0,0); |
| 549 | return res.isError ? -1 : res.val; |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 550 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 551 | #else |
| 552 | # error Unknown arch |
| 553 | #endif |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 554 | } |
| 555 | |
| 556 | Int VG_(getsockname) ( Int sd, struct vki_sockaddr *name, Int *namelen) |
| 557 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 558 | SysRes res; |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 559 | |
sewardj | 2c48c7b | 2005-11-29 13:05:56 +0000 | [diff] [blame] | 560 | #if defined(VGP_x86_linux) || defined(VGP_ppc32_linux) || defined(VGP_ppc64_linux) |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 561 | UWord args[3]; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 562 | args[0] = sd; |
| 563 | args[1] = (UWord)name; |
| 564 | args[2] = (UWord)namelen; |
| 565 | res = VG_(do_syscall2)(__NR_socketcall, VKI_SYS_GETSOCKNAME, (UWord)&args); |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 566 | return res.isError ? -1 : res.val; |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 567 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 568 | #elif defined(VGP_amd64_linux) |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 569 | res = VG_(do_syscall3)( __NR_getsockname, |
| 570 | (UWord)sd, (UWord)name, (UWord)namelen ); |
| 571 | return res.isError ? -1 : res.val; |
| 572 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 573 | #else |
| 574 | # error Unknown arch |
| 575 | #endif |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 576 | } |
| 577 | |
| 578 | Int VG_(getpeername) ( Int sd, struct vki_sockaddr *name, Int *namelen) |
| 579 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 580 | SysRes res; |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 581 | |
sewardj | 2c48c7b | 2005-11-29 13:05:56 +0000 | [diff] [blame] | 582 | #if defined(VGP_x86_linux) || defined(VGP_ppc32_linux) || defined(VGP_ppc64_linux) |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 583 | UWord args[3]; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 584 | args[0] = sd; |
| 585 | args[1] = (UWord)name; |
| 586 | args[2] = (UWord)namelen; |
| 587 | res = VG_(do_syscall2)(__NR_socketcall, VKI_SYS_GETPEERNAME, (UWord)&args); |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 588 | return res.isError ? -1 : res.val; |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 589 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 590 | #elif defined(VGP_amd64_linux) |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 591 | res = VG_(do_syscall3)( __NR_getpeername, |
| 592 | (UWord)sd, (UWord)name, (UWord)namelen ); |
| 593 | return res.isError ? -1 : res.val; |
| 594 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 595 | #else |
sewardj | 8a85128 | 2005-07-02 10:13:05 +0000 | [diff] [blame] | 596 | # error Unknown archx |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 597 | #endif |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 598 | } |
| 599 | |
| 600 | Int VG_(getsockopt) ( Int sd, Int level, Int optname, void *optval, |
| 601 | Int *optlen) |
| 602 | { |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 603 | SysRes res; |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 604 | |
sewardj | 2c48c7b | 2005-11-29 13:05:56 +0000 | [diff] [blame] | 605 | #if defined(VGP_x86_linux) || defined(VGP_ppc32_linux) || defined(VGP_ppc64_linux) |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 606 | UWord args[5]; |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 607 | args[0] = sd; |
| 608 | args[1] = level; |
| 609 | args[2] = optname; |
| 610 | args[3] = (UWord)optval; |
| 611 | args[4] = (UWord)optlen; |
| 612 | res = VG_(do_syscall2)(__NR_socketcall, VKI_SYS_GETSOCKOPT, (UWord)&args); |
sewardj | a8d8e23 | 2005-06-07 20:04:56 +0000 | [diff] [blame] | 613 | return res.isError ? -1 : res.val; |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 614 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 615 | #elif defined(VGP_amd64_linux) |
sewardj | a5c9e4a | 2005-06-09 13:21:58 +0000 | [diff] [blame] | 616 | res = VG_(do_syscall5)( __NR_getsockopt, |
| 617 | (UWord)sd, (UWord)level, (UWord)optname, |
| 618 | (UWord)optval, (UWord)optlen ); |
| 619 | return res.isError ? -1 : res.val; |
| 620 | |
cerion | 85665ca | 2005-06-20 15:51:07 +0000 | [diff] [blame] | 621 | #else |
| 622 | # error Unknown arch |
| 623 | #endif |
njn | eb8896b | 2005-06-04 20:03:55 +0000 | [diff] [blame] | 624 | } |
| 625 | |
| 626 | |
| 627 | |
| 628 | /*--------------------------------------------------------------------*/ |
| 629 | /*--- end ---*/ |
| 630 | /*--------------------------------------------------------------------*/ |
| 631 | |