int safe_accept(int fd, struct sockaddr *sa, socklen_t *sa_len_p)
{
int res;
loop:
res = accept(fd, sa, sa_len_p);
if (res < 0 && errno == EINTR)
goto loop;
if (res < 0)
log_noise("safe_accept(%d) = %s", fd, strerror(errno));
else if (cf_verbose > 2)
log_noise("safe_accept(%d) = %d (%s)", fd, res, sa2str(sa));
return res;
}
int safe_connect(int fd, const struct sockaddr *sa, socklen_t sa_len)
{
int res;
loop:
res = connect(fd, sa, sa_len);
if (res < 0 && errno == EINTR)
goto loop;
if (res < 0 && (errno != EINPROGRESS || cf_verbose > 2))
log_noise("connect(%d, %s) = %s", fd, sa2str(sa), strerror(errno));
else if (cf_verbose > 2)
log_noise("connect(%d, %s) = %d", fd, sa2str(sa), res);
return res;
}
int safe_send(int fd, const void *buf, int len, int flags)
{
int res;
loop:
res = send(fd, buf, len, flags);
if (res < 0 && errno == EINTR)
goto loop;
if (res < 0)
log_noise("safe_send(%d, %d) = %s", fd, len, strerror(errno));
else if (cf_verbose > 2)
log_noise("safe_send(%d, %d) = %d", fd, len, res);
return res;
}
void fill_remote_addr(PgSocket *sk, int fd, bool is_unix)
{
PgAddr *dst = &sk->remote_addr;
socklen_t len = sizeof(PgAddr);
int err;
if (is_unix) {
uid_t uid = 0;
gid_t gid = 0;
pid_t pid = 0;
pga_set(dst, AF_UNIX, cf_listen_port);
if (getpeercreds(fd, &uid, &gid, &pid) >= 0) {
log_noise("unix peer uid: %d", (int)uid);
} else {
log_warning("unix peer uid failed: %s", strerror(errno));
}
dst->scred.uid = uid;
dst->scred.pid = pid;
} else {
err = getpeername(fd, (struct sockaddr *)dst, &len);
if (err < 0) {
log_error("fill_remote_addr: getpeername(%d) = %s",
fd, strerror(errno));
}
}
}
int safe_sendmsg(int fd, const struct msghdr *msg, int flags)
{
int res;
int msgerr_count = 0;
loop:
res = sendmsg(fd, msg, flags);
if (res < 0 && errno == EINTR)
goto loop;
if (res < 0) {
log_warning("safe_sendmsg(%d, msg[%d,%d], %d) = %s", fd,
(int)msg->msg_iov[0].iov_len,
(int)msg->msg_controllen,
flags, strerror(errno));
/* with ancillary data on blocking socket OSX returns
* EMSGSIZE instead of blocking. try to solve it by waiting */
if (errno == EMSGSIZE && msgerr_count < 20) {
struct timeval tv = {1, 0};
log_warning("trying to sleep a bit");
select(0, NULL, NULL, NULL, &tv);
msgerr_count++;
goto loop;
}
} else if (cf_verbose > 2)
log_noise("safe_sendmsg(%d, msg, %d) = %d", fd, flags, res);
return res;
}
int safe_recvmsg(int fd, struct msghdr *msg, int flags)
{
int res;
loop:
res = recvmsg(fd, msg, flags);
if (res < 0 && errno == EINTR)
goto loop;
if (res < 0)
log_warning("safe_recvmsg(%d, msg, %d) = %s", fd, flags, strerror(errno));
else if (cf_verbose > 2)
log_noise("safe_recvmsg(%d, msg, %d) = %d", fd, flags, res);
return res;
}
/* reposition at buffer start again */
static void sbuf_try_resync(SBuf *sbuf, bool release)
{
IOBuf *io = sbuf->io;
if (io) {
log_noise("resync: done=%d, parse=%d, recv=%d",
io->done_pos, io->parse_pos, io->recv_pos);
}
AssertActive(sbuf);
if (!io)
return;
if (release && iobuf_empty(io)) {
slab_free(iobuf_cache, io);
sbuf->io = NULL;
} else {
iobuf_try_resync(io, SBUF_SMALL_PKT);
}
}
/*
* Get key=val pair from connstring. returns position it stopped
* or NULL on error. EOF is signaled by *key = 0.
*/
static char * cstr_get_pair(char *p,
char **key_p,
char **val_p)
{
p = cstr_skip_ws(p);
*key_p = *val_p = p;
if (*p == 0)
return p;
/* read key */
p = cstr_get_key(p, key_p);
if (!p)
return NULL;
/* read value */
p = cstr_get_value(p, val_p);
if (!p)
return NULL;
log_noise("cstr_get_pair: \"%s\"=\"%s\"", *key_p, *val_p);
return cstr_skip_ws(p);
}
