/***************************************************************************
 *
 * util_helper.c - HAL utilities for helper (as e.g. prober/addons) et al.
 *
 * Copyright (C) 2006 David Zeuthen, <david@fubar.dk>
 *
 * Licensed under the Academic Free License version 2.1
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 **************************************************************************/

#ifdef HAVE_CONFIG_H
#  include <config.h>
#endif

#include <grp.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <sys/time.h>
#include <time.h>
#include <pwd.h>
#include <unistd.h>

#include "logger.h"

#include "util_helper.h"

#ifdef __linux__
extern char **environ;
#endif

static char **argv_buffer = NULL;
static size_t argv_size = 0;

#ifdef sun
#include <priv.h>
void
drop_privileges(int keep_auxgroups)
{
        priv_set_t *pPrivSet;

        /*
         * Start with the 'basic' privilege set and then remove any
         * of the 'basic' privileges that will not be needed.
         */
        if ((pPrivSet = priv_allocset()) == NULL) {
                return;
        }

        /*
         * Establish the basic set of privileges.
         * Note: fork/exec required for libdevinfo devlink
         * interfaces are included in the basic set.
         */
        priv_basicset(pPrivSet);

        /* Clear privileges we will not need from the 'basic' set */
        (void) priv_delset(pPrivSet, PRIV_FILE_LINK_ANY);
        (void) priv_delset(pPrivSet, PRIV_PROC_INFO);
        (void) priv_delset(pPrivSet, PRIV_PROC_SESSION);

        /* for sysevent need to be root and have this privilege */
        (void) priv_addset(pPrivSet, PRIV_SYS_CONFIG);

        /* need proc_audit privilege */
        (void) priv_addset(pPrivSet, PRIV_PROC_AUDIT);

        /* Set the permitted privilege set. */
        (void) setppriv(PRIV_SET, PRIV_PERMITTED, pPrivSet);

        /* Set the limit privilege set. */
        (void) setppriv(PRIV_SET, PRIV_LIMIT, pPrivSet);

        priv_freeset(pPrivSet);
}
#else /* !sun */

/** Drop root privileges: Set the running user id to HAL_USER and
 *  group to HAL_GROUP, and optionally retain auxiliary groups of HAL_USER.
 */
void
drop_privileges (int keep_auxgroups)
{
        struct passwd *pw = NULL;
        struct group *gr = NULL;

        /* determine user id */
        pw = getpwnam (HAL_USER);
        if (!pw)  {
                HAL_DEBUG (("drop_privileges: user " HAL_USER " does not exist"));
                exit (-1);
        }

        /* determine primary group id */
        gr = getgrnam (HAL_GROUP);
        if (!gr) {
                HAL_DEBUG (("drop_privileges: group " HAL_GROUP " does not exist"));
                exit (-1);
        }

        if (keep_auxgroups) {
                if (initgroups (HAL_USER, gr->gr_gid)) {
                        HAL_DEBUG(("drop_privileges: could not initialize groups"));
                        exit (-1);
                }
        }

        if (setgid (gr->gr_gid)) {
                HAL_DEBUG (("drop_privileges: could not set group id"));
                exit (-1);
        }

        if (setuid (pw->pw_uid)) {
                HAL_DEBUG (("drop_privileges: could not set user id"));
                exit (-1);
        }
}
#endif /* !sun */

void
hal_set_proc_title_init (int argc, char *argv[])
{
#ifdef __linux__
        unsigned int i;
        char **new_environ, *endptr;

        /* This code is really really ugly. We make some memory layout
         * assumptions and reuse the environment array as memory to store
         * our process title in */

        for (i = 0; environ[i] != NULL; i++)
                ;

        endptr = i ? environ[i-1] + strlen (environ[i-1]) : argv[argc-1] + strlen (argv[argc-1]);

        argv_buffer = argv;
        argv_size = endptr - argv_buffer[0];

        /* Make a copy of environ */

        new_environ = malloc (sizeof(char*) * (i + 1));
        for (i = 0; environ[i] != NULL; i++)
                new_environ[i] = strdup (environ[i]);
        new_environ[i] = NULL;

        environ = new_environ;
#endif
}

/* this code borrowed from avahi-daemon's setproctitle.c (LGPL v2) */
void
hal_set_proc_title (const char *format, ...)
{
#ifdef __linux__
        size_t len;
        va_list ap;

        if (argv_buffer == NULL)
                goto out;

        va_start (ap, format);
        vsnprintf (argv_buffer[0], argv_size, format, ap);
        va_end (ap);

        len = strlen (argv_buffer[0]);

        memset (argv_buffer[0] + len, 0, argv_size - len);
        argv_buffer[1] = NULL;
out:
        ;
#endif
}