/*

 * 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 2008 Sun Microsystems, Inc.  All rights reserved.

 * Use is subject to license terms.

 */

/*

 * Copyright 2011 Nexenta Systems, Inc.  All rights reserved.

 */

#include <fcntl.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <sys/types.h>

#include <unistd.h>

#include <stropts.h>

#include <sys/kbio.h>

#include <stdio.h>

#include <libintl.h>

#include <locale.h>

#include <malloc.h>

#include <string.h>

#include <sys/types.h>

#include <sys/uio.h>

#include <unistd.h>

#include <stdlib.h>

#include <sys/param.h>

#include <sys/stat.h>

#include <errno.h>

#include <dirent.h>

#include <ctype.h>

#include "orchestrator_private.h"

#include "orchestrator_lang_codes.h"

#define	COUNTRY_SEP	'_'

#define	CODESET_SEP	'.'

#define	UTF		"UTF-8"

#define	SIMPLIFIED_CHINESE	"Chinese-Simplified"

#define	TRADITIONAL_CHINESE	"Chinese-Traditional"

#define	INSTALL_NLS_PATH	"/usr/lib/install/data/lib/locale"

#define	NLS_PATH		"/usr/lib/locale"

/* Static variables used to store language/locale system information */

static	lang_info_t 	*install_ll_list = NULL;	/* lang, locale list */

static	lang_info_t	*supported_ll_list = NULL;

static	char		*install_lang_list[MAX_NUM_LANG];

static	char		*supported_lang_list[MAX_NUM_LANG];

static	char		**install_languages = NULL;	/* sorted list */

static	char		**supported_languages = NULL;

static	char		*app_locale = NULL;

static	int		lang_initialized = 0;

static	int		install_initialized = 0;

static	int		install_lang_total = 0;

static	int		supported_lang_total = 0;

struct	chinese_values {

	char 	*lang;

	char	*lang_name;

	char	*lang_code;

} chinese_values[] = {

	{"zh", SIMPLIFIED_CHINESE, "sc"},

	{"zh_CN", SIMPLIFIED_CHINESE, "sc"},

	{"zh_HK", TRADITIONAL_CHINESE, "tc"},

	{"zh_MO", TRADITIONAL_CHINESE, "tc"},

	{"zh_SG", TRADITIONAL_CHINESE, "tc"},

	{"zh_TW", TRADITIONAL_CHINESE, "tc"},

	{ NULL }

};

static int 	add_lang_to_list(char ***list, char *locale, int *k, int j);

static void 	add_locale_entry_to_lang(lang_info_t *lp, char *locale,

    char *region, boolean_t is_default);

static int	build_language_list(char *path, char **, int *);

static void	build_install_ll_list(char *nlspath, char **list,

    int lang_total, lang_info_t **return_list, int *ll_total);

static void	build_ll_list(char **list, int lang_total,

		    lang_info_t **, int *total);

static	char	*copy_up_to(char *start, char *t);

static int	create_lang_entry(char *lang, char *locale, char *region,

static void	end_of_comp(char **t, char **start);

static char 	**get_actual_languages(char **list, int *);

static lang_info_t *get_lang_entry(char *, lang_info_t *search_list);

static char 	*get_locale_component(char **t, char **start);

static char 	*get_locale_description(char *lang, char *region);

static int 	handle_chinese_language(char *region, char **lang);

static boolean_t is_locale_in_installer_lang(char *locale_name);

static boolean_t is_valid_locale(char *locale);

static int 	list_cmp(const void *p1, const void *p2);

static int 	lang_init(char *path, char **list, int *total, int *init_var);

static int 	save_system_default_locale(char *locale);

static void 	set_lang(char *locale);

static void 	set_lc(char *, char *, char *, char *, char *, char *);

static char 	*strip_comment(char *buf);

static char 	*translate_description(char *locale, char *desc);

static void 	translate_lang_names(lang_info_t **list);

static void 	sort_lang_list(char **unsorted_list, int total);

static char 	*substitute_chinese_language(char *locale, char **code);

static char 	*substitute_C_POSIX_language(char **code);

static char 	*substitute_language(char *locale, char **code);

static void 	update_init(FILE *fp, char *locale);

static void 	update_env(char *locale);

#define	STR_LANG	"LANG="

#define	LEN_LANG	(sizeof (STR_LANG) - 1)

#define	STR_LC_COLLATE	"LC_COLLATE="

#define	LEN_LC_COLLATE	(sizeof (STR_LC_COLLATE) - 1)

#define	STR_LC_CTYPE	"LC_CTYPE="

#define	LEN_LC_CTYPE	(sizeof (STR_LC_CTYPE) - 1)

#define	STR_LC_MESSAGES	"LC_MESSAGES="

#define	LEN_LC_MESSAGES	(sizeof (STR_LC_MESSAGES) - 1)

#define	STR_LC_MONETARY	"LC_MONETARY="

#define	LEN_LC_MONETARY	(sizeof (STR_LC_MONETARY) - 1)

#define	STR_LC_NUMERIC	"LC_NUMERIC="

#define	LEN_LC_NUMERIC	(sizeof (STR_LC_NUMERIC) - 1)

#define	STR_LC_TIME	"LC_TIME="

#define	LEN_LC_TIME	(sizeof (STR_LC_TIME) - 1)

/*

 * om_get_install_lang_info

 * This function returns the locales available for use in setting

 * installer locale.

 * Input:	None

 * Output:	int *total, returns the total number of locales found

 * Return:	Pointer to lang_info_t which is a linked list of

 *		locales available for selection.

 *		NULL if no locales found.

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILIRE if no locales found.

 */

lang_info_t *

om_get_install_lang_info(int *total)

{

	int		ll_total;

	int		ret;

	/*

	 * Path to look for locale data.

	 */

	*total = 0;

	/*

	 * Build the lang list for the install application support.

	 */

	if (!install_initialized) {

		ret = lang_init(INSTALL_NLS_PATH, (char **)&install_lang_list,

		    &install_lang_total, &install_initialized);

		if (ret)

			return (NULL);

	}

	/*

	 * For the install languages we build the locale list associated

	 * with them by reading the lcctab file in the /usr/lib/locale/

	 * directory.

	 */

	build_install_ll_list(NLS_PATH, (char **)&install_lang_list,

	    install_lang_total, &install_ll_list, &ll_total);

	*total = ll_total;

	return (install_ll_list);

}

/*

 * om_get_install_lang_names

 * This function returns the lang names available for use in setting

 * installer locale system

 * Input:	None

 * Output:	int *total, returns the total number of locales found

 * Return:	char ** list of lang names available for selection.

 *		NULL if no locales found.

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILIRE if no locales found.

 */

char **

om_get_install_lang_names(int *total)

{

	int		ret;

	*total = 0;

	/*

	 * Reads directory that contains installation application

	 * languages. Fills install_lang_list with this data,

	 * if this has not been initialized yet.

	 */

	if (!install_initialized) {

		ret = lang_init(INSTALL_NLS_PATH, (char **)&install_lang_list,

		    &install_lang_total, &install_initialized);

		if (ret)

			return (NULL);

	}

	install_languages = get_actual_languages((char **)install_lang_list,

	    total);

	sort_lang_list(install_languages, *total);

	return (install_languages);

}

/*

 * om_get_lang_info

 * This function returns the available locales for installation on to

 * system

 * Input:	None

 * Output:	int *total, returns the total number of locales found

 * Return:	Pointer to lang_info_t which is a linked list of

 *		locales available for selection to install.

 *		NULL if no locales found.

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILIRE if no locales found.

 */

lang_info_t *

om_get_lang_info(int *total)

{

	int	ret;

	int	locale_total;

	*total = 0;

	if (!lang_initialized) {

		ret = lang_init(NLS_PATH, (char **)&supported_lang_list,

		    &supported_lang_total, &lang_initialized);

		if (ret)

			return (NULL);

	}

	/*

	 * This function looks in usr/lib/locale on the installation

	 * media to determine languages that we provide support for.

	 */

	sort_lang_list((char **)&supported_lang_list, supported_lang_total);

	build_ll_list((char **)&supported_lang_list, supported_lang_total,

	    &supported_ll_list, &locale_total);

	*total = locale_total;

	return (supported_ll_list);

}

/*

 * om_get_lang_names

 * This function returns the language strings that can are supported for

 * installation on to system.

 * Input:	None

 * Output:	int *total, returns the total number of lang names found

 * Return:	char ** list of lang names

 *		NULL if no lang names found.

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILURE if no locales found.

 */

char **

om_get_lang_names(int *total)

{

	int	ret = 0;

	*total = 0;

	/*

	 * We only gather 'supported lang names'. Supported means

	 * full locale, UTF-8 only. For display to the user for

	 * default locale selection.

	 */

	if (!lang_initialized) {

		ret = lang_init(NLS_PATH, (char **)&supported_lang_list,

		    &supported_lang_total, &lang_initialized);

		if (ret) {

			return (NULL);

		}

	}

	supported_languages = get_actual_languages(supported_lang_list, total);

	sort_lang_list(supported_languages, *total);

	return (supported_languages);

}

/*

 * om_get_locale_info

 * This function returns a linked list of locale_info_t's that correspond

 * to the language specified.

 * Input:	char *lang - language for which to return locale data.

 * Output:	int *total, returns the total number of lang names found

 * Return:	locale_info_t * list of locale_names.

 *		NULL if no locale names found.

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILURE if no locales found.

 */

locale_info_t *

om_get_locale_info(char *lang, int *total)

{

	lang_info_t *langp;

	locale_info_t *localep = NULL;

	lang_info_t	*tmp = supported_ll_list;

	if (tmp == NULL)

		tmp = install_ll_list;

	if (tmp == NULL)

		return (NULL);

	/*

	 * The lang is passed in as the lang code. Not the translated

	 * name.

	 */

	for (langp = tmp; langp != NULL; langp = langp->next) {

		if (strcmp(langp->lang, lang) == 0) {

			localep = langp->locale_info;

			break;

		}

	}

	*total = langp->n_locales;

	return (localep);

}

/*

 * om_get_locale_names

 * This function returns a list of locale names that correspond

 * to the language specified.

 * Input:	char *lang - language for which to return locale data.

 * Output:	int *total, returns the total number of lang names found

 * Return:	char ** list of locale_names.

 *		NULL if no locale names found.

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILURE if no locales found.

 */

char **

om_get_locale_names(char *lang, int *total)

{

	lang_info_t 	*langp;

	locale_info_t 	*localep = NULL;

	char		**locale_names;

	int		i = 0;

	/*

	 * locale_names is a an array of char *. Allocate each array

	 * member before strduping the locale_info in to this array.

	 */

	/*

	 * The lang is passed in as the lang code. Not the translated

	 * name.

	 */

	for (langp = supported_ll_list; langp != NULL; langp = langp->next) {

		if (strcmp(langp->lang, lang) == 0) {

			localep = langp->locale_info;

			break;

		}

	}

	/*

	 * allocate number of char * pointers to correspond to number

	 * of locale names.

	 */

	locale_names = (char **)malloc(langp->n_locales * sizeof (char *));

	if (locale_names == NULL) {

		om_set_error(OM_NO_SPACE);

		return (NULL);

	}

	for (; localep; localep = localep->next) {

		locale_names[i] = strdup(localep->locale_name);

		if (locale_names[i] == NULL) {

			om_free_lang_names(locale_names);

			om_set_error(OM_NO_SPACE);

			return (NULL);

		}

		i++;

	}

	*total = i;

	return (locale_names);

}

/*

 * om_set_install_lang_by_value

 * This function sets the installer application language, using the

 * lang_info_t * data passed in.

 * Input:	lang_info_t	* lang/locale info to set

 * Output:	none

 * Return:	Success or Failure

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILURE if no locales found.

 */

int

om_set_install_lang_by_value(lang_info_t *localep)

{

	/*

	 * For install applications, there will only be 1 locale

	 * associated with that language.

	 */

	locale_info_t *locp = localep->locale_info;

	if (locp == NULL) {

		om_set_error(OM_INVALID_LOCALE);

		return (OM_FAILURE);

	}

	om_save_locale(locp->locale_name, B_TRUE);

	return (OM_SUCCESS);

}

/*

 * om_set_install_lang_by_name

 * This function sets the installer application language, using the

 * lang name passed in.

 * Input:	char *lang name to set.

 * Output:	none

 * Return:	int, Success or Failure

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILURE if no locales found.

 */

int

om_set_install_lang_by_name(char *lang)

{

	locale_info_t	*locp;

	int		total;

	/*

	 * Find the locale entry based on the name passed in. If no

	 * locales for this language use the default locale of C/POSIX.

	 */

	locp = om_get_locale_info(lang, &total);

	if (locp == NULL) {

		om_set_error(OM_INVALID_LOCALE);

		return (OM_FAILURE);

	}

	/*

	 * Now, set the environment for the installation application.

	 */

	om_save_locale(locp->locale_name, B_TRUE);

	om_free_locale_info(locp);

	return (OM_SUCCESS);

}

/*

 * om_set_default_locale_by_name

 * This function sets the system default locale by name

 * Input:	char *lang name to set.

 * Output:	none

 * Return:	int, Success or Failure

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILURE if no locales found.

 */

int

om_set_default_locale_by_name(char *localep)

{

	int ret = 0;

	/*

	 * C/Posix is the default, there is no need to specify it

	 * in the /etc/default/init file.

	 */

	if (strcasecmp(localep, "C/Posix") == 0 ||

	    strcasecmp(localep, "Posix") == 0) {

		return (ret);

	}

	ret = save_system_default_locale(localep);

	if (!ret)

		om_save_locale(localep, B_FALSE);

	return (ret);

}

/*

 * om_get_default_locale

 * This function returns the default locale as set by a call to

 * set_default_locale.

 * Input:	locale_info_t *loclistp - list to search

 * Output:	none

 * Return:	locale_info_t * pointer to structure for default locale.

 * 		or NULL if none found.

 * Error Handling:

 *		OM_SUCCESS if locales found.

 *		OM_FAILURE if no locales found.

 */

locale_info_t *

om_get_def_locale(locale_info_t *loclistp)

{

	locale_info_t *lp;

	for (lp = loclistp; lp != NULL; lp = lp->next)

		if (lp->def_locale == B_TRUE)

			return (lp);

	return (NULL);

}

/*

 * om_free_lang_names

 * This function frees the memory associated with the char **list of lang

 * names passed in.

 * Input:	char **lang name list.

 * Output:	none

 * Return:	none

 * Error Handling:

 *		none

 */

void

om_free_lang_names(char **listp)

{

	int	i = 0;

	for (i = 0; listp[i] != NULL; i++) {

		free(listp[i]);

	}