/*
 *
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 *
 */

/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
*/


#include <stdlib.h>
#include <unistd.h>
#include <strings.h>
#include <libscf.h>

#define MAX_TRY 15
static boolean_t fs_temporarily_enabled = B_FALSE;
char svm_core_svcs[] = "system/filesystem/local:default";

/*
 * Name:        enable_local_fs
 * Description: If the SMF service system/filesystem/local:default is not
 *              enabled, then this function enables the service, so that,
 *              all the local filesystems are mounted.
 * Arguments:   None
 * Returns:     B_TRUE on success; B_FALSE on error.
 */
boolean_t
enable_local_fs(void)
{
        char *cur_smf_state;
        int i;
        boolean_t fs_enabled_here = B_FALSE;

        if (fs_temporarily_enabled)  {
                return (B_TRUE);
        }

        if ((cur_smf_state = smf_get_state(svm_core_svcs)) != NULL) {
                if (strcmp(cur_smf_state, SCF_STATE_STRING_DISABLED) == 0) {
                        if (smf_enable_instance(svm_core_svcs, SMF_TEMPORARY)
                                != 0) {
                                free(cur_smf_state);
                                return (B_FALSE);
                        }

                        fs_enabled_here = B_TRUE;

                } else if (strcmp(cur_smf_state, SCF_STATE_STRING_ONLINE)
                                == 0) {
                        free(cur_smf_state);
                        return (B_TRUE);
                } else if (strcmp(cur_smf_state, SCF_STATE_STRING_OFFLINE)
                                != 0) {
                        free(cur_smf_state);
                        return (B_FALSE);
                }

                free(cur_smf_state);

        } else {
                return (B_FALSE);
        }

        for (i = 0; i < MAX_TRY; i++) {
                if ((cur_smf_state = smf_get_state(svm_core_svcs)) != NULL) {
                        if (strcmp(cur_smf_state, SCF_STATE_STRING_ONLINE)
                                == 0) {
                                free(cur_smf_state);
                                if (fs_enabled_here) {
                                        fs_temporarily_enabled = B_TRUE;
                                }
                                return (B_TRUE);
                        } else if ((strcmp(cur_smf_state,
                                SCF_STATE_STRING_OFFLINE) == 0) ||
                (strcmp(cur_smf_state, SCF_STATE_STRING_DISABLED) == 0)) {
                                (void) sleep(1);
                                free(cur_smf_state);
                        } else {
                                free(cur_smf_state);
                                return (B_FALSE);
                        }
                } else {
                        return (B_FALSE);
                }
        }

        return (B_FALSE);
}

/*
 * Name:        restore_local_fs
 * Description: If the SMF service system/filesystem/local:default was
 *              enabled using enable_local_fs(), then this function disables
 *              the service.
 * Arguments:   None
 * Returns:     B_TRUE on success; B_FALSE on error.
 */
boolean_t
restore_local_fs(void)
{
        int i;
        char *cur_smf_state;

        if (!fs_temporarily_enabled) {
                return (B_TRUE);
        }

        if (smf_disable_instance(svm_core_svcs, SMF_TEMPORARY) != 0) {
                return (B_FALSE);
        }

        for (i = 0; i < MAX_TRY; i++) {
                if ((cur_smf_state = smf_get_state(svm_core_svcs)) != NULL) {
                        if (strcmp(cur_smf_state, SCF_STATE_STRING_DISABLED)
                                == 0) {
                                fs_temporarily_enabled = B_FALSE;
                                free(cur_smf_state);
                                break;
                        }
                        (void) sleep(1);

                        free(cur_smf_state);
                } else {
                        return (B_FALSE);
                }
        }

        return (!fs_temporarily_enabled);
}