[mirror_edk2.git] / SecurityPkg / UserIdentification / UserProfileManagerDxe / UserProfileDelete.c

/** @file\r
  The functions to delete a user profile.\r
    \r
Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>\r
This program and the accompanying materials \r
are licensed and made available under the terms and conditions of the BSD License \r
which accompanies this distribution.  The full text of the license may be found at \r
http://opensource.org/licenses/bsd-license.php\r
\r
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
\r
**/\r
\r
#include "UserProfileManager.h"\r
\r
/**\r
  Get the username from the specified user.\r
\r
  @param[in]   User              Handle of a user profile. \r
\r
  @retval EFI_STRING_ID          The String Id of the user's username.\r
\r
**/\r
EFI_STRING_ID \r
GetUserName (\r
  IN  EFI_USER_PROFILE_HANDLE                   User\r
  )\r
{\r
  EFI_STATUS            Status;\r
  EFI_USER_INFO_HANDLE  UserInfo;\r
  EFI_USER_INFO         *Info;\r
  UINTN                 InfoSize;\r
  UINTN                 MemSize;\r
  UINTN                 NameLen;\r
  CHAR16                UserName[USER_NAME_LENGTH];\r
  EFI_STRING_ID         UserId;\r
  \r
  //\r
  // Allocate user information memory.\r
  //\r
  MemSize = sizeof (EFI_USER_INFO) + 63;\r
  Info    = AllocateZeroPool (MemSize);\r
  ASSERT (Info != NULL);\r
  \r
  //\r
  // Get user name information.\r
  //\r
  UserInfo = NULL;\r
  while (TRUE) {\r
    InfoSize = MemSize;\r
    //\r
    // Get next user information.\r
    //\r
    Status = mUserManager->GetNextInfo (\r
                             mUserManager,\r
                             User,\r
                             &UserInfo\r
                             );\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    Status = mUserManager->GetInfo (\r
                             mUserManager,\r
                             User,\r
                             UserInfo,\r
                             Info,\r
                             &InfoSize\r
                             );\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      MemSize = InfoSize;\r
      FreePool (Info);\r
      Info = AllocateZeroPool (MemSize);\r
      ASSERT (Info != NULL);\r
\r
      Status = mUserManager->GetInfo (\r
                               mUserManager,\r
                               User,\r
                               UserInfo,\r
                               Info,\r
                               &InfoSize\r
                               );\r
    }\r
    //\r
    // Check user information.\r
    //\r
    if (Status == EFI_SUCCESS) {\r
      if (Info->InfoType == EFI_USER_INFO_NAME_RECORD) {\r
        NameLen = Info->InfoSize - sizeof (EFI_USER_INFO);\r
        if (NameLen > USER_NAME_LENGTH * sizeof (CHAR16)) {\r
          NameLen = USER_NAME_LENGTH * sizeof (CHAR16);\r
        }\r
        ASSERT (NameLen >= sizeof (CHAR16));\r
        CopyMem (UserName, (UINT8 *) (Info + 1), NameLen);\r
        UserName[NameLen / sizeof (CHAR16) - 1] = 0;\r
        UserId = HiiSetString (\r
                   mCallbackInfo->HiiHandle,\r
                   0,\r
                   UserName,\r
                   NULL\r
                   );\r
        if (UserId != 0) {\r
          FreePool (Info);\r
          return UserId;\r
        }\r
      }\r
    }\r
  }\r
\r
  FreePool (Info);\r
  return 0;\r
}\r
\r
\r
/**\r
  Add a username item in form.\r
\r
  @param[in]  User          Points to the user profile whose username is added. \r
  @param[in]  Index         The index of the user in the user name list\r
  @param[in]  OpCodeHandle  Points to container for dynamic created opcodes.\r
\r
**/\r
VOID\r
AddUserToForm (\r
  IN  EFI_USER_PROFILE_HANDLE                   User,\r
  IN  UINT16                                    Index,\r
  IN  VOID                                      *OpCodeHandle\r
  )\r
{\r
  EFI_STRING_ID NameId;\r
\r
  //\r
  // Get user name\r
  //\r
  NameId = GetUserName (User);\r
  if (NameId == 0) {\r
    return ;\r
  }\r
  \r
  //\r
  // Create user name option.\r
  //\r
  switch (Index & KEY_FIRST_FORM_MASK) {\r
  case KEY_MODIFY_USER:\r
    HiiCreateGotoOpCode (\r
      OpCodeHandle,                   // Container for dynamic created opcodes\r
      FORMID_USER_INFO,               // Target Form ID\r
      NameId,                         // Prompt text\r
      STRING_TOKEN (STR_NULL_STRING), // Help text\r
      EFI_IFR_FLAG_CALLBACK,          // Question flag\r
      Index                           // Question ID\r
      );\r
    break;\r
\r
  case KEY_DEL_USER:\r
    HiiCreateActionOpCode (\r
      OpCodeHandle,                   // Container for dynamic created opcodes\r
      Index,                          // Question ID\r
      NameId,                         // Prompt text\r
      STRING_TOKEN (STR_NULL_STRING), // Help text\r
      EFI_IFR_FLAG_CALLBACK,          // Question flag\r
      0                               // Action String ID\r
      );\r
    break;\r
\r
  default:\r
    break;\r
  }\r
}\r
\r
\r
/**\r
  Delete the user specified by UserIndex in user profile database.\r
\r
  @param[in]  UserIndex       The index of user in the user name list \r
                              to be deleted.\r
\r
**/\r
VOID\r
DeleteUser (\r
  IN UINT8                                      UserIndex\r
  )\r
{\r
  EFI_STATUS              Status;\r
  EFI_USER_PROFILE_HANDLE User;\r
  EFI_INPUT_KEY           Key;\r
  EFI_USER_INFO_HANDLE    UserInfo;\r
  EFI_USER_INFO           *Info;\r
  UINTN                   InfoSize;\r
\r
  //\r
  // Find specified user profile and delete it.\r
  //\r
  User    = NULL;\r
  Status  = mUserManager->GetNext (mUserManager, &User);\r
  if (EFI_ERROR (Status)) {\r
    goto Done;\r
  }\r
  \r
  while (UserIndex > 1) {\r
    Status = mUserManager->GetNext (mUserManager, &User);\r
    if (EFI_ERROR (Status)) {\r
      goto Done;\r
    }\r
    UserIndex--;\r
  }\r
\r
  if (UserIndex == 1) {\r
    //\r
    // Get the identification policy.\r
    //\r
    Status = FindInfoByType (User, EFI_USER_INFO_IDENTITY_POLICY_RECORD, &UserInfo);\r
    if (EFI_ERROR (Status)) {\r
      goto Done;\r
    }\r
\r
    InfoSize = 0;\r
    Info = NULL;\r
    Status   = mUserManager->GetInfo (mUserManager, User, UserInfo, Info, &InfoSize);\r
    if (Status == EFI_BUFFER_TOO_SMALL) {\r
      Info = AllocateZeroPool (InfoSize);\r
      if (Info == NULL) {\r
        goto Done;\r
      }\r
      Status = mUserManager->GetInfo (mUserManager, User, UserInfo, Info, &InfoSize);\r
    }\r
\r
    //\r
    // Delete the user on the credential providers by its identification policy.\r
    //\r
    DeleteCredentialFromProviders ((UINT8 *)(Info + 1), Info->InfoSize - sizeof (EFI_USER_INFO), User);\r
    FreePool (Info);\r
    \r
    Status = mUserManager->Delete (mUserManager, User);\r
    if (EFI_ERROR (Status)) {\r
      goto Done;\r
    }\r
    CreatePopUp (\r
      EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE,\r
      &Key,\r
      L"Delete User Succeed!",\r
      L"",\r
      L"Please Press Any Key to Continue ...",\r
      NULL\r
      );\r
    return ;  \r
  }\r
\r
Done:\r
  CreatePopUp (\r
    EFI_LIGHTGRAY | EFI_BACKGROUND_BLUE,\r
    &Key,\r
    L"Delete User Failed!",\r
    L"",\r
    L"Please Press Any Key to Continue ...",\r
    NULL\r
    );\r
}\r
\r
\r
/**\r
  Display user select form, cab select a user to delete.\r
\r
**/\r
VOID\r
SelectUserToDelete (\r
  VOID\r
  )\r
{\r
  EFI_STATUS              Status;\r
  UINT8                   Index;\r
  EFI_USER_PROFILE_HANDLE User;\r
  EFI_USER_PROFILE_HANDLE CurrentUser;\r
  VOID                    *StartOpCodeHandle;\r
  VOID                    *EndOpCodeHandle;\r
  EFI_IFR_GUID_LABEL      *StartLabel;\r
  EFI_IFR_GUID_LABEL      *EndLabel;\r
\r
  //\r
  // Initialize the container for dynamic opcodes.\r
  //\r
  StartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
  ASSERT (StartOpCodeHandle != NULL);\r
\r
  EndOpCodeHandle = HiiAllocateOpCodeHandle ();\r
  ASSERT (EndOpCodeHandle != NULL);\r
\r
  //\r
  // Create Hii Extend Label OpCode.\r
  //\r
  StartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (\r
                                        StartOpCodeHandle,\r
                                        &gEfiIfrTianoGuid,\r
                                        NULL,\r
                                        sizeof (EFI_IFR_GUID_LABEL)\r
                                        );\r
  StartLabel->ExtendOpCode  = EFI_IFR_EXTEND_OP_LABEL;\r
  StartLabel->Number        = LABEL_USER_DEL_FUNC;\r
\r
  EndLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (\r
                                      EndOpCodeHandle,\r
                                      &gEfiIfrTianoGuid,\r
                                      NULL,\r
                                      sizeof (EFI_IFR_GUID_LABEL)\r
                                      );\r
  EndLabel->ExtendOpCode  = EFI_IFR_EXTEND_OP_LABEL;\r
  EndLabel->Number        = LABEL_END;\r
\r
  //\r
  // Add each user can be deleted.\r
  //\r
  User  = NULL;\r
  Index = 1;\r
  mUserManager->Current (mUserManager, &CurrentUser);\r
  while (TRUE) {\r
    Status = mUserManager->GetNext (mUserManager, &User);\r
    if (EFI_ERROR (Status)) {\r
      break;\r
    }\r
\r
    if (User != CurrentUser) {\r
      AddUserToForm (\r
        User,\r
        (UINT16)(KEY_DEL_USER | KEY_SELECT_USER | Index),\r
        StartOpCodeHandle\r
        );\r
    }\r
    Index++;\r
  }\r
\r
  HiiUpdateForm (\r
    mCallbackInfo->HiiHandle, // HII handle\r
    &gUserProfileManagerGuid, // Formset GUID\r
    FORMID_DEL_USER,          // Form ID\r
    StartOpCodeHandle,        // Label for where to insert opcodes\r
    EndOpCodeHandle           // Replace data\r
    );\r
\r
  HiiFreeOpCodeHandle (StartOpCodeHandle);\r
  HiiFreeOpCodeHandle (EndOpCodeHandle);\r
}\r
Commit	Line	Data
0c18794e	1	/** @file\r
	2	The functions to delete a user profile.\r
	3	\r
a0c56a82	4	Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>\r
0c18794e	5	This program and the accompanying materials \r
	6	are licensed and made available under the terms and conditions of the BSD License \r
	7	which accompanies this distribution. The full text of the license may be found at \r
	8	http://opensource.org/licenses/bsd-license.php\r
	9	\r
	10	THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, \r
	11	WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	12	\r
	13	**/\r
	14	\r
	15	#include "UserProfileManager.h"\r
	16	\r
	17	/**\r
	18	Get the username from the specified user.\r
	19	\r
	20	@param[in] User Handle of a user profile. \r
	21	\r
	22	@retval EFI_STRING_ID The String Id of the user's username.\r
	23	\r
	24	**/\r
	25	EFI_STRING_ID \r
	26	GetUserName (\r
	27	IN EFI_USER_PROFILE_HANDLE User\r
	28	)\r
	29	{\r
	30	EFI_STATUS Status;\r
	31	EFI_USER_INFO_HANDLE UserInfo;\r
	32	EFI_USER_INFO *Info;\r
	33	UINTN InfoSize;\r
	34	UINTN MemSize;\r
	35	UINTN NameLen;\r
	36	CHAR16 UserName[USER_NAME_LENGTH];\r
	37	EFI_STRING_ID UserId;\r
	38	\r
	39	//\r
	40	// Allocate user information memory.\r
	41	//\r
	42	MemSize = sizeof (EFI_USER_INFO) + 63;\r
	43	Info = AllocateZeroPool (MemSize);\r
	44	ASSERT (Info != NULL);\r
	45	\r
	46	//\r
	47	// Get user name information.\r
	48	//\r
	49	UserInfo = NULL;\r
	50	while (TRUE) {\r
	51	InfoSize = MemSize;\r
	52	//\r
	53	// Get next user information.\r
	54	//\r
	55	Status = mUserManager->GetNextInfo (\r
	56	mUserManager,\r
	57	User,\r
	58	&UserInfo\r
	59	);\r
	60	if (EFI_ERROR (Status)) {\r
	61	break;\r
	62	}\r
	63	\r
	64	Status = mUserManager->GetInfo (\r
	65	mUserManager,\r
	66	User,\r
	67	UserInfo,\r
	68	Info,\r
69	&InfoSize\r
70	);\r
71	if (Status == EFI_BUFFER_TOO_SMALL) {\r
72	MemSize = InfoSize;\r
73	FreePool (Info);\r
74	Info = AllocateZeroPool (MemSize);\r
75	ASSERT (Info != NULL);\r
76	\r
77	Status = mUserManager->GetInfo (\r
78	mUserManager,\r
79	User,\r
80	UserInfo,\r
81	Info,\r
82	&InfoSize\r
83	);\r
84	}\r
85	//\r
86	// Check user information.\r
87	//\r
88	if (Status == EFI_SUCCESS) {\r
89	if (Info->InfoType == EFI_USER_INFO_NAME_RECORD) {\r
90	NameLen = Info->InfoSize - sizeof (EFI_USER_INFO);\r
91	if (NameLen > USER_NAME_LENGTH * sizeof (CHAR16)) {\r
92	NameLen = USER_NAME_LENGTH * sizeof (CHAR16);\r
93	}\r
94	ASSERT (NameLen >= sizeof (CHAR16));\r
95	CopyMem (UserName, (UINT8 *) (Info + 1), NameLen);\r
96	UserName[NameLen / sizeof (CHAR16) - 1] = 0;\r
97	UserId = HiiSetString (\r
98	mCallbackInfo->HiiHandle,\r
99	0,\r
100	UserName,\r
101	NULL\r
102	);\r
103	if (UserId != 0) {\r
104	FreePool (Info);\r
105	return UserId;\r
106	}\r
107	}\r
108	}\r
109	}\r
110	\r
111	FreePool (Info);\r
112	return 0;\r
113	}\r
114	\r
115	\r
116	/**\r
117	Add a username item in form.\r
118	\r
119	@param[in] User Points to the user profile whose username is added. \r
120	@param[in] Index The index of the user in the user name list\r
121	@param[in] OpCodeHandle Points to container for dynamic created opcodes.\r
122	\r
123	**/\r
124	VOID\r
125	AddUserToForm (\r
126	IN EFI_USER_PROFILE_HANDLE User,\r
127	IN UINT16 Index,\r
128	IN VOID *OpCodeHandle\r
129	)\r
130	{\r
131	EFI_STRING_ID NameId;\r
132	\r
133	//\r
134	// Get user name\r
135	//\r
136	NameId = GetUserName (User);\r
137	if (NameId == 0) {\r
138	return ;\r
139	}\r
140	\r
141	//\r
142	// Create user name option.\r
143	//\r
144	switch (Index & KEY_FIRST_FORM_MASK) {\r
145	case KEY_MODIFY_USER:\r
146	HiiCreateGotoOpCode (\r
147	OpCodeHandle, // Container for dynamic created opcodes\r
148	FORMID_USER_INFO, // Target Form ID\r
149	NameId, // Prompt text\r
150	STRING_TOKEN (STR_NULL_STRING), // Help text\r
151	EFI_IFR_FLAG_CALLBACK, // Question flag\r
152	Index // Question ID\r
153	);\r
154	break;\r
155	\r
156	case KEY_DEL_USER:\r
157	HiiCreateActionOpCode (\r
158	OpCodeHandle, // Container for dynamic created opcodes\r
159	Index, // Question ID\r
160	NameId, // Prompt text\r
161	STRING_TOKEN (STR_NULL_STRING), // Help text\r
162	EFI_IFR_FLAG_CALLBACK, // Question flag\r
163	0 // Action String ID\r
164	);\r
165	break;\r
166	\r
167	default:\r
168	break;\r
169	}\r
170	}\r
171	\r
172	\r
173	/**\r
174	Delete the user specified by UserIndex in user profile database.\r
175	\r
176	@param[in] UserIndex The index of user in the user name list \r
177	to be deleted.\r
178	\r
179	**/\r
180	VOID\r
181	DeleteUser (\r
182	IN UINT8 UserIndex\r
183	)\r
184	{\r
185	EFI_STATUS Status;\r
186	EFI_USER_PROFILE_HANDLE User;\r
187	EFI_INPUT_KEY Key;\r
0c5b25f0	188	EFI_USER_INFO_HANDLE UserInfo;\r
	189	EFI_USER_INFO *Info;\r
	190	UINTN InfoSize;\r
0c18794e	191	\r
	192	//\r
	193	// Find specified user profile and delete it.\r
	194	//\r
	195	User = NULL;\r
	196	Status = mUserManager->GetNext (mUserManager, &User);\r
	197	if (EFI_ERROR (Status)) {\r
	198	goto Done;\r
	199	}\r
	200	\r
	201	while (UserIndex > 1) {\r
	202	Status = mUserManager->GetNext (mUserManager, &User);\r
	203	if (EFI_ERROR (Status)) {\r
	204	goto Done;\r
	205	}\r
	206	UserIndex--;\r
	207	}\r
	208	\r
	209	if (UserIndex == 1) {\r
0c5b25f0	210	//\r
	211	// Get the identification policy.\r
	212	//\r
	213	Status = FindInfoByType (User, EFI_USER_INFO_IDENTITY_POLICY_RECORD, &UserInfo);\r
	214	if (EFI_ERROR (Status)) {\r
	215	goto Done;\r
	216	}\r
	217	\r
	218	InfoSize = 0;\r
	219	Info = NULL;\r
	220	Status = mUserManager->GetInfo (mUserManager, User, UserInfo, Info, &InfoSize);\r
	221	if (Status == EFI_BUFFER_TOO_SMALL) {\r
	222	Info = AllocateZeroPool (InfoSize);\r
	223	if (Info == NULL) {\r
	224	goto Done;\r
	225	}\r
	226	Status = mUserManager->GetInfo (mUserManager, User, UserInfo, Info, &InfoSize);\r
	227	}\r
	228	\r
	229	//\r
	230	// Delete the user on the credential providers by its identification policy.\r
	231	//\r
	232	DeleteCredentialFromProviders ((UINT8 *)(Info + 1), Info->InfoSize - sizeof (EFI_USER_INFO), User);\r
	233	FreePool (Info);\r
	234	\r
0c18794e	235	Status = mUserManager->Delete (mUserManager, User);\r
	236	if (EFI_ERROR (Status)) {\r
	237	goto Done;\r
	238	}\r
	239	CreatePopUp (\r
	240	EFI_LIGHTGRAY \| EFI_BACKGROUND_BLUE,\r
	241	&Key,\r
	242	L"Delete User Succeed!",\r
	243	L"",\r
	244	L"Please Press Any Key to Continue ...",\r
	245	NULL\r
	246	);\r
	247	return ; \r
	248	}\r
	249	\r
	250	Done:\r
	251	CreatePopUp (\r
	252	EFI_LIGHTGRAY \| EFI_BACKGROUND_BLUE,\r
	253	&Key,\r
	254	L"Delete User Failed!",\r
	255	L"",\r
	256	L"Please Press Any Key to Continue ...",\r
	257	NULL\r
	258	);\r
	259	}\r
	260	\r
	261	\r
	262	/**\r
	263	Display user select form, cab select a user to delete.\r
	264	\r
	265	**/\r
	266	VOID\r
	267	SelectUserToDelete (\r
	268	VOID\r
	269	)\r
	270	{\r
	271	EFI_STATUS Status;\r
	272	UINT8 Index;\r
	273	EFI_USER_PROFILE_HANDLE User;\r
	274	EFI_USER_PROFILE_HANDLE CurrentUser;\r
	275	VOID *StartOpCodeHandle;\r
	276	VOID *EndOpCodeHandle;\r
	277	EFI_IFR_GUID_LABEL *StartLabel;\r
	278	EFI_IFR_GUID_LABEL *EndLabel;\r
	279	\r
	280	//\r
	281	// Initialize the container for dynamic opcodes.\r
	282	//\r
	283	StartOpCodeHandle = HiiAllocateOpCodeHandle ();\r
	284	ASSERT (StartOpCodeHandle != NULL);\r
	285	\r
	286	EndOpCodeHandle = HiiAllocateOpCodeHandle ();\r
	287	ASSERT (EndOpCodeHandle != NULL);\r
	288	\r
	289	//\r
	290	// Create Hii Extend Label OpCode.\r
	291	//\r
	292	StartLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (\r
	293	StartOpCodeHandle,\r
	294	&gEfiIfrTianoGuid,\r
	295	NULL,\r
	296	sizeof (EFI_IFR_GUID_LABEL)\r
	297	);\r
	298	StartLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
299	StartLabel->Number = LABEL_USER_DEL_FUNC;\r
300	\r
301	EndLabel = (EFI_IFR_GUID_LABEL *) HiiCreateGuidOpCode (\r
302	EndOpCodeHandle,\r
303	&gEfiIfrTianoGuid,\r
304	NULL,\r
305	sizeof (EFI_IFR_GUID_LABEL)\r
306	);\r
307	EndLabel->ExtendOpCode = EFI_IFR_EXTEND_OP_LABEL;\r
308	EndLabel->Number = LABEL_END;\r
309	\r
310	//\r
311	// Add each user can be deleted.\r
312	//\r
313	User = NULL;\r
314	Index = 1;\r
315	mUserManager->Current (mUserManager, &CurrentUser);\r
316	while (TRUE) {\r
317	Status = mUserManager->GetNext (mUserManager, &User);\r
318	if (EFI_ERROR (Status)) {\r
319	break;\r
320	}\r
321	\r
322	if (User != CurrentUser) {\r
323	AddUserToForm (\r
324	User,\r
325	(UINT16)(KEY_DEL_USER \| KEY_SELECT_USER \| Index),\r
326	StartOpCodeHandle\r
327	);\r
328	}\r
329	Index++;\r
330	}\r
331	\r
332	HiiUpdateForm (\r
333	mCallbackInfo->HiiHandle, // HII handle\r
a0c56a82	334	&gUserProfileManagerGuid, // Formset GUID\r
0c18794e	335	FORMID_DEL_USER, // Form ID\r
	336	StartOpCodeHandle, // Label for where to insert opcodes\r
	337	EndOpCodeHandle // Replace data\r
	338	);\r
	339	\r
	340	HiiFreeOpCodeHandle (StartOpCodeHandle);\r
	341	HiiFreeOpCodeHandle (EndOpCodeHandle);\r
	342	}\r