libexec/rpc.rusersd/rusers_proc.c

   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
  32 #ifdef DEBUG
  33 #include <errno.h>
  34 #endif
  35 #include <stdio.h>
  36 #include <string.h>
  37 #include <sys/param.h>
  38 #include <sys/stat.h>
  39 #include <stdlib.h>
  40 #include <syslog.h>
  41 #include <utmpx.h>
  42 #ifdef XIDLE
  43 #include <setjmp.h>
  44 #include <X11/Xlib.h>
  45 #include <X11/extensions/xidle.h>
  46 #endif
  47 #include <rpcsvc/rnusers.h>
  48
  49 #include "extern.h"
  50
  51 #ifndef _PATH_DEV
  52 #define _PATH_DEV "/dev"
  53 #endif
  54
  55 static utmpidle utmp_idle[MAXUSERS];
  56 static utmp old_utmp[MAXUSERS];
  57 static struct utmpx utmp_list[MAXUSERS];
  58
  59 #ifdef XIDLE
  60 static Display *dpy;
  61
  62 static jmp_buf openAbort;
  63
  64 static void
  65 abortOpen(void)
  66 {
  67     longjmp (openAbort, 1);
  68 }
  69
  70 XqueryIdle(char *display)
  71 {
  72         int first_event, first_error;
  73         Time IdleTime;
  74
  75         (void) signal (SIGALRM, abortOpen);
  76         (void) alarm ((unsigned) 10);
  77         if (!setjmp (openAbort)) {
  78                 if (!(dpy= XOpenDisplay(display))) {
  79                         syslog(LOG_ERR, "Cannot open display %s", display);
  80                         return(-1);
  81                 }
  82                 if (XidleQueryExtension(dpy, &first_event, &first_error)) {
  83                         if (!XGetIdleTime(dpy, &IdleTime)) {
  84                                 syslog(LOG_ERR, "%s: unable to get idle time", display);
  85                                 return(-1);
  86                         }
  87                 } else {
  88                         syslog(LOG_ERR, "%s: Xidle extension not loaded", display);
  89                         return(-1);
  90                 }
  91                 XCloseDisplay(dpy);
  92         } else {
  93                 syslog(LOG_ERR, "%s: server grabbed for over 10 seconds", display);
  94                 return(-1);
  95         }
  96         (void) signal (SIGALRM, SIG_DFL);
  97         (void) alarm ((unsigned) 0);
  98
  99         IdleTime /= 1000;
 100         return((IdleTime + 30) / 60);
 101 }
 102 #endif
 103
 104 static u_int
 105 getidle(const char *tty, const char *display __unused)
 106 {
 107         struct stat st;
 108         char ttyname[PATH_MAX];
 109         time_t now;
 110         u_long idle;
 111
 112         /*
 113          * If this is an X terminal or console, then try the
 114          * XIdle extension
 115          */
 116 #ifdef XIDLE
 117         if (display && *display && (idle = XqueryIdle(display)) >= 0)
 118                 return(idle);
 119 #endif
 120         idle = 0;
 121         if (*tty == 'X') {
 122                 u_long kbd_idle, mouse_idle;
 123 #if     !defined(__FreeBSD__)
 124                 kbd_idle = getidle("kbd", NULL);
 125 #else
 126                 kbd_idle = getidle("vga", NULL);
 127 #endif
 128                 mouse_idle = getidle("mouse", NULL);
 129                 idle = (kbd_idle < mouse_idle)?kbd_idle:mouse_idle;
 130         } else {
 131                 sprintf(ttyname, "%s/%s", _PATH_DEV, tty);
 132                 if (stat(ttyname, &st) < 0) {
 133 #ifdef DEBUG
 134                         printf("%s: %s\n", ttyname, strerror(errno));
 135 #endif
 136                         return(-1);
 137                 }
 138                 time(&now);
 139 #ifdef DEBUG
 140                 printf("%s: now=%d atime=%d\n", ttyname, now,
 141                        st.st_atime);
 142 #endif
 143                 idle = now - st.st_atime;
 144                 idle = (idle + 30) / 60; /* secs->mins */
 145         }
 146
 147         return(idle);
 148 }
 149
 150 static utmpidlearr *
 151 do_names_2(void)
 152 {
 153         static utmpidlearr ut;
 154         struct utmpx *usr;
 155         int nusers = 0;
 156
 157         memset(&ut, 0, sizeof(ut));
 158         ut.utmpidlearr_val = &utmp_idle[0];
 159
 160         setutxent();
 161         while ((usr = getutxent()) != NULL && nusers < MAXUSERS) {
 162                 if (usr->ut_type != USER_PROCESS)
 163                         continue;
 164
 165                 memcpy(&utmp_list[nusers], usr, sizeof(*usr));
 166                 utmp_idle[nusers].ui_utmp.ut_time = usr->ut_tv.tv_sec;
 167                 utmp_idle[nusers].ui_idle =
 168                     getidle(usr->ut_line, usr->ut_host);
 169                 utmp_idle[nusers].ui_utmp.ut_line =
 170                     utmp_list[nusers].ut_line;
 171                 utmp_idle[nusers].ui_utmp.ut_name =
 172                     utmp_list[nusers].ut_user;
 173                 utmp_idle[nusers].ui_utmp.ut_host =
 174                     utmp_list[nusers].ut_host;
 175
 176                 nusers++;
 177         }
 178         endutxent();
 179
 180         ut.utmpidlearr_len = nusers;
 181         return(&ut);
 182 }
 183
 184 static int *
 185 rusers_num(void *argp __unused, struct svc_req *rqstp __unused)
 186 {
 187         static int num_users = 0;
 188         struct utmpx *usr;
 189
 190         setutxent();
 191         while ((usr = getutxent()) != NULL) {
 192                 if (usr->ut_type != USER_PROCESS)
 193                         continue;
 194                 num_users++;
 195         }
 196         endutxent();
 197
 198         return(&num_users);
 199 }
 200
 201 static utmparr *
 202 do_names_1(void)
 203 {
 204         utmpidlearr *utidle;
 205         static utmparr ut;
 206         unsigned int i;
 207
 208         bzero((char *)&ut, sizeof(ut));
 209
 210         utidle = do_names_2();
 211         if (utidle) {
 212                 ut.utmparr_len = utidle->utmpidlearr_len;
 213                 ut.utmparr_val = &old_utmp[0];
 214                 for (i = 0; i < ut.utmparr_len; i++)
 215                         bcopy(&utmp_idle[i].ui_utmp, &old_utmp[i],
 216                               sizeof(old_utmp[0]));
 217
 218         }
 219
 220         return(&ut);
 221 }
 222
 223 utmpidlearr *
 224 rusersproc_names_2_svc(void *argp __unused, struct svc_req *rqstp __unused)
 225 {
 226
 227         return (do_names_2());
 228 }
 229
 230 utmpidlearr *
 231 rusersproc_allnames_2_svc(void *argp __unused, struct svc_req *rqstp __unused)
 232 {
 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
 241         return (do_names_1());
 242 }
 243
 244 utmparr *
 245 rusersproc_allnames_1_svc(void *argp __unused, struct svc_req *rqstp __unused)
 246 {
 247
 248         return (do_names_1());
 249 }
 250
 251 typedef void *(*rusersproc_t)(void *, struct svc_req *);
 252
 253 void
 254 rusers_service(struct svc_req *rqstp, SVCXPRT *transp)
 255 {
 256         union {
 257                 int fill;
 258         } argument;
 259         char *result;
 260         xdrproc_t xdr_argument, xdr_result;
 261         rusersproc_t local;
 262
 263         switch (rqstp->rq_proc) {
 264         case NULLPROC:
 265                 (void)svc_sendreply(transp, (xdrproc_t)xdr_void, NULL);
 266                 goto leave;
 267
 268         case RUSERSPROC_NUM:
 269                 xdr_argument = (xdrproc_t)xdr_void;
 270                 xdr_result = (xdrproc_t)xdr_int;
 271                 local = (rusersproc_t)rusers_num;
 272                 break;
 273
 274         case RUSERSPROC_NAMES:
 275                 xdr_argument = (xdrproc_t)xdr_void;
 276                 xdr_result = (xdrproc_t)xdr_utmpidlearr;
 277                 switch (rqstp->rq_vers) {
 278                 case RUSERSVERS_ORIG:
 279                         local = (rusersproc_t)rusersproc_names_1_svc;
 280                         break;
 281                 case RUSERSVERS_IDLE:
 282                         local = (rusersproc_t)rusersproc_names_2_svc;
 283                         break;
 284                 default:
 285                         svcerr_progvers(transp, RUSERSVERS_ORIG, RUSERSVERS_IDLE);
 286                         goto leave;
 287                         /*NOTREACHED*/
 288                 }
 289                 break;
 290
 291         case RUSERSPROC_ALLNAMES:
 292                 xdr_argument = (xdrproc_t)xdr_void;
 293                 xdr_result = (xdrproc_t)xdr_utmpidlearr;
 294                 switch (rqstp->rq_vers) {
 295                 case RUSERSVERS_ORIG:
 296                         local = (rusersproc_t)rusersproc_allnames_1_svc;
 297                         break;
 298                 case RUSERSVERS_IDLE:
 299                         local = (rusersproc_t)rusersproc_allnames_2_svc;
 300                         break;
 301                 default:
 302                         svcerr_progvers(transp, RUSERSVERS_ORIG, RUSERSVERS_IDLE);
 303                         goto leave;
 304                         /*NOTREACHED*/
 305                 }
 306                 break;
 307
 308         default:
 309                 svcerr_noproc(transp);
 310                 goto leave;
 311         }
 312         bzero(&argument, sizeof(argument));
 313         if (!svc_getargs(transp, (xdrproc_t)xdr_argument, &argument)) {
 314                 svcerr_decode(transp);
 315                 goto leave;
 316         }
 317         result = (*local)(&argument, rqstp);
 318         if (result != NULL &&
 319             !svc_sendreply(transp, (xdrproc_t)xdr_result, result)) {
 320                 svcerr_systemerr(transp);
 321         }
 322         if (!svc_freeargs(transp, (xdrproc_t)xdr_argument, &argument)) {
 323                 syslog(LOG_ERR, "unable to free arguments");
 324                 exit(1);
 325         }
 326 leave:
 327         if (from_inetd)
 328                 exit(0);
 329 }