OLD | NEW |
(Empty) | |
| 1 #include <sys/socket.h> |
| 2 #include <byteswap.h> |
| 3 #include <unistd.h> |
| 4 #include <stdio.h> |
| 5 #include <string.h> |
| 6 #include <errno.h> |
| 7 #include <limits.h> |
| 8 #include "nscd.h" |
| 9 |
| 10 static const struct { |
| 11 short sun_family; |
| 12 char sun_path[21]; |
| 13 } addr = { |
| 14 AF_UNIX, |
| 15 "/var/run/nscd/socket" |
| 16 }; |
| 17 |
| 18 FILE *__nscd_query(int32_t req, const char *key, int32_t *buf, size_t len, int *
swap) |
| 19 { |
| 20 size_t i; |
| 21 int fd; |
| 22 FILE *f = 0; |
| 23 int32_t req_buf[REQ_LEN] = { |
| 24 NSCDVERSION, |
| 25 req, |
| 26 strnlen(key,LOGIN_NAME_MAX)+1 |
| 27 }; |
| 28 struct msghdr msg = { |
| 29 .msg_iov = (struct iovec[]){ |
| 30 {&req_buf, sizeof(req_buf)}, |
| 31 {(char*)key, strlen(key)+1} |
| 32 }, |
| 33 .msg_iovlen = 2 |
| 34 }; |
| 35 int errno_save = errno; |
| 36 |
| 37 *swap = 0; |
| 38 retry: |
| 39 memset(buf, 0, len); |
| 40 buf[0] = NSCDVERSION; |
| 41 |
| 42 fd = socket(PF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0); |
| 43 if (fd < 0) return NULL; |
| 44 |
| 45 if(!(f = fdopen(fd, "r"))) { |
| 46 close(fd); |
| 47 return 0; |
| 48 } |
| 49 |
| 50 if (req_buf[2] > LOGIN_NAME_MAX) |
| 51 return f; |
| 52 |
| 53 if (connect(fd, (struct sockaddr*)&addr, sizeof(addr)) < 0) { |
| 54 /* If there isn't a running nscd we simulate a "not found" |
| 55 * result and the caller is responsible for calling |
| 56 * fclose on the (unconnected) socket. The value of |
| 57 * errno must be left unchanged in this case. */ |
| 58 if (errno == EACCES || errno == ECONNREFUSED || errno == ENOENT)
{ |
| 59 errno = errno_save; |
| 60 return f; |
| 61 } |
| 62 goto error; |
| 63 } |
| 64 |
| 65 if (sendmsg(fd, &msg, MSG_NOSIGNAL) < 0) |
| 66 goto error; |
| 67 |
| 68 if (!fread(buf, len, 1, f)) { |
| 69 /* If the VERSION entry mismatches nscd will disconnect. The |
| 70 * most likely cause is that the endianness mismatched. So, we |
| 71 * byteswap and try once more. (if we already swapped, just |
| 72 * fail out) |
| 73 */ |
| 74 if (ferror(f)) goto error; |
| 75 if (!*swap) { |
| 76 fclose(f); |
| 77 for (i = 0; i < sizeof(req_buf)/sizeof(req_buf[0]); i++)
{ |
| 78 req_buf[i] = bswap_32(req_buf[i]); |
| 79 } |
| 80 *swap = 1; |
| 81 goto retry; |
| 82 } else { |
| 83 errno = EIO; |
| 84 goto error; |
| 85 } |
| 86 } |
| 87 |
| 88 if (*swap) { |
| 89 for (i = 0; i < len/sizeof(buf[0]); i++) { |
| 90 buf[i] = bswap_32(buf[i]); |
| 91 } |
| 92 } |
| 93 |
| 94 /* The first entry in every nscd response is the version number. This |
| 95 * really shouldn't happen, and is evidence of some form of malformed |
| 96 * response. |
| 97 */ |
| 98 if(buf[0] != NSCDVERSION) { |
| 99 errno = EIO; |
| 100 goto error; |
| 101 } |
| 102 |
| 103 return f; |
| 104 error: |
| 105 fclose(f); |
| 106 return 0; |
| 107 } |
OLD | NEW |