/*

 * Copyright © 2020 Amazon.com, Inc. or its affiliates.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 *     * Redistributions of source code must retain the above

 *       copyright notice, this list of conditions and the

 *       following disclaimer.

 *     * Redistributions in binary form must reproduce the

 *       above copyright notice, this list of conditions and

 *       the following disclaimer in the documentation and/or

 *       other materials provided with the distribution.

 *     * The names of contributors to this software may not be

 *       used to endorse or promote products derived from this

 *       software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS

 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE

 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,

 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,

 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS

 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,

 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF

 * THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH

 * DAMAGE.

 *

 * Author: David Woodhouse <dwmw2@infradead.org>

 */

#include "config.h"

#include "vsock.h"

#include <limits.h>

#include <string.h>

#include <stdlib.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#ifdef HAVE_VSOCK

#include <sys/socket.h>

#include <linux/vm_sockets.h>

#include <sys/ioctl.h>

#endif

/* This generic parsing utility doesn't actually require the

 * vm_sockets.h header and thus doesn't require conditional

 * compiliation... except for this one definition. */

#ifndef VMADDR_CID_ANY

#define VMADDR_CID_ANY -1U

#endif

bool

p11_vsock_parse_addr (const char *target,

		      unsigned int *cid,

		      unsigned int *port)

{

	bool cid_found = false;

	bool port_found = false;

	unsigned long val;

	char *endptr;

	while (*target) {

		if (strncmp (target, "cid=", 4) == 0) {

			val = strtoul(target + 4, &endptr, 0);

			if (val > UINT_MAX || endptr == target + 4)

				return false;

			*cid = val;

			cid_found = true;

		} else if (strncmp (target, "port=", 5) == 0) {

			val = strtoul (target + 5, &endptr, 0);

			if (val > UINT_MAX || endptr == target + 5)

				return false;

			*port = val;

			port_found = true;

		} else {

			return false;

		}

		target = endptr;

		if (*target == ';')

			target++;

		else if (*target)

			return false;

	}

	/* Port is mandatory */

	if (!port_found)

		return false;

	/* CID is optional, defaulting to VMADDR_CID_ANY */

	if (!cid_found)

		*cid = VMADDR_CID_ANY;

	return true;

}

bool

p11_vsock_get_local_cid (unsigned int *cid)

{

#ifndef HAVE_VSOCK

	return false;

#else

	int fd = open ("/dev/vsock", O_RDONLY);

	int rc;

	if (fd == -1)

		return false;

	rc = ioctl (fd, IOCTL_VM_SOCKETS_GET_LOCAL_CID, cid, sizeof(*cid));

	close (fd);

	return (rc == 0);

#endif

}