kc3-lang/gnulib/tests/test-idpriv-drop.c

• Show log

  Commit

• Hash : 7b089321
  Author :
  Date : 2025-01-01T09:24:36
  

  maint: run 'make update-copyright'
  

Download

• tests/test-idpriv-drop.c

• /* Test of dropping uid/gid privileges of the current process permanently.
     Copyright (C) 2009-2025 Free Software Foundation, Inc.
  
     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 3 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, see <https://www.gnu.org/licenses/>.  */
  
  #include <config.h>
  
  #include "idpriv.h"
  
  #include <stdio.h>
  #include <stdlib.h>
  #include <string.h>
  #include <unistd.h>
  
  #include "macros.h"
  
  static void
  show_uids ()
  {
  #if HAVE_GETRESUID /* glibc, FreeBSD, OpenBSD, HP-UX */
    uid_t real;
    uid_t effective;
    uid_t saved;
    ASSERT (getresuid (&real, &effective, &saved) >= 0);
    printf ("uids: real=%d effective=%d saved=%d",
            (int) real, (int) effective, (int) saved);
  #elif HAVE_GETEUID
    printf ("uids: real=%d effective=%d",
            (int) getuid (), (int) geteuid ());
  #elif HAVE_GETUID
    printf ("uids: real=%d",
            (int) getuid ());
  #endif
  }
  
  static void
  show_gids ()
  {
  #if HAVE_GETRESGID /* glibc, FreeBSD, OpenBSD, HP-UX */
    gid_t real;
    gid_t effective;
    gid_t saved;
    ASSERT (getresgid (&real, &effective, &saved) >= 0);
    printf ("gids: real=%d effective=%d saved=%d",
            (int) real, (int) effective, (int) saved);
  #elif HAVE_GETEGID
    printf ("gids: real=%d effective=%d",
            (int) getgid (), (int) getegid ());
  #elif HAVE_GETGID
    printf ("gids: real=%d",
            (int) getgid ());
  #endif
  }
  
  static void
  show (const char *prefix)
  {
    printf ("%s  ", prefix);
    show_uids ();
    printf ("  ");
    show_gids ();
    printf ("\n");
  }
  
  int
  main (int argc, char *argv[])
  {
    bool verbose = false;
    int i;
  
  #if HAVE_GETUID
    int uid = getuid ();
  #endif
  #if HAVE_GETGID
    int gid = getgid ();
  #endif
  
    /* Parse arguments.
       -v  enables verbose output.
     */
    for (i = 1; i < argc; i++)
      {
        const char *arg = argv[i];
        if (strcmp (arg, "-v") == 0)
          verbose = true;
      }
  
    if (verbose)
      show ("before drop:");
  
    ASSERT (idpriv_drop () == 0);
  
    if (verbose)
      show ("after drop: ");
  
    /* Verify that the privileges have really been dropped.  */
  #if HAVE_GETRESUID /* glibc, FreeBSD, OpenBSD, HP-UX */
    {
      uid_t real;
      uid_t effective;
      uid_t saved;
      if (getresuid (&real, &effective, &saved) < 0
          || real != uid
          || effective != uid
          || saved != uid)
        abort ();
    }
  #else
  # if HAVE_GETEUID
    if (geteuid () != uid)
      abort ();
  # endif
  # if HAVE_GETUID
    if (getuid () != uid)
      abort ();
  # endif
  #endif
  #if HAVE_GETRESGID /* glibc, FreeBSD, OpenBSD, HP-UX */
    {
      gid_t real;
      gid_t effective;
      gid_t saved;
      if (getresgid (&real, &effective, &saved) < 0
          || real != gid
          || effective != gid
          || saved != gid)
        abort ();
    }
  #else
  # if HAVE_GETEGID
    if (getegid () != gid)
      abort ();
  # endif
  # if HAVE_GETGID
    if (getgid () != gid)
      abort ();
  # endif
  #endif
  
    return test_exit_status;
  }
  

Download