File: [local] / src / usr.sbin / iscsid / util.c (download)
Revision 1.9, Mon Apr 12 10:03:33 2021 UTC (3 years, 1 month ago) by claudio
Branch: MAIN
CVS Tags: OPENBSD_7_5_BASE, OPENBSD_7_5, OPENBSD_7_4_BASE, OPENBSD_7_4, OPENBSD_7_3_BASE, OPENBSD_7_3, OPENBSD_7_2_BASE, OPENBSD_7_2, OPENBSD_7_1_BASE, OPENBSD_7_1, OPENBSD_7_0_BASE, OPENBSD_7_0, OPENBSD_6_9_BASE, OPENBSD_6_9, HEAD Changes since 1.8: +2 -2 lines
Spaces, no functional change
|
/* $OpenBSD: util.c,v 1.9 2021/04/12 10:03:33 claudio Exp $ */
/*
* Copyright (c) 2009 Claudio Jeker <claudio@openbsd.org>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <sys/queue.h>
#include <sys/socket.h>
#include <sys/uio.h>
#include <scsi/iscsi.h>
#include <errno.h>
#include <event.h>
#include <fcntl.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "iscsid.h"
#include "log.h"
struct pdu *
pdu_new(void)
{
struct pdu *p;
if (!(p = calloc(1, sizeof(*p))))
return NULL;
return p;
}
void *
pdu_alloc(size_t len)
{
return malloc(PDU_LEN(len));
}
void *
pdu_dup(void *data, size_t len)
{
void *p;
if ((p = malloc(PDU_LEN(len))))
memcpy(p, data, len);
return p;
}
int
pdu_addbuf(struct pdu *p, void *buf, size_t len, unsigned int elm)
{
if (len & 0x3) {
bzero((char *)buf + len, 4 - (len & 0x3));
len += 4 - (len & 0x3);
}
if (elm < PDU_MAXIOV)
if (!p->iov[elm].iov_base) {
p->iov[elm].iov_base = buf;
p->iov[elm].iov_len = len;
return 0;
}
/* no space left */
return -1;
}
void *
pdu_getbuf(struct pdu *p, size_t *len, unsigned int elm)
{
if (len)
*len = 0;
if (elm < PDU_MAXIOV)
if (p->iov[elm].iov_base) {
if (len)
*len = p->iov[elm].iov_len;
return p->iov[elm].iov_base;
}
return NULL;
}
void
pdu_free(struct pdu *p)
{
unsigned int j;
for (j = 0; j < PDU_MAXIOV; j++)
free(p->iov[j].iov_base);
free(p);
}
int
socket_setblockmode(int fd, int nonblocking)
{
int flags;
if ((flags = fcntl(fd, F_GETFL)) == -1)
return -1;
if (nonblocking)
flags |= O_NONBLOCK;
else
flags &= ~O_NONBLOCK;
if ((flags = fcntl(fd, F_SETFL, flags)) == -1)
return -1;
return 0;
}
const char *
log_sockaddr(void *arg)
{
struct sockaddr *sa = arg;
char port[6];
char host[NI_MAXHOST];
static char buf[NI_MAXHOST + 8];
if (getnameinfo(sa, sa->sa_len, host, sizeof(host), port, sizeof(port),
NI_NUMERICHOST))
return "unknown";
if (port[0] == '0')
strlcpy(buf, host, sizeof(buf));
else if (sa->sa_family == AF_INET)
snprintf(buf, sizeof(buf), "%s:%s", host, port);
else
snprintf(buf, sizeof(buf), "[%s]:%s", host, port);
return buf;
}
int
control_compose(void *ch, u_int16_t type, void *buf, size_t len)
{
return control_build(ch, type, 1, CTRLARGV({ buf, len }));
}
int
control_build(void *ch, u_int16_t type, int argc, struct ctrldata *argv)
{
struct pdu *pdu;
struct ctrlmsghdr *cmh;
size_t size = 0;
int i;
if (argc > (int)nitems(cmh->len))
return -1;
for (i = 0; i < argc; i++)
size += argv[i].len;
if (PDU_LEN(size) > CONTROL_READ_SIZE - PDU_LEN(sizeof(*cmh)))
return -1;
if ((pdu = pdu_new()) == NULL)
return -1;
if ((cmh = pdu_alloc(sizeof(*cmh))) == NULL)
goto fail;
bzero(cmh, sizeof(*cmh));
cmh->type = type;
pdu_addbuf(pdu, cmh, sizeof(*cmh), 0);
for (i = 0; i < argc; i++)
if (argv[i].len > 0) {
void *ptr;
cmh->len[i] = argv[i].len;
if ((ptr = pdu_alloc(argv[i].len)) == NULL)
goto fail;
memcpy(ptr, argv[i].buf, argv[i].len);
pdu_addbuf(pdu, ptr, argv[i].len, i + 1);
}
control_queue(ch, pdu);
return 0;
fail:
pdu_free(pdu);
return -1;
}