/*

 * 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 OS_UNIX

#include <unistd.h>

#endif

#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

}