/*      $NetBSD: extattr.c,v 1.5 2017/03/09 11:39:41 maya Exp $ */

/*-
* Copyright (c) 2001 Robert N. M. Watson
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. 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.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR 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 AUTHOR 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.
*/

/*
* TrustedBSD: Utility functions for extended attributes.
*/

#include <sys/cdefs.h>
#if defined(LIBC_SCCS) && !defined(lint)
__RCSID("$NetBSD: extattr.c,v 1.5 2017/03/09 11:39:41 maya Exp $");
#endif /* LIBC_SCCS and not lint */

#include "namespace.h"
#include <sys/types.h>
#include <sys/param.h>
#include <sys/extattr.h>

#include <errno.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>

const int extattr_namespaces[] = {
       EXTATTR_NAMESPACE_USER,
       EXTATTR_NAMESPACE_SYSTEM,
       0,
};

int
extattr_namespace_to_string(int attrnamespace, char **string)
{

       switch(attrnamespace) {
       case EXTATTR_NAMESPACE_USER:
               if (string != NULL) {
                       if ((*string =
                            strdup(EXTATTR_NAMESPACE_USER_STRING)) == NULL)
                               return (-1);
               }
               return (0);

       case EXTATTR_NAMESPACE_SYSTEM:
               if (string != NULL)
                       if ((*string =
                            strdup(EXTATTR_NAMESPACE_SYSTEM_STRING)) == NULL)
                               return (-1);
               return (0);

       default:
               errno = EINVAL;
               return (-1);
       }
}

int
extattr_string_to_namespace(const char *string, int *attrnamespace)
{

       if (strcmp(string, EXTATTR_NAMESPACE_USER_STRING) == 0) {
               if (attrnamespace != NULL)
                       *attrnamespace = EXTATTR_NAMESPACE_USER;
               return (0);
       } else if (strcmp(string, EXTATTR_NAMESPACE_SYSTEM_STRING) == 0) {
               if (attrnamespace != NULL)
                       *attrnamespace = EXTATTR_NAMESPACE_SYSTEM;
               return (0);
       } else {
               errno = EINVAL;
               return (-1);
       }
}


int
extattr_copy_fd(int from_fd, int to_fd, int namespace)
{
       ssize_t llen, vlen, maxvlen;
       size_t alen;
       void *alist = NULL;
       void *aval = NULL;
       size_t i;
       int error = -1;

       llen = extattr_list_fd(from_fd, namespace, NULL, 0);
       if (llen == -1) {
               /* Silently ignore when EA are not supported */
               if (errno == EOPNOTSUPP)
                       error = 0;
               goto out;
       }

       if (llen == 0) {
               error = 0;
               goto out;
       }

       if ((alist = malloc((size_t)llen)) == NULL)
               goto out;

       llen = extattr_list_fd(from_fd, namespace, alist, (size_t)llen);
       if (llen == -1)
               goto out;

       maxvlen = 1024;
       if ((aval = malloc((size_t)maxvlen)) == NULL)
               goto out;

       for (i = 0; i < (size_t)llen; i += alen + 1) {
               char aname[NAME_MAX + 1];
               char *ap;

               alen = ((uint8_t *)alist)[i];
               ap = ((char *)alist) + i + 1;
               (void)memcpy(aname, ap, alen);
               aname[alen] = '\0';

               vlen = extattr_get_fd(from_fd, namespace, aname, NULL, 0);
               if (vlen == -1)
                       goto out;

               if (vlen > maxvlen) {
                       if ((aval = realloc(aval, (size_t)vlen)) == NULL)
                               goto out;
                       maxvlen = vlen;
               }

               if ((vlen = extattr_get_fd(from_fd, namespace, aname,
                                          aval, (size_t)vlen)) == -1)
                       goto out;

               if (extattr_set_fd(to_fd, namespace, aname,
                                  aval, (size_t)vlen) != vlen)
                       goto out;
       }

       error = 0;
out:
       free(aval);
       free(alist);

       return error;
}

int
extattr_copy_file(const char *from, const char *to, int namespace)
{
       ssize_t llen, vlen, maxvlen;
       size_t alen;
       void *alist = NULL;
       void *aval = NULL;
       size_t i;
       int error = -1;

       llen = extattr_list_file(from, namespace, NULL, 0);
       if (llen == -1) {
               /* Silently ignore when EA are not supported */
               if (errno == EOPNOTSUPP)
                       error = 0;
               goto out;
       }

       if (llen == 0) {
               error = 0;
               goto out;
       }

       if ((alist = malloc((size_t)llen)) == NULL)
               goto out;

       llen = extattr_list_file(from, namespace, alist, (size_t)llen);
       if (llen == -1)
               goto out;

       maxvlen = 1024;
       if ((aval = malloc((size_t)maxvlen)) == NULL)
               goto out;

       for (i = 0; i < (size_t)llen; i += alen + 1) {
               char aname[NAME_MAX + 1];
               char *ap;

               alen = ((uint8_t *)alist)[i];
               ap = ((char *)alist) + i + 1;
               (void)memcpy(aname, ap, alen);
               aname[alen] = '\0';

               vlen = extattr_get_file(from, namespace, aname, NULL, 0);
               if (vlen == -1)
                       goto out;

               if (vlen > maxvlen) {
                       if ((aval = realloc(aval, (size_t)vlen)) == NULL)
                               goto out;
                       maxvlen = vlen;
               }

               if ((vlen = extattr_get_file(from, namespace, aname,
                                               aval, (size_t)vlen)) == -1)
                       goto out;

               if (extattr_set_file(to, namespace, aname,
                                    aval, (size_t)vlen) != vlen)
                       goto out;
       }

       error = 0;
out:
       free(aval);
       free(alist);

       return error;
}

int
extattr_copy_link(const char *from, const char *to, int namespace)
{
       ssize_t llen, vlen, maxvlen;
       size_t alen;
       void *alist = NULL;
       void *aval = NULL;
       size_t i;
       int error = -1;

       llen = extattr_list_link(from, namespace, NULL, 0);
       if (llen == -1) {
               /* Silently ignore when EA are not supported */
               if (errno == EOPNOTSUPP)
                       error = 0;
               goto out;
       }

       if (llen == 0) {
               error = 0;
               goto out;
       }

       if ((alist = malloc((size_t)llen)) == NULL)
               goto out;

       llen = extattr_list_link(from, namespace, alist, (size_t)llen);
       if (llen == -1)
               goto out;

       maxvlen = 1024;
       if ((aval = malloc((size_t)maxvlen)) == NULL)
               goto out;

       for (i = 0; i < (size_t)llen; i += alen + 1) {
               char aname[NAME_MAX + 1];
               char *ap;

               alen = ((uint8_t *)alist)[i];
               ap = ((char *)alist) + i + 1;
               (void)memcpy(aname, ap, alen);
               aname[alen] = '\0';

               vlen = extattr_get_link(from, namespace, aname, NULL, 0);
               if (vlen == -1)
                       goto out;

               if (vlen > maxvlen) {
                       if ((aval = realloc(aval, (size_t)vlen)) == NULL)
                               goto out;
                       maxvlen = vlen;
               }

               if ((vlen = extattr_get_link(from, namespace, aname,
                                            aval, (size_t)vlen)) == -1)
                       goto out;

               if (extattr_set_link(to, namespace, aname,
                                    aval, (size_t)vlen) != vlen)
                       goto out;
       }

       error = 0;
out:
       free(aval);
       free(alist);

       return error;
}

static int
extattr_namespace_access(int namespace, int mode)
{
       switch (namespace) {
       case EXTATTR_NAMESPACE_SYSTEM:
               if ((mode & (R_OK|W_OK)) && getuid() != 0)
                       return -1;
               break;
       default:
               break;
       }

       return 0;
}

int
fcpxattr(int from_fd, int to_fd)
{
       const int *ns;
       int error;

       for (ns = extattr_namespaces; *ns; ns++) {
               if (extattr_namespace_access(*ns, R_OK|W_OK) != 0)
                       continue;

               if ((error = extattr_copy_fd(from_fd, to_fd, *ns)) != 0)
                       return error;
       }

       return 0;
}

int
cpxattr(const char *from, const char *to)
{
       const int *ns;
       int error;

       for (ns = extattr_namespaces; *ns; ns++) {
               if (extattr_namespace_access(*ns, R_OK|W_OK) != 0)
                       continue;

               if ((error = extattr_copy_file(from, to, *ns)) != 0)
                       return error;
       }

       return 0;
}

int
lcpxattr(const char *from, const char *to)
{
       const int *ns;
       int error;

       for (ns = extattr_namespaces; *ns; ns++) {
               if (extattr_namespace_access(*ns, R_OK|W_OK) != 0)
                       continue;

               if ((error = extattr_copy_link(from, to, *ns)) != 0)
                       return error;
       }

       return 0;
}