//*****************************************************************//
//*****************************************************************//
//*****************************************************************//
//**                                                             **//
//**         (C)Copyright 2014, American Megatrends, Inc.        **//
//**                                                             **//
//**                     All Rights Reserved.                    **//
//**                                                             **//
//**   5555 Oakbrook Pkwy, Building 200,Norcross, Georgia 30093  **//
//**                                                             **//
//**                     Phone (770)-246-8600                    **//
//**                                                             **//
//*****************************************************************//
//*****************************************************************//
//*****************************************************************//
// $Archive: /Alaska/SOURCE/Modules/AMITSE2_0/AMITSE/TseLite/PopupEdit.c $
//
// $Author: Premkumara $
//
// $Revision: 28 $
//
// $Date: 5/03/14 4:02p $
//
//*****************************************************************//
//*****************************************************************//
// Revision History
// ----------------
// $Log: /Alaska/SOURCE/Modules/AMITSE2_0/AMITSE/TseLite/PopupEdit.c $
// 
// 28    5/03/14 4:02p Premkumara
// [TAG]	EIP135665
// [Category]	Bug Fix
// [Severity:]	Important
// [Symptom:]	TSE get hangs, when long string is given as title for
// password
// [Root Cause]	TSE is not handling properl when PasswordTitle length
// exceeds screen width 
// [Solution]	Handled password title length based on screen width and
// given support to display title in multiple lines.
// [Files]	postmgmtext.c, Popup.c, PopupEdit.c, PopupPassword.c 
// 
// 27    5/01/14 3:53p Premkumara
// [TAG]  		EIP158293 
// [Category]  	Bug Fix
// [Severity]  	Normal
// [Symptom]  	Closing Password popup window displaying invalid password
// msg box, when SINGLE_CLICK_ACTIVATION token is enabled.
// [RootCause]  	Once clicked outside password popup window,
// ControlActionAbort action is not setting so it returning EFFI_SUCCESS
// status. If status is not success invalid password will display.
// [Solution]  	When mouse is clicked outside password/string popup
// window, ControlActionAbort action will be set so finally it will return
// EFI_SUCCESS and "invalid password" msg box will not display.
// [Files]  		PopupEdit.c
// 
// 26    2/11/14 8:40p Arunsb
// Changes reverted for 2.16.1243 label
// 
// 25    12/03/13 8:57a Premkumara
// [TAG]	EIP135665
// [Category]	Bug Fix
// [Severity:]	Important
// [Symptom:]	TSE get hangs, when long string is given as title for
// password
// [Root Cause]	TSE is not handling properl when PasswordTitle length
// exceeds screen width 
// [Solution]	Handled password title length based on screen width and
// given support to display title in multiple lines.
// [Files]	postmgmtext.c, Popup.c, PopupEdit.c, PopupPassword.c 
// 
// 24    5/15/13 4:06a Premkumara
// [TAG]  		EIP123311
// [Issue Faced]	Popup string control accepts ENTER character while
// pressing CTRL+ENTER. So causing page corruption
// [Root Cause]	When CTRL+ENTER is pressed valid shift key is set and
// unicode char set with carriage return. But action get as unknown so for
// unknow case we are simply filling text with unicode char value. So this
// cause issue.
// [Solution]		Accepting character other than carriage return in popup
// edit for string controls
// [Modified files]	PopupEdit.c
// 
// 23    1/22/13 12:23a Rajashakerg
// [TAG]  		EIP103449
// [Category]  	Bug Fix
// [Severity]  	Important
// [Symptom]  	In Setup click\Touch from navigation window, option are not
// displayed properly
// [RootCause]  	Latest mouse driver providing LEFT_DOWN when we hold the
// mouse left down button.
// [Solution]  	Provided fix resolves the issue.
// [Files]  		ezport.c, PopupEdit.c, ListBox.c, MessageBox.c
// 
// 22    10/18/12 6:02a Arunsb
// Updated for 2.16.1235 QA submission
// 
// 18    10/10/12 12:38p Arunsb
// Synched the source for v2.16.1232, backup with Aptio
// 
// 21    4/27/12 2:02p Arunsb
// Build error rectified
// 
// 20    4/27/12 2:40a Rajashakerg
// [TAG]  		EIP86346 
// [Category]  	Improvement
// [Description]  	Cursor visibility is not proper in multiline string
// controls
// [Files]  		PopupEdit.c, PopupString.c
// 
// 19    4/05/12 7:19a Rajashakerg
// [TAG]  		EIP87122,85508,86449
// [Category]  	Improvement
// [Description]  	Numeric in old style, softkbd issues
// [Files]  		CommonHelper.c, Mouse.c, Date.c, edit.c, frame.c,
// minisetupext.h, numeric.c, numeric.h, PopupEdit.c, PopupEdit.h, time.c
// 
// 18    2/02/12 1:18p Premkumara
// [TAG]  		EIP75351,75352,75384
// [Category]  	Improvement
// [Description]  	Suppress the warnings from static code analyzer
// [Files]  		String.c, boot.c, TseAdvanced.c,Time.c, PopupEdit.c,
// MessageBox.c, Label.c, Edit.c, Date.c
// 
// 17    12/27/11 12:37p Arunsb
// [TAG]  		EIP79486
// [Category]  	Bug Fix
// [Severity]  	Normal
// [Symptom]  	Cursor is not blinking when "Softkbd" support enabled
// [RootCause]  	Unknown action also consumed
// [Solution]  	Unknow action not consumed.
// Mouse stopped instead of freezing when the mouse pointer is over the
// cursor.
// [Files]  		popupedit.c and popuppassword.c
// 
// 16    12/01/11 1:38a Premkumara
// [TAG]  		EIP73236
// [Category]  	Improvement
// [Description]  	Large amounts of allocated memory are not freed
// [Files]  		Expression.c, PopupEdit.c, Popup.c, MessageBox.c, Menu.c,
// Memo.c, 
// 
// 15    11/30/11 12:38a Premkumara
// [TAG]  		EIP75351
// [Category]  	Improvement
// [Description]  	Static code analysis.Suppress the warnings from static
// code analyzer
// [Files]  		String.c, HiiString21.c, TseAdvanced.c, Special.c,
// UefiAction., Time.c, PopupEdit.c, MessageBox.c, Label.c, Edit.c, Date.c
// 
// 14    11/28/11 1:42a Premkumara
// [TAG]  		EIP75384 
// [Category]  	Improvement
// [Description]  	Suppress the warnings from static code analyzer
// [Files]  		UefiWapper.c, Hii.c, Expression.c, CtrlCond.c, PopupSel.c,
// Minisetupext.c, Menu.c, Date.c, Ezport\Stylecommon.c,
// EzportPlus\StyleCommon.c,
// 
// 13    11/20/11 8:08a Rajashakerg
// [TAG]  		EIP62763
// [Category]  	Improvement
// [Description]  	Utilize the Improvements done from mouse driver in
// AMITSE
// [Files]  		HookAnchor.h, TseCommon.h, AMITSE.sdl, CommonHelper.c,
// commonoem.c, commonoem.h, buffer.c, globals.c, HookAnchor.c,
// minisetup.h, notify.c, postmgmt.c, protocol.c, ezport.c, stylecommon.c,
// Mouse.c, Action.c, Date.c, frame.c, MessageBox.c, minisetupext.c,
// minisetupext.h, numeric.c, numeric.h, page.c, PopupEdit.c, PopupEdit.h,
// PopupPassword.c, postmgmtext.c, time.c.
// 
// 12    11/10/11 12:49a Arunsb
// [TAG]  		EIP67735
// [Category]  	Improvement
// [Description]  	Zeroing password buffers before freeing
// [Files]  		mem.c, mem.h, popupedit.c, popuppassword.c and tseadvanced.c
// 
// 11    10/19/11 3:52p Blaines
// [TAG] - EIP 66860
// [Category]- Function Request
// [Synopsis]- AmiPostManager interface for text entry. Add cursor
// support.
// [Files] - LogoLib.h, AMIPostMgr.h, protocol.c, protocol.h,
// TseAdvanced.c, TseLitehelper.c, TseUefiHii.h, Uefi21Wapper.c,
// uefi20Wapper.c, PopupEdit.c
// 
// 10    12/24/10 12:37p Mallikarjunanv
// [TAG]  		EIP 46144
// [Category]  	Bug Fix
// [Severity]  	Normal
// [Symptom]  	String exceeds the msg dialog boundry in "Path for boot
// [RootCause]  	The width coordinate of the popup edit is not handled
// propely.
// [Solution]  	Updated the width to write the string with in the
// boundary.
// [Files]  		PopupEdit.c
// 
// 9     10/20/10 1:44p Mallikarjunanv
// reverting back to v25
// 
// 7     6/17/10 2:59p Madhans
// Dynamic parsing support in TSE.
// 
// 6     5/07/10 6:40p Madhans
// EIP: 38197 To make the Scan code support work correctly with 9 and =
// Keys.
// 
// 5     4/16/10 5:13p Madhans
// Changes for Tse 2.02. Please see Changelog.log for more details.
// 
// 4     2/19/10 1:04p Madhans
// Updated for TSE 2.01. Refer Changelog.log for File change history.
// 
// 8     2/19/10 8:18a Mallikarjunanv
// updated year in copyright message
// 
// 7     2/04/10 11:17p Madhans
// Mouse support related code optimized
// 
// 6     1/09/10 7:01a Mallikarjunanv
// Updated TSE2.01 Release sources with coding standards
// 
// 5     8/19/09 6:34p Madhans
// InValidActionHook Added. for PopupEdit case
// 
// 4     7/20/09 1:22p Mallikarjunanv
// updated the code with token PASSWORD_WITH_SPECIAL_CHAR_SUPPORT
// 
// 3     6/23/09 6:52p Blaines
// Coding standard update, 
// Remove spaces from file header to allow proper chm function list
// creation.
// 
// 2     6/12/09 7:44p Presannar
// Initial implementation of coding standards for AMITSE2.0
// 
// 1     6/04/09 8:05p Madhans
// 
// 1     4/28/09 11:06p Madhans
// Tse 2.0 Code complete Checkin.
// 
// 2     1/30/09 6:06p Madhans
// Function headers added. 
// 
// 1     12/18/08 7:58p Madhans
// Intial version of TSE Lite sources
//*****************************************************************//
//*****************************************************************//

//<AMI_FHDR_START>
//----------------------------------------------------------------------------
//
// Name:		PopupEdit.c
//
// Description:	This file contains code to handle Popup Edit operations
//
//----------------------------------------------------------------------------
//<AMI_FHDR_END>

#include "minisetup.h"
extern UINTN gCursorLeft,gCursorTop ; 

POPUPEDIT_METHODS gPopupEdit =
{
	PopupEditCreate,
	PopupEditDestroy,
	PopupEditInitialize,
	PopupEditDraw,
	PopupEditHandleAction,
	PopupEditSetCallback,
	PopupEditSetFocus,
	PopupEditSetPosition,
	PopupEditSetDimensions,
	PopupEditSetAttributes,
	PopupEditGetControlHight,
	PopupEditSetType,
	PopupEditSetText
};

BOOLEAN CheckIsAllowedPasswordChar(CHAR16 Char);
//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditCreate
//
// Description:	this function uses the create function of control
//					and creates the Popup Edit
//
// Input:	POPUPEDIT_DATA **object
//
// Output:	status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditCreate( POPUPEDIT_DATA **object )
{
	EFI_STATUS Status = EFI_OUT_OF_RESOURCES;

	if ( *object == NULL )
	{
		*object = EfiLibAllocateZeroPool( sizeof(POPUPEDIT_DATA) );

		if ( *object == NULL )
			return Status;
	}

	Status = gPopup.Create( object );
	if ( ! EFI_ERROR(Status) )
		(*object)->Methods = &gPopupEdit;

	return Status;
}

//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditDestroy
//
// Description:	this function uses the destroy function of control
//					and destroys the Popup Edit
//
// Input:	POPUPEDIT_DATA **object, BOOLEAN freeMem
//
// Output:	status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditDestroy( POPUPEDIT_DATA *PopupEdit, BOOLEAN freeMem )
{
	EFI_STATUS Status;

	if(NULL == PopupEdit)
	  return EFI_SUCCESS;
	
	Status = gPopup.Destroy( PopupEdit, FALSE );

	if(PopupEdit->Title)
		MemFreePointer( (VOID **)&PopupEdit->Title);

	RestoreScreen( PopupEdit->ScreenBuf );
	if (freeMem)
	{
		if (POPUPEDIT_TYPE_PASSWORD == PopupEdit->PopupEditType)
		{
			gBS->SetMem (PopupEdit->Text, EfiStrLen (PopupEdit->Text) * sizeof (CHAR16), 0);	//EIP67735 zeroing the password prompt memories
		}
        MemFreePointer( (VOID **)&PopupEdit->Text );//EIP-73236 Memory leak
        MemFreePointer( (VOID **)&PopupEdit->TempText );
        MemFreePointer( (VOID **)&PopupEdit->EmptyString ); //Free the allocated memory for the EmptyString field
		MemFreePointer( (VOID **)&PopupEdit );
	}

	return EFI_SUCCESS;
}

//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditInitialize
//
// Description:	this function uses the Initialize function of control
//					and initializes the Popup Edit
//
// Input:	POPUPEDIT_DATA *PopupEdit, VOID *data
//
// Output:	status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditInitialize( POPUPEDIT_DATA *PopupEdit, VOID *data )
{
	EFI_STATUS Status = EFI_UNSUPPORTED;
	UINT16 Length =0/*,height=0*/, Size =0; 
	CHAR16 *TempText;

	// initializes the heigth and width
 	Status = gPopup.Initialize( PopupEdit, data );
	if (EFI_ERROR(Status))
		return Status;

	// add extra initialization here...
	SetControlColorsHook( NULL, NULL , NULL, NULL , NULL, NULL ,
	                      NULL , NULL,  NULL,NULL ,
						  NULL,NULL ,NULL,&(PopupEdit->FGColor),&(PopupEdit->BGColor)  );


	PopupEdit->Container = NULL;
	
	PopupEdit->Border =TRUE;
	PopupEdit->Shadow = TRUE;
	PopupEdit->Title = HiiGetString(PopupEdit->ControlData.ControlHandle, UefiGetPromptField(PopupEdit->ControlData.ControlPtr));

	if ( Length < (TestPrintLength( PopupEdit->Title ) / (NG_SIZE)))
		Length = (UINT16)TestPrintLength( PopupEdit->Title ) / (NG_SIZE);

	if ( Length < (UINT16)UefiGetMaxValue(PopupEdit->ControlData.ControlPtr) )
		Length = (UINT16)UefiGetMaxValue(PopupEdit->ControlData.ControlPtr);

	PopupEdit->Width = (UINT16)(Length + 4);
	if ( PopupEdit->Width > MAX_POPUPEDIT_WIDTH )
	{
		//EIP-135665 If multiline is supported
		if ( IsTSEMultilineControlSupported() )
			PopupEdit->Height += (UINT16)(PopupEdit->Width / MAX_POPUPEDIT_WIDTH )+(((PopupEdit->Width % MAX_POPUPEDIT_WIDTH) > 0)?1:0)-1;
		PopupEdit->Width = (UINT16)MAX_POPUPEDIT_WIDTH;	
	}
	PopupEdit->TextWidth = (UINT16)UefiGetMaxValue(((CONTROL_INFO *)data)->ControlPtr);

     
	// create text memory area with max available size
	TempText = EfiLibAllocateZeroPool(PopupEdit->TextWidth*2+2);
    EfiStrCpy(TempText,PopupEdit->Text);
	MemFreePointer((CHAR16**)&PopupEdit->Text);
	PopupEdit->Text =TempText;

	PopupEdit->ScreenBuf = SaveScreen();

    PopupEdit->TempText =  EfiLibAllocateZeroPool( PopupEdit->TextWidth*2 +2 );

	PopupEdit->Height += 2+ (PopupEdit->TextWidth/PopupEdit->Width +1)/*height*/;
	gPopupEdit.SetDimensions( PopupEdit, PopupEdit->Width , PopupEdit->Height );

	return Status;
}

//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditDraw
//
// Description:	function to draw the popup to edit with attributes
//
// Input:	POPUPEDIT_DATA *PopupEdit
//
// Output:	status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditDraw( POPUPEDIT_DATA *PopupEdit )
{
	EFI_STATUS Status;
	CHAR16 *text = NULL, *newText = NULL;
	CHAR16 *line;
	UINT16 pos, height=1,i;
	UINTN Wrappedtextlen= 0;
	
    text = StrDup(PopupEdit->Text);

	Status = gPopup.Draw( PopupEdit );
	if(EFI_ERROR(Status))
	{
		//EIP-75384 Static code
		MemFreePointer( (VOID **)&text );
		return Status;
	}

	// clear out old wrapped string
	for ( pos = PopupEdit->Height- (2+ (PopupEdit->TextWidth/PopupEdit->Width +1))+1; pos < PopupEdit->Height-2; pos++ )
		DrawStringWithAttribute( PopupEdit->Left+1, PopupEdit->Top + pos, PopupEdit->EmptyString, PopupEdit->FGColor | PopupEdit->BGColor );

	if(PopupEdit->PopupEditType != POPUPEDIT_TYPE_PASSWORD ) {
		//EIP_46144 - Updated the width to write the string with in the boundary.
		newText = StringWrapText( text, PopupEdit->Width-2, &height );	
	}
	else {
		newText = text;
	}
		
	//EIP-75351 Suppress the warnings from static code analyzer
	if (newText == NULL)
	{
		MemFreePointer( (VOID **)&text );
		return EFI_NOT_FOUND;
	}

	if ( newText != text )
		MemFreePointer( (VOID **)&text );

	line = newText;

	for ( pos = PopupEdit->Height- (2+ (PopupEdit->TextWidth/PopupEdit->Width +1))+1; pos < PopupEdit->Height; pos++ )
	{
		CHAR16 save;

		text = line;
		if ( *line == L'\0' )
			break;

		if(PopupEdit->PopupEditType != POPUPEDIT_TYPE_PASSWORD )
		{
			while ( ( *line != L'\n' ) &&( *line != L'\r' ) && ( *line != L'\0' ) )
				line++;
		}
		else
		{
			while ( *line != L'\0' ) 
				line++;
		}

		save = *line;
		*line = L'\0';
		i=0;
		while( (text[i] != L'\0') && (PopupEdit->Chr !=0x0 ) )
		{		
//			if(text[i] != L' ') //all chars should be treated the same
				text[i] = PopupEdit->Chr;
			i++;
		}

		DrawStringWithAttribute( PopupEdit->Left+1, PopupEdit->Top + pos, text, PopupEdit->FGColor | PopupEdit->BGColor );

		if ( ( *line = save ) != L'\0' )
		{
			line++;
			if ( ( *line == L'\r' ) || ( *line == L'\n' ) )
				line++;
		}
		
		if(L'\0' == *line)
			Wrappedtextlen = EfiStrLen(text);//EIP 86346 : Getting the wrapped text length
			 
	}
	MemFreePointer( (VOID **)&newText );

	if ( pos < height )
		height = pos;

	gCursorLeft = PopupEdit->Left + Wrappedtextlen;
	gCursorTop = PopupEdit->Top + (pos-1);//EIP 86346 : Providing the Cursor Postion depending on the lenght of string
	if((Wrappedtextlen == 0)||(((UINT16)Wrappedtextlen) > (PopupEdit->Width-2)))
	{
		gCursorLeft = PopupEdit->Left+1;
		gCursorTop++;
	}
	PopupEdit->DisplayLines = height;
	FlushLines( PopupEdit->Top, PopupEdit->Top + PopupEdit->Height  );
	return  EFI_SUCCESS;

}


//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditHandleAction
//
// Description:	function to handle the Popup Edit actions
//
// Input:	POPUPEDIT_DATA *PopupEdit, ACTION_DATA *Data
//
// Output:	status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditHandleAction( POPUPEDIT_DATA *PopupEdit, ACTION_DATA *Data)
{
	UINT8 Redraw =0;
	//Always return success to consume the message, the popupedit is always modal.
	EFI_STATUS Status = EFI_UNSUPPORTED;
	CONTROL_ACTION Action=ControlActionUnknown;

	if ( (Data->Input.Type == ACTION_TYPE_TIMER) || (Data->Input.Type == ACTION_TYPE_NULL) )	//EIP79486 cursor not blinking when mouse driver presents
		return Status;

	if ( Data->Input.Type == ACTION_TYPE_MOUSE )
	{
		Action = MapControlMouseActionHook(&Data->Input.Data.MouseInfo);

	//EIP 103449  : In Setup click\Touch from navigation window, option are not displayed properly
	   if( (
	        (Data->Input.Data.MouseInfo.Top < PopupEdit->Top) ||
	        (Data->Input.Data.MouseInfo.Top > (UINT32)(PopupEdit->Top + PopupEdit->Height)) ||
	        (Data->Input.Data.MouseInfo.Left < PopupEdit->Left) ||
	        (Data->Input.Data.MouseInfo.Left > (UINT32)(PopupEdit->Left + PopupEdit->Width)) )
			&& ( (ControlActionChoose == Action) || (ControlActionSelect == Action) )
	      )
		{	// Any action outside the popupedit about
			Action = ControlActionAbort;
		}

	}

	if (Data->Input.Type == ACTION_TYPE_KEY)
	{
		//Get mapping
		Action = MapControlKeysHook(Data->Input.Data.AmiKey);
	}

	switch(Action)
	{
	case ControlActionNextUp:
	case ControlActionNextDown:
	case ControlActionNextLeft:
	case ControlActionNextRight:
	case ControlActionNextSelection: 
		Redraw = TRUE;
		break;

	case ControlActionAbort:
		if(PopupEdit->ControlActive == TRUE)
		{
			MemCopy(PopupEdit->Text,PopupEdit->TempText,PopupEdit->TextWidth*sizeof(UINT16)); //UNICODE CHARS
		}
		if(PopupEdit->Callback != NULL)
		{
			if ( PopupEdit->Cookie != NULL )
			{
				VOID *ifrData = (VOID *)PopupEdit->ControlData.ControlPtr;
				CALLBACK_VARIABLE *callbackData = (CALLBACK_VARIABLE *)PopupEdit->Cookie;

				callbackData->Variable = PopupEdit->ControlData.ControlVariable;
				callbackData->Offset = UefiGetQuestionOffset(ifrData);
				callbackData->Length = UefiGetWidth(ifrData);
				callbackData->Data = (VOID *)PopupEdit->Text;

				PopupEdit->Callback( PopupEdit->Container,PopupEdit, NULL );	
			}
		}

		PopupEdit->ControlActive = FALSE;
		Redraw = 1;
		Status = EFI_SUCCESS;
		break;

	case ControlActionSelect:
		if(PopupEdit->Callback != NULL)
		{
			if ( PopupEdit->Cookie != NULL )
			{
				VOID *ifrData = PopupEdit->ControlData.ControlPtr;
				CALLBACK_VARIABLE *callbackData = (CALLBACK_VARIABLE *)PopupEdit->Cookie;

				callbackData->Variable = PopupEdit->ControlData.ControlVariable;
				callbackData->Offset = UefiGetQuestionOffset(ifrData);
				callbackData->Length = UefiGetWidth(ifrData);
				callbackData->Data = (VOID *)PopupEdit->Text;

				PopupEdit->Callback( PopupEdit->Container,PopupEdit, PopupEdit->Cookie );	
			}

		}

		Redraw =1;
		PopupEdit->ControlActive =FALSE;
		Status = EFI_SUCCESS;
		break;

	case ControlActionBackSpace:
		if(EfiStrLen(PopupEdit->Text)>0)
		{
			if(PopupEdit->ControlActive ==FALSE)
			{
				MemCopy(PopupEdit->TempText,PopupEdit->Text,PopupEdit->TextWidth * sizeof(UINT16)); //UNICODE CHARS
				PopupEdit->ControlActive = TRUE; 
			}
			PopupEdit->Text[EfiStrLen(PopupEdit->Text)-1] = 0x0;
			Redraw =1;
		}
		break;

	default:
		if ( Data->Input.Type == ACTION_TYPE_MOUSE )
			break;

		if(PopupEditHandlePassword( PopupEdit, Data) == EFI_SUCCESS)
			Redraw =1;
		else
		{
			if ((Data->Input.Data.AmiKey.Key.UnicodeChar) && (CHAR_CARRIAGE_RETURN != Data->Input.Data.AmiKey.Key.UnicodeChar)) //EIP-123306 Avoid carriage return when press CTRL+ENTER in popupedit controls.
			{
				if ( ( CheckIsAllowedPasswordChar(Data->Input.Data.AmiKey.Key.UnicodeChar) || ( PopupEdit->PopupEditType != POPUPEDIT_TYPE_PASSWORD ) ) &&
					( EfiStrLen(PopupEdit->Text) < PopupEdit->TextWidth ) )
				{
					if(PopupEdit->ControlActive ==FALSE)
					{
						MemCopy(PopupEdit->TempText,PopupEdit->Text,PopupEdit->TextWidth* sizeof(UINT16));
						PopupEdit->ControlActive = TRUE; 
					}

					PopupEdit->Text[EfiStrLen(PopupEdit->Text)] = Data->Input.Data.AmiKey.Key.UnicodeChar ; 
					PopupEdit->Text[EfiStrLen(PopupEdit->Text)] = 0x0 ; 
					Redraw =1;
				}
				else {
					InvalidActionHookHook();		///
				}
			}
			else if(!IsToggleStateKey(Data))	
			{
				InvalidActionHookHook();
			}

		}
		break;
	}

	if ( Redraw == TRUE )
	{	
		MouseFreeze();

		gPopupEdit.Draw( PopupEdit );

		MouseRefresh();	
	}

	return EFI_SUCCESS;
}

//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditSetCallback
//
// Description:	Function to set the PopupSel callback
//
// Input:		POPUPEDIT_DATA *PopupEdit, OBJECT_DATA *container, OBJECT_CALLBACK callback, VOID *cookie
//
// Output:		STATUS
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditSetCallback( POPUPEDIT_DATA *PopupEdit, OBJECT_DATA *container, OBJECT_CALLBACK callback, VOID *cookie )
{
	return gControl.SetCallback( PopupEdit, container, callback, cookie );
}

//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditSetType
//
// Description:	Function to set the PopupEdit type
//
// Input:		POPUPEDIT_DATA *PopupEdit, UINT8 Type
//
// Output:		STATUS
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditSetType( POPUPEDIT_DATA *PopupEdit, UINT8 Type )
{
	PopupEdit->PopupEditType = Type;
	if ( Type == POPUPEDIT_TYPE_PASSWORD )
		PopupEdit->Chr = PASSWORD_CHAR;

	return EFI_SUCCESS;
}

//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditSetText
//
// Description:	Function to set the PopupEdit text
//
// Input:		POPUPEDIT_DATA *PopupEdit, CHAR16 *String 
//
// Output:		STATUS
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditSetText(POPUPEDIT_DATA *PopupEdit, CHAR16 *String )
{
    if(PopupEdit->Text==NULL)
    {
        if(String != NULL)
		{
            PopupEdit->Text = EfiLibAllocateZeroPool( (EfiStrLen(String )+1) * sizeof(CHAR16) );
		    EfiStrCpy(PopupEdit->Text, String);
		}
	    else
		{
			PopupEdit->Text = EfiLibAllocateZeroPool( (2) * sizeof(CHAR16) );
            EfiStrCpy(PopupEdit->Text, L"");
		}
	}
    else
	{
        if(String != NULL)
		  EfiStrCpy(PopupEdit->Text,String);
	    else
           EfiStrCpy(PopupEdit->Text, L"");

	}

	return EFI_SUCCESS;
}
//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditSetFocus
//
// Description:	Function to set focus
//
// Input:		POPUPEDIT_DATA *PopupEdit, BOOLEAN focus
//
// Output:		Status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditSetFocus(POPUPEDIT_DATA *PopupEdit, BOOLEAN focus)
{
	PopupEdit->ControlFocus = focus;
	return EFI_SUCCESS;
}
//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditSetPosition
//
// Description:	Function to set focus
//
// Input:		POPUPEDIT_DATA *PopupEdit, UINT16 Left, UINT16 Top
//
// Output:		Status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditSetPosition(POPUPEDIT_DATA *PopupEdit, UINT16 Left, UINT16 Top )
{
	return gPopup.SetPosition( PopupEdit, Left, Top );
}

//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditSetDimensions
//
// Description:	function to set the dimentions of the editable popup menu
//
// Input:	POPUPEDIT_DATA *PopupEdit, UINT16 Width, UINT16 Height
//
// Output:	status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditSetDimensions(POPUPEDIT_DATA *PopupEdit, UINT16 Width, UINT16 Height )
{ 
	EFI_STATUS Status= EFI_SUCCESS;
    
	// dimensions include top, bottom, left and right lines for popup
	Status = gPopup.SetDimensions( PopupEdit, Width, Height ); 

	MemFreePointer( (VOID **)&PopupEdit->EmptyString );
	PopupEdit->EmptyString = EfiLibAllocateZeroPool( (Width-1 ) * sizeof(CHAR16) );
	if ( PopupEdit->EmptyString == NULL )
		return EFI_OUT_OF_RESOURCES;

	MemFillUINT16Buffer( PopupEdit->EmptyString, Width-2, L' ' );

	return(Status);
}
//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditSetAttributes
//
// Description:	Function to set the dimentions of the editable popup menu
//
// Input:	POPUPEDIT_DATA *PopupEdit, UINT8 FGColor, UINT8 BGColor
//
// Output:	status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditSetAttributes(POPUPEDIT_DATA *PopupEdit, UINT8 FGColor, UINT8 BGColor )
{
	return gControl.SetAttributes( PopupEdit, FGColor, BGColor );
}

//<AMI_PHDR_START>
//----------------------------------------------------------------------------
// Procedure:	PopupEditGetControlHight
//
// Description:	Function unsuppored.
//
// Input:	VOID *object,VOID *frame, UINT16 *height
//
// Output:	status
//
//----------------------------------------------------------------------------
//<AMI_PHDR_END>
EFI_STATUS PopupEditGetControlHight( VOID *object,VOID *frame, UINT16 *height )
{
	return EFI_UNSUPPORTED;
}


//*****************************************************************//
//*****************************************************************//
//*****************************************************************//
//**                                                             **//
//**       (C)Copyright 1985-2014, American Megatrends, Inc.     **//
//**                                                             **//
//**                     All Rights Reserved.                    **//
//**                                                             **//
//**    5555 Oakbrook Pkwy, Norcross, Suite 200, Georgia 30093   **//
//**                                                             **//
//**                     Phone (770)-246-8600                    **//
//**                                                             **//
//*****************************************************************//
//*****************************************************************//
//*****************************************************************//