1 /*- 2 * SPDX-License-Identifier: BSD-3-Clause 3 * 4 * Copyright (c) 1993, John Brezak 5 * All rights reserved. 6 * 7 * Redistribution and use in source and binary forms, with or without 8 * modification, are permitted provided that the following conditions 9 * are met: 10 * 1. Redistributions of source code must retain the above copyright 11 * notice, this list of conditions and the following disclaimer. 12 * 2. Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in the 14 * documentation and/or other materials provided with the distribution. 15 * 3. Neither the name of the University nor the names of its contributors 16 * may be used to endorse or promote products derived from this software 17 * without specific prior written permission. 18 * 19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 29 * SUCH DAMAGE. 30 * 31 * $FreeBSD: head/libexec/rpc.rusersd/rusers_proc.c 326025 2017-11-20 19:49:47Z pfg $ 32 */ 33 34 #ifdef DEBUG 35 #include <errno.h> 36 #endif 37 #include <stdio.h> 38 #include <stdlib.h> 39 #include <string.h> 40 #include <sys/param.h> 41 #include <sys/stat.h> 42 #include <syslog.h> 43 #include <utmpx.h> 44 #ifdef XIDLE 45 #include <setjmp.h> 46 #include <X11/Xlib.h> 47 #include <X11/extensions/xidle.h> 48 #endif 49 #include <rpcsvc/rnusers.h> 50 51 #include "extern.h" 52 53 #ifndef _PATH_DEV 54 #define _PATH_DEV "/dev" 55 #endif 56 57 static utmpidle utmp_idle[MAXUSERS]; 58 static utmp old_utmp[MAXUSERS]; 59 static struct utmpx utmp_list[MAXUSERS]; 60 61 #ifdef XIDLE 62 static Display *dpy; 63 64 static jmp_buf openAbort; 65 66 static void 67 abortOpen(void) 68 { 69 longjmp (openAbort, 1); 70 } 71 72 XqueryIdle(char *display) 73 { 74 int first_event, first_error; 75 Time IdleTime; 76 77 (void) signal (SIGALRM, abortOpen); 78 (void) alarm ((unsigned) 10); 79 if (!setjmp (openAbort)) { 80 if (!(dpy= XOpenDisplay(display))) { 81 syslog(LOG_ERR, "Cannot open display %s", display); 82 return(-1); 83 } 84 if (XidleQueryExtension(dpy, &first_event, &first_error)) { 85 if (!XGetIdleTime(dpy, &IdleTime)) { 86 syslog(LOG_ERR, "%s: unable to get idle time", display); 87 return(-1); 88 } 89 } else { 90 syslog(LOG_ERR, "%s: Xidle extension not loaded", display); 91 return(-1); 92 } 93 XCloseDisplay(dpy); 94 } else { 95 syslog(LOG_ERR, "%s: server grabbed for over 10 seconds", display); 96 return(-1); 97 } 98 (void) signal (SIGALRM, SIG_DFL); 99 (void) alarm ((unsigned) 0); 100 101 IdleTime /= 1000; 102 return((IdleTime + 30) / 60); 103 } 104 #endif 105 106 static u_int 107 getidle(const char *tty, const char *display __unused) 108 { 109 struct stat st; 110 char ttyname[PATH_MAX]; 111 time_t now; 112 u_long idle; 113 114 /* 115 * If this is an X terminal or console, then try the 116 * XIdle extension 117 */ 118 #ifdef XIDLE 119 if (display && *display && (idle = XqueryIdle(display)) >= 0) 120 return(idle); 121 #endif 122 idle = 0; 123 if (*tty == 'X') { 124 u_long kbd_idle, mouse_idle; 125 #if !defined(__DragonFly__) && !defined(__FreeBSD__) 126 kbd_idle = getidle("kbd", NULL); 127 #else 128 kbd_idle = getidle("vga", NULL); 129 #endif 130 mouse_idle = getidle("mouse", NULL); 131 idle = (kbd_idle < mouse_idle)?kbd_idle:mouse_idle; 132 } else { 133 sprintf(ttyname, "%s/%s", _PATH_DEV, tty); 134 if (stat(ttyname, &st) < 0) { 135 #ifdef DEBUG 136 printf("%s: %s\n", ttyname, strerror(errno)); 137 #endif 138 return(-1); 139 } 140 time(&now); 141 #ifdef DEBUG 142 printf("%s: now=%d atime=%d\n", ttyname, now, 143 st.st_atime); 144 #endif 145 idle = now - st.st_atime; 146 idle = (idle + 30) / 60; /* secs->mins */ 147 } 148 149 return(idle); 150 } 151 152 static utmpidlearr * 153 do_names_2(void) 154 { 155 static utmpidlearr ut; 156 struct utmpx *usr; 157 int nusers = 0; 158 159 memset(&ut, 0, sizeof(ut)); 160 ut.utmpidlearr_val = &utmp_idle[0]; 161 162 setutxent(); 163 while ((usr = getutxent()) != NULL && nusers < MAXUSERS) { 164 if (usr->ut_type != USER_PROCESS) 165 continue; 166 167 memcpy(&utmp_list[nusers], usr, sizeof(*usr)); 168 utmp_idle[nusers].ui_utmp.ut_time = usr->ut_tv.tv_sec; 169 utmp_idle[nusers].ui_idle = 170 getidle(usr->ut_line, usr->ut_host); 171 utmp_idle[nusers].ui_utmp.ut_line = 172 utmp_list[nusers].ut_line; 173 utmp_idle[nusers].ui_utmp.ut_name = 174 utmp_list[nusers].ut_user; 175 utmp_idle[nusers].ui_utmp.ut_host = 176 utmp_list[nusers].ut_host; 177 178 nusers++; 179 } 180 endutxent(); 181 182 ut.utmpidlearr_len = nusers; 183 return(&ut); 184 } 185 186 static int * 187 rusers_num(void *argp __unused, struct svc_req *rqstp __unused) 188 { 189 static int num_users = 0; 190 struct utmpx *usr; 191 192 setutxent(); 193 while ((usr = getutxent()) != NULL) { 194 if (usr->ut_type != USER_PROCESS) 195 continue; 196 num_users++; 197 } 198 endutxent(); 199 200 return(&num_users); 201 } 202 203 static utmparr * 204 do_names_1(void) 205 { 206 utmpidlearr *utidle; 207 static utmparr ut; 208 unsigned int i; 209 210 bzero((char *)&ut, sizeof(ut)); 211 212 utidle = do_names_2(); 213 if (utidle) { 214 ut.utmparr_len = utidle->utmpidlearr_len; 215 ut.utmparr_val = &old_utmp[0]; 216 for (i = 0; i < ut.utmparr_len; i++) 217 bcopy(&utmp_idle[i].ui_utmp, &old_utmp[i], 218 sizeof(old_utmp[0])); 219 220 } 221 222 return(&ut); 223 } 224 225 utmpidlearr * 226 rusersproc_names_2_svc(void *argp __unused, struct svc_req *rqstp __unused) 227 { 228 return (do_names_2()); 229 } 230 231 utmpidlearr * 232 rusersproc_allnames_2_svc(void *argp __unused, struct svc_req *rqstp __unused) 233 { 234 return (do_names_2()); 235 } 236 237 utmparr * 238 rusersproc_names_1_svc(void *argp __unused, struct svc_req *rqstp __unused) 239 { 240 return (do_names_1()); 241 } 242 243 utmparr * 244 rusersproc_allnames_1_svc(void *argp __unused, struct svc_req *rqstp __unused) 245 { 246 return (do_names_1()); 247 } 248 249 typedef void *(*rusersproc_t)(void *, struct svc_req *); 250 251 void 252 rusers_service(struct svc_req *rqstp, SVCXPRT *transp) 253 { 254 union { 255 int fill; 256 } argument; 257 char *result; 258 xdrproc_t xdr_argument, xdr_result; 259 rusersproc_t local; 260 261 switch (rqstp->rq_proc) { 262 case NULLPROC: 263 (void)svc_sendreply(transp, (xdrproc_t)xdr_void, NULL); 264 goto leave; 265 266 case RUSERSPROC_NUM: 267 xdr_argument = (xdrproc_t)xdr_void; 268 xdr_result = (xdrproc_t)xdr_int; 269 local = (rusersproc_t)rusers_num; 270 break; 271 272 case RUSERSPROC_NAMES: 273 xdr_argument = (xdrproc_t)xdr_void; 274 xdr_result = (xdrproc_t)xdr_utmpidlearr; 275 switch (rqstp->rq_vers) { 276 case RUSERSVERS_ORIG: 277 local = (rusersproc_t)rusersproc_names_1_svc; 278 break; 279 case RUSERSVERS_IDLE: 280 local = (rusersproc_t)rusersproc_names_2_svc; 281 break; 282 default: 283 svcerr_progvers(transp, RUSERSVERS_ORIG, RUSERSVERS_IDLE); 284 goto leave; 285 /*NOTREACHED*/ 286 } 287 break; 288 289 case RUSERSPROC_ALLNAMES: 290 xdr_argument = (xdrproc_t)xdr_void; 291 xdr_result = (xdrproc_t)xdr_utmpidlearr; 292 switch (rqstp->rq_vers) { 293 case RUSERSVERS_ORIG: 294 local = (rusersproc_t)rusersproc_allnames_1_svc; 295 break; 296 case RUSERSVERS_IDLE: 297 local = (rusersproc_t)rusersproc_allnames_2_svc; 298 break; 299 default: 300 svcerr_progvers(transp, RUSERSVERS_ORIG, RUSERSVERS_IDLE); 301 goto leave; 302 /*NOTREACHED*/ 303 } 304 break; 305 306 default: 307 svcerr_noproc(transp); 308 goto leave; 309 } 310 bzero(&argument, sizeof(argument)); 311 if (!svc_getargs(transp, (xdrproc_t)xdr_argument, &argument)) { 312 svcerr_decode(transp); 313 goto leave; 314 } 315 result = (*local)(&argument, rqstp); 316 if (result != NULL && 317 !svc_sendreply(transp, (xdrproc_t)xdr_result, result)) { 318 svcerr_systemerr(transp); 319 } 320 if (!svc_freeargs(transp, (xdrproc_t)xdr_argument, &argument)) { 321 syslog(LOG_ERR, "unable to free arguments"); 322 exit(1); 323 } 324 leave: 325 if (from_inetd) 326 exit(0); 327 } 328