1*4887Schin /***********************************************************************
2*4887Schin *                                                                      *
3*4887Schin *               This software is part of the ast package               *
4*4887Schin *           Copyright (c) 1985-2007 AT&T Knowledge Ventures            *
5*4887Schin *                      and is licensed under the                       *
6*4887Schin *                  Common Public License, Version 1.0                  *
7*4887Schin *                      by AT&T Knowledge Ventures                      *
8*4887Schin *                                                                      *
9*4887Schin *                A copy of the License is available at                 *
10*4887Schin *            http://www.opensource.org/licenses/cpl1.0.txt             *
11*4887Schin *         (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9)         *
12*4887Schin *                                                                      *
13*4887Schin *              Information and Software Systems Research               *
14*4887Schin *                            AT&T Research                             *
15*4887Schin *                           Florham Park NJ                            *
16*4887Schin *                                                                      *
17*4887Schin *                 Glenn Fowler <gsf@research.att.com>                  *
18*4887Schin *                  David Korn <dgk@research.att.com>                   *
19*4887Schin *                   Phong Vo <kpv@research.att.com>                    *
20*4887Schin *                                                                      *
21*4887Schin ***********************************************************************/
22*4887Schin #pragma prototyped
23*4887Schin /*
24*4887Schin  * Glenn Fowler
25*4887Schin  * AT&T Bell Laboratories
26*4887Schin  *
27*4887Schin  * uid name -> number
28*4887Schin  */
29*4887Schin 
30*4887Schin #if defined(__STDPP__directive) && defined(__STDPP__hide)
31*4887Schin __STDPP__directive pragma pp:hide getpwnam getpwuid
32*4887Schin #else
33*4887Schin #define getpwnam	______getpwnam
34*4887Schin #define getpwuid	______getpwuid
35*4887Schin #endif
36*4887Schin 
37*4887Schin #include <ast.h>
38*4887Schin #include <cdt.h>
39*4887Schin #include <pwd.h>
40*4887Schin 
41*4887Schin #if defined(__STDPP__directive) && defined(__STDPP__hide)
42*4887Schin __STDPP__directive pragma pp:nohide getpwnam getpwuid
43*4887Schin #else
44*4887Schin #undef	getpwnam
45*4887Schin #undef	getpwuid
46*4887Schin #endif
47*4887Schin 
48*4887Schin extern struct passwd*	getpwnam(const char*);
49*4887Schin extern struct passwd*	getpwuid(uid_t);
50*4887Schin 
51*4887Schin typedef struct Id_s
52*4887Schin {
53*4887Schin 	Dtlink_t	link;
54*4887Schin 	int		id;
55*4887Schin 	char		name[1];
56*4887Schin } Id_t;
57*4887Schin 
58*4887Schin /*
59*4887Schin  * return uid number given uid name
60*4887Schin  * -1 on first error for a given name
61*4887Schin  * -2 on subsequent errors for a given name
62*4887Schin  */
63*4887Schin 
64*4887Schin int
65*4887Schin struid(const char* name)
66*4887Schin {
67*4887Schin 	register Id_t*		ip;
68*4887Schin 	register struct passwd*	pw;
69*4887Schin 	int			id;
70*4887Schin 	char*			e;
71*4887Schin 
72*4887Schin 	static Dt_t*		dict;
73*4887Schin 	static Dtdisc_t		disc;
74*4887Schin 
75*4887Schin 	if (!dict)
76*4887Schin 	{
77*4887Schin 		disc.key = offsetof(Id_t, name);
78*4887Schin 		dict = dtopen(&disc, Dthash);
79*4887Schin 	}
80*4887Schin 	else if (ip = (Id_t*)dtmatch(dict, name))
81*4887Schin 		return ip->id;
82*4887Schin 	if (pw = getpwnam(name))
83*4887Schin 		id = pw->pw_uid;
84*4887Schin 	else
85*4887Schin 	{
86*4887Schin 		id = strtol(name, &e, 0);
87*4887Schin #if _WINIX
88*4887Schin 		if (!*e)
89*4887Schin 		{
90*4887Schin 			if (!getpwuid(id))
91*4887Schin 				id = -1;
92*4887Schin 		}
93*4887Schin 		else if (streq(name, "root") && (pw = getpwnam("Administrator")))
94*4887Schin 			id = pw->pw_uid;
95*4887Schin 		else
96*4887Schin 			id = -1;
97*4887Schin #else
98*4887Schin 		if (*e || !getpwuid(id))
99*4887Schin 			id = -1;
100*4887Schin #endif
101*4887Schin 	}
102*4887Schin 	if (dict && (ip = newof(0, Id_t, 1, strlen(name))))
103*4887Schin 	{
104*4887Schin 		strcpy(ip->name, name);
105*4887Schin 		ip->id = id >= 0 ? id : -2;
106*4887Schin 		dtinsert(dict, ip);
107*4887Schin 	}
108*4887Schin 	return id;
109*4887Schin }
110