Branch data     Line data    Source code

       1                 :             : /*-------------------------------------------------------------------------
       2                 :             :  *
       3                 :             :  * username.c
       4                 :             :  *        get user name
       5                 :             :  *
       6                 :             :  * Portions Copyright (c) 1996-2026, PostgreSQL Global Development Group
       7                 :             :  * Portions Copyright (c) 1994, Regents of the University of California
       8                 :             :  *
       9                 :             :  * IDENTIFICATION
      10                 :             :  *        src/common/username.c
      11                 :             :  *
      12                 :             :  *-------------------------------------------------------------------------
      13                 :             :  */
      14                 :             : 
      15                 :             : #ifndef FRONTEND
      16                 :             : #include "postgres.h"
      17                 :             : #else
      18                 :             : #include "postgres_fe.h"
      19                 :             : #endif
      20                 :             : 
      21                 :             : #include <pwd.h>
      22                 :             : #include <unistd.h>
      23                 :             : 
      24                 :             : #include "common/username.h"
      25                 :             : 
      26                 :             : /*
      27                 :             :  * Returns the current user name in a static buffer
      28                 :             :  * On error, returns NULL and sets *errstr to point to a palloc'd message
      29                 :             :  */
      30                 :             : const char *
      31                 :           2 : get_user_name(char **errstr)
      32                 :             : {
      33                 :             : #ifndef WIN32
      34                 :           2 :         struct passwd *pw;
      35                 :           2 :         uid_t           user_id = geteuid();
      36                 :             : 
      37                 :           2 :         *errstr = NULL;
      38                 :             : 
      39                 :           2 :         errno = 0;                                      /* clear errno before call */
      40                 :           2 :         pw = getpwuid(user_id);
      41         [ +  - ]:           2 :         if (!pw)
      42                 :             :         {
      43                 :           0 :                 *errstr = psprintf(_("could not look up effective user ID %ld: %s"),
      44                 :           0 :                                                    (long) user_id,
      45         [ #  # ]:           0 :                                                    errno ? strerror(errno) : _("user does not exist"));
      46                 :           0 :                 return NULL;
      47                 :             :         }
      48                 :             : 
      49                 :           2 :         return pw->pw_name;
      50                 :             : #else
      51                 :             :         /* Microsoft recommends buffer size of UNLEN+1, where UNLEN = 256 */
      52                 :             :         /* "static" variable remains after function exit */
      53                 :             :         static char username[256 + 1];
      54                 :             :         DWORD           len = sizeof(username);
      55                 :             : 
      56                 :             :         *errstr = NULL;
      57                 :             : 
      58                 :             :         if (!GetUserName(username, &len))
      59                 :             :         {
      60                 :             :                 *errstr = psprintf(_("user name lookup failure: error code %lu"),
      61                 :             :                                                    GetLastError());
      62                 :             :                 return NULL;
      63                 :             :         }
      64                 :             : 
      65                 :             :         return username;
      66                 :             : #endif
      67                 :           2 : }
      68                 :             : 
      69                 :             : 
      70                 :             : /*
      71                 :             :  * Returns the current user name in a static buffer or exits
      72                 :             :  */
      73                 :             : const char *
      74                 :           2 : get_user_name_or_exit(const char *progname)
      75                 :             : {
      76                 :           2 :         const char *user_name;
      77                 :           2 :         char       *errstr;
      78                 :             : 
      79                 :           2 :         user_name = get_user_name(&errstr);
      80                 :             : 
      81         [ +  - ]:           2 :         if (!user_name)
      82                 :             :         {
      83                 :           0 :                 fprintf(stderr, "%s: %s\n", progname, errstr);
      84                 :           0 :                 exit(1);
      85                 :             :         }
      86                 :           4 :         return user_name;
      87                 :           2 : }