usr/src/cmd/nscd/nscd_smfmonitor.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21 /*
  22  * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
  23  * Use is subject to license terms.
  24  *
  25  * Copyright 2018 Joyent, Inc.
  26  */
  27
  28 #include <stdlib.h>
  29 #include <libscf.h>
  30 #include <string.h>
  31 #include "nscd_switch.h"
  32 #include "nscd_log.h"
  33 #include "nscd_door.h"
  34
  35 extern int      _whoami;
  36
  37 /*
  38  * Service states monitored by nscd. Protected by
  39  * readers/writer lock nscd_smf_service_state_lock
  40  */
  41 nscd_smf_state_t *nscd_smf_service_state;
  42 static rwlock_t nscd_smf_service_state_lock = DEFAULTRWLOCK;
  43 /*
  44  * init service state table
  45  */
  46 nscd_rc_t
  47 _nscd_alloc_service_state_table()
  48 {
  49         int i;
  50
  51         nscd_smf_service_state = calloc(NSCD_NUM_SMF_FMRI,
  52             sizeof (nscd_smf_state_t));
  53
  54         if (nscd_smf_service_state == NULL)
  55                 return (NSCD_NO_MEMORY);
  56
  57         for (i = 1; i < NSCD_NUM_SMF_FMRI; i++)
  58                 NSCD_SMF_SVC_STATE(i) = NSCD_SVC_STATE_UNINITED;
  59
  60         return (NSCD_SUCCESS);
  61 }
  62
  63 static int
  64 query_smf_state(int srci)
  65 {
  66
  67         int     ret = NSCD_SVC_STATE_UNINITED;
  68         char    *state = NULL;
  69         char    *me = "query_smf_state";
  70
  71         state = smf_get_state(NSCD_SMF_SVC_FMRI(srci));
  72         if (state == NULL)
  73                 return (ret);
  74
  75         _NSCD_LOG(NSCD_LOG_SMF_MONITOR, NSCD_LOG_LEVEL_DEBUG)
  76                 (me, "%s -- %s\n", state, NSCD_SMF_SVC_FMRI(srci));
  77
  78         (void) rw_wrlock(&nscd_smf_service_state_lock);
  79
  80         if (nscd_smf_service_state[srci].src_name == NULL)
  81                 nscd_smf_service_state[srci].src_name =
  82                     NSCD_NSW_SRC_NAME(srci);
  83
  84         if (strcmp(state, SCF_STATE_STRING_UNINIT) == 0)
  85                 NSCD_SMF_SVC_STATE(srci) = SCF_STATE_UNINIT;
  86         else if (strcmp(state, SCF_STATE_STRING_MAINT) == 0)
  87                 NSCD_SMF_SVC_STATE(srci) = SCF_STATE_MAINT;
  88         else if (strcmp(state, SCF_STATE_STRING_OFFLINE) == 0)
  89                 NSCD_SMF_SVC_STATE(srci) = SCF_STATE_OFFLINE;
  90         else if (strcmp(state, SCF_STATE_STRING_DISABLED) == 0)
  91                 NSCD_SMF_SVC_STATE(srci) = SCF_STATE_DISABLED;
  92         else if (strcmp(state, SCF_STATE_STRING_ONLINE) == 0)
  93                 NSCD_SMF_SVC_STATE(srci) = SCF_STATE_ONLINE;
  94         else if (strcmp(state, SCF_STATE_STRING_DEGRADED) == 0)
  95                 NSCD_SMF_SVC_STATE(srci) = SCF_STATE_DEGRADED;
  96
  97         ret = NSCD_SMF_SVC_STATE(srci);
  98         (void) rw_unlock(&nscd_smf_service_state_lock);
  99
 100         free(state);
 101         return (ret);
 102 }
 103
 104 /* ARGSUSED */
 105 static void *
 106 set_smf_state(void *arg)
 107 {
 108
 109         int     i;
 110         int     st;
 111
 112         (void) thr_setname(thr_self(), "set_smf_state");
 113
 114         /*
 115          * the forker nscd needs not monitor the state
 116          * of the client services
 117          */
 118         if (_whoami == NSCD_FORKER)
 119                 thr_exit(0);
 120
 121         /*CONSTCOND*/
 122         while (1) {
 123
 124                 /* skip the first service which is nscd */
 125                 for (i = 1; i < NSCD_NUM_SMF_FMRI; i++) {
 126                         st = query_smf_state(i);
 127                         if (st == NSCD_SVC_STATE_UNINITED)
 128                                 break;
 129                 }
 130
 131                 (void) sleep(NSCD_SW_CFG_G.check_smf_state_interval_g);
 132         }
 133         /* NOTREACHED */
 134         /*LINTED E_FUNC_HAS_NO_RETURN_STMT*/
 135 }
 136
 137 nscd_rc_t
 138 _nscd_init_smf_monitor() {
 139
 140         int     errnum;
 141         char    *me = "_nscd_init_smf_monitor";
 142
 143         _NSCD_LOG(NSCD_LOG_SMF_MONITOR, NSCD_LOG_LEVEL_DEBUG)
 144         (me, "initializing the smf monitor\n");
 145
 146         /*
 147          * start a thread to check the state of the client services
 148          */
 149         if (thr_create(NULL, NULL, set_smf_state,
 150                 NULL, THR_DETACHED, NULL) != 0) {
 151                 errnum = errno;
 152                 _NSCD_LOG(NSCD_LOG_SMF_MONITOR, NSCD_LOG_LEVEL_ERROR)
 153                 (me, "thr_create: %s\n", strerror(errnum));
 154                 return (NSCD_THREAD_CREATE_ERROR);
 155         }
 156
 157         return (NSCD_SUCCESS);
 158 }
 159
 160 int
 161 _nscd_get_smf_state(int srci, int dbi, int recheck)
 162 {
 163         int     s;
 164         char    *n;
 165
 166         n = NSCD_NSW_SRC_NAME(srci);
 167
 168         /* the files, compat, and dns backends are always available */
 169         if ((*n == 'f' || *n == 'c' || *n == 'd' || *n == 'a') &&
 170             (strcmp(NSCD_NSW_SRC_NAME(srci), "files") == 0 ||
 171             strcmp(NSCD_NSW_SRC_NAME(srci), "compat") == 0 ||
 172             strcmp(NSCD_NSW_SRC_NAME(srci), "ad") == 0 ||
 173             strcmp(NSCD_NSW_SRC_NAME(srci), "dns") == 0)) {
 174                 return (SCF_STATE_ONLINE);
 175         }
 176
 177         /*
 178          * for the printer database and user backend, treat the
 179          * backend as a unsupported one, as nscd can not access
 180          * the home directory of the user
 181          */
 182         if (*n == 'u' && strcmp(NSCD_NSW_SRC_NAME(srci), "user") == 0) {
 183                 if (strcmp(NSCD_NSW_DB_NAME(dbi), NSS_DBNAM_PRINTERS) == 0)
 184                         return (NSCD_SVC_STATE_UNSUPPORTED_SRC);
 185                 else
 186                         return (SCF_STATE_ONLINE);
 187         }
 188
 189         /*
 190          * Foreign backend is not supported by nscd unless
 191          * the backend supports the nss2 interface (global
 192          * symbol _nss_<backname name>_version is present),
 193          * tell the switch engine to return NSS_TRYLOCAL
 194          * if needed via rc NSCD_SVC_STATE_FOREIGN_SRC.
 195          */
 196         if (srci >= _nscd_cfg_num_nsw_src)
 197                 return (NSCD_SVC_STATE_FOREIGN_SRC);
 198
 199         if (recheck == 1)
 200                 return (query_smf_state(srci));
 201
 202         (void) rw_rdlock(&nscd_smf_service_state_lock);
 203         s = NSCD_SMF_SVC_STATE(srci);
 204         (void) rw_unlock(&nscd_smf_service_state_lock);
 205
 206         /*
 207          * if the state has been queried at least once but is
 208          * still not online, query one more time
 209          */
 210         if (s != NSCD_SVC_STATE_UNINITED && s < SCF_STATE_ONLINE)
 211                 s = query_smf_state(srci);
 212
 213         return (s);
 214 }