qrtr/src/lib.c

#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <errno.h>
#include <poll.h>

#include "qrtr.h"
#include "lib.h"
#include "ns.h"

#define LOGW(fmt, ...) do { fprintf(stderr, "W|qrtr: " fmt "\n", ##__VA_ARGS__); } while (0)
#define LOGE_errno(fmt, ...) do { fprintf(stderr, "E|qrtr: " fmt ": %s\n", ##__VA_ARGS__, strerror(errno)); } while (0)

int qrtr_open(int rport)
{
	struct timeval tv;
	int sock;
	int rc;

	sock = socket(AF_QIPCRTR, SOCK_DGRAM, 0);
	if (sock < 0) {
		LOGE_errno("socket(AF_QIPCRTR)");
		return -1;
	}

	tv.tv_sec = 1;
	tv.tv_usec = 0;

	rc = setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
	if (rc) {
		LOGE_errno("setsockopt(SO_RCVTIMEO)");
		goto err;
	}

	if (rport != 0) {
		struct sockaddr_qrtr sq;

		sq.sq_family = AF_QIPCRTR;
		sq.sq_node = 0;
		sq.sq_port = rport;

		rc = bind(sock, (void *)&sq, sizeof(sq));
		if (rc < 0) {
			LOGE_errno("bind(%d)", rport);
			goto err;
		}
	}

	return sock;
err:
	close(sock);
	return -1;
}

void qrtr_close(int sock)
{
	close(sock);
}

int qrtr_sendto(int sock, u32 node, u32 port, const void *data, unsigned int sz)
{
	struct sockaddr_qrtr sq;
	int rc;

	sq.sq_family = AF_QIPCRTR;
	sq.sq_node = node;
	sq.sq_port = port;

	rc = sendto(sock, data, sz, 0, (void *)&sq, sizeof(sq));
	if (rc < 0) {
		LOGE_errno("sendto()");
		return -1;
	}

	return 0;
}

int qrtr_publish(int sock, u32 service, u32 instance)
{
	struct ns_pkt pkt;

	memset(&pkt, 0, sizeof(pkt));

	pkt.type = cpu_to_le32(NS_PKT_PUBLISH);
	pkt.publish.service = cpu_to_le32(service);
	pkt.publish.instance = cpu_to_le32(instance);

	return qrtr_sendto(sock, 0, NS_PORT, &pkt, sizeof(pkt));
}

int qrtr_bye(int sock, u32 service, u32 instance)
{
	struct ns_pkt pkt;

	memset(&pkt, 0, sizeof(pkt));

	pkt.type = cpu_to_le32(NS_PKT_BYE);
	pkt.bye.service = cpu_to_le32(service);
	pkt.bye.instance = cpu_to_le32(instance);

	return qrtr_sendto(sock, 0, NS_PORT, &pkt, sizeof(pkt));
}

int qrtr_poll(int sock, unsigned int ms)
{
	struct pollfd fds;

	fds.fd = sock;
	fds.revents = 0;
	fds.events = POLLIN | POLLERR;

	return poll(&fds, 1, ms);
}

int qrtr_recv(int sock, void *buf, unsigned int bsz)
{
	int rc;

	rc = recv(sock, buf, bsz, 0);
	if (rc < 0)
		LOGE_errno("recv()");
	return rc;
}

int qrtr_recvfrom(int sock, void *buf, unsigned int bsz, u32 *node, u32 *port)
{
	struct sockaddr_qrtr sq;
	socklen_t sl;
	int rc;

	sl = sizeof(sq);
	rc = recvfrom(sock, buf, bsz, 0, (void *)&sq, &sl);
	if (rc < 0) {
		LOGE_errno("recvfrom()");
		return rc;
	}
	if (node)
		*node = sq.sq_node;
	if (port)
		*port = sq.sq_port;
	return rc;
}

int qrtr_lookup(int sock, u32 service, u32 instance, u32 ifilter,
		void (* cb)(void *,u32,u32,u32,u32), void *udata)
{
	struct ns_pkt pkt;
	int len;
	int rc;

	memset(&pkt, 0, sizeof(pkt));

	pkt.type = cpu_to_le32(NS_PKT_QUERY);
	pkt.query.ifilter = cpu_to_le32(ifilter);
	pkt.query.instance = cpu_to_le32(instance);
	pkt.query.service = cpu_to_le32(service);

	rc = qrtr_sendto(sock, 0, NS_PORT, &pkt, sizeof(pkt));
	if (rc < 0)
		return -1;

	while ((len = recv(sock, &pkt, sizeof(pkt), 0)) > 0) {
		unsigned int type = le32_to_cpu(pkt.type);

		if (len < sizeof(pkt) || type != NS_PKT_NOTICE) {
			LOGW("invalid/short packet");
			continue;
		}

		if (pkt.notice.seq == 0)
			break;

		pkt.notice.service = le32_to_cpu(pkt.notice.service);
		pkt.notice.instance = le32_to_cpu(pkt.notice.instance);
		pkt.notice.node = le32_to_cpu(pkt.notice.node);
		pkt.notice.port = le32_to_cpu(pkt.notice.port);

		cb(udata, pkt.notice.service, pkt.notice.instance,
			pkt.notice.node, pkt.notice.port);
	}

	if (len < 0) {
		LOGE_errno("recv()");
		return -1;
	}

	return 0;
}