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


/*
 * WARNING WARNING WARNING WARNING WARNING WARNING WARNING WARNING WARNING
 *
 *	Openvision retains the copyright to derivative works of
 *	this source code.  Do *NOT* create a derivative of this
 *	source code before consulting with your legal department.
 *	Do *NOT* integrate *ANY* of this source code into another
 *	product before consulting with your legal department.
 *
 *	For further information, read the top-level Openvision
 *	copyright which is contained in the top-level MIT Kerberos
 *	copyright.
 *
 * WARNING WARNING WARNING WARNING WARNING WARNING WARNING WARNING WARNING
 *
 */


/*
 * Copyright 1993 OpenVision Technologies, Inc., All Rights Reserved.
 */


#include <stdio.h>
#ifdef HAVE_MEMORY_H
#include <memory.h>
#endif
#include <time.h>
#include <locale.h>

#include <kadm5/admin.h>
#include "admin_internal.h"

#include <krb5.h>
#include <strings.h>

#define string_text error_message

const char *chpw_error_message(kadm5_ret_t code);

/*
 * Function: kadm5_chpass_principal_util
 *
 * Purpose: Wrapper around chpass_principal. We can read new pw, change pw and return useful messages
 *
 * Arguments:
 *
 *      princ          (input) a krb5b_principal structure for the 
 *                     principal whose password we should change.
 *
 *      new_password   (input) NULL or a null terminated string with the 
 *                     the principal's desired new password.  If new_password
 *                     is NULL then this routine will read a new password.
 *    
 *	pw_ret		(output) if non-NULL, points to a static buffer
 *			containing the new password (if password is prompted
 *			internally), or to the new_password argument (if
 *			that is non-NULL).  If the former, then the buffer
 *			is only valid until the next call to the function,
 *			and the caller should be sure to zero it when
 *			it is no longer needed.
 *
 *      msg_ret         (output) a useful message is copied here.
 *
 *      <return value>  exit status of 0 for success, else the com err code 
 *                      for the last significant routine called.
 *      
 * Requires:
 *      
 *      A msg_ret should point to a buffer large enough for the messasge.
 *
 * Effects:
 *      
 * Modifies:
 *
 *
 */

kadm5_ret_t _kadm5_chpass_principal_util(void *server_handle,
					 void *lhandle,
					 krb5_principal princ,
					 char *new_pw, 
					 char **ret_pw,
					 char *msg_ret,
					 unsigned int msg_len)
{
  int code, code2;
  unsigned int pwsize;
  static char buffer[255];
  char *new_password;
  kadm5_principal_ent_rec princ_ent;
  kadm5_policy_ent_rec policy_ent;
  krb5_chgpwd_prot passwd_protocol;

  _KADM5_CHECK_HANDLE(server_handle);

  if (ret_pw)
    *ret_pw = NULL;

  if (new_pw != NULL) {
    new_password = new_pw;
  } else { /* read the password */
    krb5_context context;

    if ((code = (int) kadm5_init_krb5_context(&context)) == 0) {
      pwsize = sizeof(buffer);
      code = krb5_read_password(context, KADM5_PW_FIRST_PROMPT,
				KADM5_PW_SECOND_PROMPT,
				buffer, &pwsize);
      krb5_free_context(context);
    }

    if (code == 0) 
      new_password = buffer;
    else {
#ifdef ZEROPASSWD    
      memset(buffer, 0, sizeof(buffer));
#endif      
      if (code == KRB5_LIBOS_BADPWDMATCH) {
	(void) strncpy(msg_ret, string_text(CHPASS_UTIL_NEW_PASSWORD_MISMATCH),
		msg_len - 1);
	msg_ret[msg_len - 1] = '\0';
	return(code);
      } else {
        (void) strncpy(msg_ret, error_message(code), msg_len - 1);
        (void) strncat(msg_ret, " ", msg_len - 1);
        (void) strncat(msg_ret, string_text(CHPASS_UTIL_WHILE_READING_PASSWORD),
		msg_len - 1);
        (void) strncat(msg_ret, string_text(CHPASS_UTIL_PASSWORD_NOT_CHANGED),
		msg_len - 1);
	msg_ret[msg_len - 1] = '\0';
	return(code);
      }
    }
    if (pwsize == 0) {
#ifdef ZEROPASSWD    
      memset(buffer, 0, sizeof(buffer));
#endif      
      strncpy(msg_ret, string_text(CHPASS_UTIL_NO_PASSWORD_READ), msg_len - 1);
      msg_ret[msg_len - 1] = '\0';
      return(KRB5_LIBOS_CANTREADPWD); /* could do better */
    }
  }

  if (ret_pw)
    *ret_pw = new_password;

	passwd_protocol = _kadm5_get_kpasswd_protocol(server_handle);
	if (passwd_protocol == KRB5_CHGPWD_CHANGEPW_V2) {
		kadm5_ret_t srvr_rsp_code;
		krb5_data   srvr_msg;

		srvr_msg.length = 0;
		srvr_msg.data = NULL;

		code = kadm5_chpass_principal_v2(server_handle, princ,
						new_password,
						&srvr_rsp_code,
						&srvr_msg);
		if (srvr_rsp_code) {
			sprintf(msg_ret, "%s%s%.*s\n",
				chpw_error_message(srvr_rsp_code),
				srvr_msg.length? ": " : "",
				srvr_msg.length,
				srvr_msg.data ? srvr_msg.data : "");

			return (srvr_rsp_code);
		}
		return (code);

	} else if (passwd_protocol == KRB5_CHGPWD_RPCSEC) {
		code = kadm5_chpass_principal(server_handle, princ,
					    new_password);

#ifdef ZEROPASSWD
  if (!ret_pw)
    memset(buffer, 0, sizeof(buffer)); /* in case we read a new password */
#endif    

  if (code == KADM5_OK) {
    strncpy(msg_ret, string_text(CHPASS_UTIL_PASSWORD_CHANGED), msg_len - 1);
    msg_ret[msg_len - 1] = '\0';
    return(0);
  }

  if ((code != KADM5_PASS_Q_TOOSHORT) && 
      (code != KADM5_PASS_REUSE) &&(code != KADM5_PASS_Q_CLASS) && 
      (code != KADM5_PASS_Q_DICT) && (code != KADM5_PASS_TOOSOON)) {
    /* Can't get more info for other errors */
    sprintf(buffer, "%s %s", error_message(code), 
	    string_text(CHPASS_UTIL_WHILE_TRYING_TO_CHANGE));
    sprintf(msg_ret, "%s\n%s\n", string_text(CHPASS_UTIL_PASSWORD_NOT_CHANGED), 
	    buffer);
    return(code);
  }

  /* Ok, we have a password quality error. Return a good message */

  if (code == KADM5_PASS_REUSE) {
    strncpy(msg_ret, string_text(CHPASS_UTIL_PASSWORD_REUSE), msg_len - 1);
    msg_ret[msg_len - 1] = '\0';
    return(code);
  }

  if (code == KADM5_PASS_Q_DICT) {
    strncpy(msg_ret, string_text(CHPASS_UTIL_PASSWORD_IN_DICTIONARY),
	    msg_len - 1);
    msg_ret[msg_len - 1] = '\0';
    return(code);
  }
  
  /* Look up policy for the remaining messages */

  code2 = kadm5_get_principal (lhandle, princ, &princ_ent,
			       KADM5_PRINCIPAL_NORMAL_MASK);
  if (code2 != 0) {
    strncpy(msg_ret, error_message(code2), msg_len - 1);
    strncat(msg_ret, " ", msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, string_text(CHPASS_UTIL_GET_PRINC_INFO), msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, "\n", msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, error_message(code), msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, " ", msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, string_text(CHPASS_UTIL_WHILE_TRYING_TO_CHANGE),
	    msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, "\n\n", msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, string_text(CHPASS_UTIL_PASSWORD_NOT_CHANGED),
	    msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, "\n", msg_len - 1 - strlen(msg_ret));
    msg_ret[msg_len - 1] = '\0';
    return(code);
  }
  
  if ((princ_ent.aux_attributes & KADM5_POLICY) == 0) {
    strncpy(msg_ret, error_message(code), msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, " ", msg_len - 1 - strlen(msg_ret));
    strncpy(msg_ret, string_text(CHPASS_UTIL_NO_POLICY_YET_Q_ERROR),
	    msg_len - 1 - strlen(msg_ret));
    strncat(msg_ret, "\n\n", msg_len - 1 - strlen(msg_ret));
    strncpy(msg_ret, string_text(CHPASS_UTIL_PASSWORD_NOT_CHANGED),
	    msg_len - 1 - strlen(msg_ret));
    msg_ret[msg_len - 1] = '\0';

    (void) kadm5_free_principal_ent(lhandle, &princ_ent);
    return(code);
  }

  code2 = kadm5_get_policy(lhandle, princ_ent.policy,
			   &policy_ent);
  if (code2 != 0) {
    sprintf(msg_ret, "%s %s\n%s %s\n\n%s\n ", error_message(code2), 
	    string_text(CHPASS_UTIL_GET_POLICY_INFO),
	    error_message(code),
	    string_text(CHPASS_UTIL_WHILE_TRYING_TO_CHANGE),
	    string_text(CHPASS_UTIL_PASSWORD_NOT_CHANGED));
    (void) kadm5_free_principal_ent(lhandle, &princ_ent);
    return(code);
  }
  
  if (code == KADM5_PASS_Q_TOOSHORT) {
    sprintf(msg_ret, string_text(CHPASS_UTIL_PASSWORD_TOO_SHORT), 
	    policy_ent.pw_min_length);
    (void) kadm5_free_principal_ent(lhandle, &princ_ent);
    (void) kadm5_free_policy_ent(lhandle, &policy_ent);
    return(code);
  }


  if (code == KADM5_PASS_Q_CLASS) {
    sprintf(msg_ret, string_text(CHPASS_UTIL_TOO_FEW_CLASSES), 
	    policy_ent.pw_min_classes);
    (void) kadm5_free_principal_ent(lhandle, &princ_ent);
    (void) kadm5_free_policy_ent(lhandle, &policy_ent);
    return(code);
  }

  if (code == KADM5_PASS_TOOSOON) {
    time_t until;
    char *time_string, *ptr;

    until = princ_ent.last_pwd_change + policy_ent.pw_min_life;

    time_string = ctime(&until);
    if (*(ptr = &time_string[strlen(time_string)-1]) == '\n')
      *ptr = '\0';

    sprintf(msg_ret, string_text(CHPASS_UTIL_PASSWORD_TOO_SOON), 
	    time_string);
    (void) kadm5_free_principal_ent(lhandle, &princ_ent);
    (void) kadm5_free_policy_ent(lhandle, &policy_ent);
    return(code);
  } else {

  /* We should never get here, but just in case ... */
  sprintf(buffer, "%s %s", error_message(code), 
	  string_text(CHPASS_UTIL_WHILE_TRYING_TO_CHANGE));
  sprintf(msg_ret, "%s\n%s\n", string_text(CHPASS_UTIL_PASSWORD_NOT_CHANGED), 
	  buffer);
  (void) kadm5_free_principal_ent(lhandle, &princ_ent);
  (void) kadm5_free_policy_ent(lhandle, &policy_ent);
  return(code);
		}
	} else {
		sprintf(msg_ret, "%s\n%s\n",
			string_text(CHPASS_UTIL_PASSWORD_NOT_CHANGED),
			"Password protocol in krb5.conf is not supported\n");
		return (-1);
	}
}

/*
 * krb5_chpw_result_code_string
 *
 * convert the return code received from the password server
 * to a human-readable string.
 */
const char *
chpw_error_message(kadm5_ret_t result_code)
{
	switch (result_code) {
	case KRB5_KPASSWD_MALFORMED:
		return (dgettext(TEXT_DOMAIN, "Malformed request error"));
	case KRB5_KPASSWD_HARDERROR:
		return (dgettext(TEXT_DOMAIN, "Server error"));
	case KRB5_KPASSWD_AUTHERROR:
		return (dgettext(TEXT_DOMAIN, "Authentication error"));
	case KRB5_KPASSWD_SOFTERROR:
		return (dgettext(TEXT_DOMAIN, "Password change rejected"));
	case KRB5_KPASSWD_ACCESSDENIED:
		return (dgettext(TEXT_DOMAIN,
				"Not authorized to change password"));
	case KRB5_KPASSWD_BAD_VERSION:
		return (dgettext(TEXT_DOMAIN, "Protocol version unsupported"));
	case KRB5_KPASSWD_INITIAL_FLAG_NEEDED:
		return (dgettext(TEXT_DOMAIN,
				"initial flag required in changepw request"));
	case KRB5_KPASSWD_POLICY_REJECT:
		return (dgettext(TEXT_DOMAIN, "new password fails policy"));
	case KRB5_KPASSWD_BAD_PRINCIPAL:
		return (dgettext(TEXT_DOMAIN,
		    "target principal does not exist for "
		    "changepw request"));
	case KRB5_KPASSWD_ETYPE_NOSUPP:
		return (dgettext(TEXT_DOMAIN,
		    "changepw request key sequence has an "
		    "unsupported Etype"));
	default:
		return (dgettext(TEXT_DOMAIN, "Password change failed"));
	}
}