[mirror_edk2.git] / NetworkPkg / HttpUtilitiesDxe / HttpUtilitiesProtocol.c

/** @file\r
  Implementation of EFI_HTTP_PROTOCOL protocol interfaces.\r
\r
  Copyright (c) 2015 - 2018, Intel Corporation. All rights reserved.<BR>\r
  (C) Copyright 2016 Hewlett Packard Enterprise Development LP<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "HttpUtilitiesDxe.h"\r
\r
EFI_HTTP_UTILITIES_PROTOCOL mHttpUtilitiesProtocol = {\r
  HttpUtilitiesBuild,\r
  HttpUtilitiesParse\r
};\r
\r
\r
/**\r
  Create HTTP header based on a combination of seed header, fields\r
  to delete, and fields to append.\r
\r
  The Build() function is used to manage the headers portion of an\r
  HTTP message by providing the ability to add, remove, or replace\r
  HTTP headers.\r
\r
  @param[in]  This                Pointer to EFI_HTTP_UTILITIES_PROTOCOL instance.\r
  @param[in]  SeedMessageSize     Size of the initial HTTP header. This can be zero.\r
  @param[in]  SeedMessage         Initial HTTP header to be used as a base for\r
                                  building a new HTTP header. If NULL,\r
                                  SeedMessageSize is ignored.\r
  @param[in]  DeleteCount         Number of null-terminated HTTP header field names\r
                                  in DeleteList.\r
  @param[in]  DeleteList          List of null-terminated HTTP header field names to\r
                                  remove from SeedMessage. Only the field names are\r
                                  in this list because the field values are irrelevant\r
                                  to this operation.\r
  @param[in]  AppendCount         Number of header fields in AppendList.\r
  @param[in]  AppendList          List of HTTP headers to populate NewMessage with.\r
                                  If SeedMessage is not NULL, AppendList will be\r
                                  appended to the existing list from SeedMessage in\r
                                  NewMessage.\r
  @param[out] NewMessageSize      Pointer to number of header fields in NewMessage.\r
  @param[out] NewMessage          Pointer to a new list of HTTP headers based on.\r
\r
  @retval EFI_SUCCESS             Add, remove, and replace operations succeeded.\r
  @retval EFI_OUT_OF_RESOURCES    Could not allocate memory for NewMessage.\r
  @retval EFI_INVALID_PARAMETER   One or more of the following conditions is TRUE:\r
                                  This is NULL.\r
**/\r
EFI_STATUS\r
EFIAPI\r
HttpUtilitiesBuild (\r
  IN     EFI_HTTP_UTILITIES_PROTOCOL *This,\r
  IN     UINTN                       SeedMessageSize,\r
  IN     VOID                        *SeedMessage, OPTIONAL\r
  IN     UINTN                       DeleteCount,\r
  IN     CHAR8                       *DeleteList[], OPTIONAL\r
  IN     UINTN                       AppendCount,\r
  IN     EFI_HTTP_HEADER             *AppendList[], OPTIONAL\r
     OUT UINTN                       *NewMessageSize,\r
     OUT VOID                        **NewMessage\r
  )\r
{\r
  EFI_STATUS                Status;\r
  EFI_HTTP_HEADER           *SeedHeaderFields;\r
  UINTN                     SeedFieldCount;\r
  UINTN                     Index;\r
  EFI_HTTP_HEADER           *TempHeaderFields;\r
  UINTN                     TempFieldCount;\r
  EFI_HTTP_HEADER           *NewHeaderFields;\r
  UINTN                     NewFieldCount;\r
  EFI_HTTP_HEADER           *HttpHeader;\r
  UINTN                     StrLength;\r
  UINT8                     *NewMessagePtr;\r
\r
  SeedHeaderFields = NULL;\r
  SeedFieldCount   = 0;\r
  TempHeaderFields = NULL;\r
  TempFieldCount   = 0;\r
  NewHeaderFields  = NULL;\r
  NewFieldCount    = 0;\r
\r
  HttpHeader       = NULL;\r
  StrLength        = 0;\r
  NewMessagePtr    = NULL;\r
  *NewMessageSize  = 0;\r
  Status           = EFI_SUCCESS;\r
\r
  if (This == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  if (SeedMessage != NULL) {\r
    Status = This->Parse (\r
                     This,\r
                     SeedMessage,\r
                     SeedMessageSize,\r
                     &SeedHeaderFields,\r
                     &SeedFieldCount\r
                     );\r
    if (EFI_ERROR (Status)) {\r
      goto ON_EXIT;\r
    }\r
  }\r
\r
  //\r
  // Handle DeleteList\r
  //\r
  if (SeedFieldCount != 0 && DeleteCount != 0) {\r
    TempHeaderFields = AllocateZeroPool (SeedFieldCount * sizeof(EFI_HTTP_HEADER));\r
    if (TempHeaderFields == NULL) {\r
      Status = EFI_OUT_OF_RESOURCES;\r
      goto ON_EXIT;\r
    }\r
\r
    for (Index = 0, TempFieldCount = 0; Index < SeedFieldCount; Index++) {\r
      //\r
      // Check whether each SeedHeaderFields member is in DeleteList\r
      //\r
      if (HttpIsValidHttpHeader( DeleteList, DeleteCount, SeedHeaderFields[Index].FieldName)) {\r
        Status = HttpSetFieldNameAndValue (\r
                   &TempHeaderFields[TempFieldCount],\r
                   SeedHeaderFields[Index].FieldName,\r
                   SeedHeaderFields[Index].FieldValue\r
                   );\r
        if (EFI_ERROR (Status)) {\r
          goto ON_EXIT;\r
        }\r
        TempFieldCount++;\r
      }\r
    }\r
  } else {\r
    TempHeaderFields = SeedHeaderFields;\r
    TempFieldCount = SeedFieldCount;\r
  }\r
\r
  //\r
  // Handle AppendList\r
  //\r
  NewHeaderFields = AllocateZeroPool ((TempFieldCount + AppendCount) * sizeof (EFI_HTTP_HEADER));\r
  if (NewHeaderFields == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  for (Index = 0; Index < TempFieldCount; Index++) {\r
    Status = HttpSetFieldNameAndValue (\r
               &NewHeaderFields[Index],\r
               TempHeaderFields[Index].FieldName,\r
               TempHeaderFields[Index].FieldValue\r
               );\r
    if (EFI_ERROR (Status)) {\r
      goto ON_EXIT;\r
    }\r
  }\r
\r
  NewFieldCount = TempFieldCount;\r
\r
  for (Index = 0; Index < AppendCount; Index++) {\r
    HttpHeader = HttpFindHeader (NewFieldCount, NewHeaderFields, AppendList[Index]->FieldName);\r
    if (HttpHeader != NULL) {\r
      Status = HttpSetFieldNameAndValue (\r
                 HttpHeader,\r
                 AppendList[Index]->FieldName,\r
                 AppendList[Index]->FieldValue\r
                 );\r
      if (EFI_ERROR (Status)) {\r
        goto ON_EXIT;\r
      }\r
    } else {\r
      Status = HttpSetFieldNameAndValue (\r
                 &NewHeaderFields[NewFieldCount],\r
                 AppendList[Index]->FieldName,\r
                 AppendList[Index]->FieldValue\r
                 );\r
      if (EFI_ERROR (Status)) {\r
        goto ON_EXIT;\r
      }\r
      NewFieldCount++;\r
    }\r
  }\r
\r
  //\r
  // Calculate NewMessageSize, then build NewMessage\r
  //\r
  for (Index = 0; Index < NewFieldCount; Index++) {\r
    HttpHeader = &NewHeaderFields[Index];\r
\r
    StrLength = AsciiStrLen (HttpHeader->FieldName);\r
    *NewMessageSize += StrLength;\r
\r
    StrLength = sizeof(": ") - 1;\r
    *NewMessageSize += StrLength;\r
\r
    StrLength = AsciiStrLen (HttpHeader->FieldValue);\r
    *NewMessageSize += StrLength;\r
\r
    StrLength = sizeof("\r\n") - 1;\r
    *NewMessageSize += StrLength;\r
  }\r
  StrLength = sizeof("\r\n") - 1;\r
  *NewMessageSize += StrLength;\r
\r
  *NewMessage = AllocateZeroPool (*NewMessageSize);\r
  if (*NewMessage == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  NewMessagePtr = (UINT8 *)(*NewMessage);\r
\r
  for (Index = 0; Index < NewFieldCount; Index++) {\r
    HttpHeader = &NewHeaderFields[Index];\r
\r
    StrLength = AsciiStrLen (HttpHeader->FieldName);\r
    CopyMem (NewMessagePtr, HttpHeader->FieldName, StrLength);\r
    NewMessagePtr += StrLength;\r
\r
    StrLength = sizeof(": ") - 1;\r
    CopyMem (NewMessagePtr, ": ", StrLength);\r
    NewMessagePtr += StrLength;\r
\r
    StrLength = AsciiStrLen (HttpHeader->FieldValue);\r
    CopyMem (NewMessagePtr, HttpHeader->FieldValue, StrLength);\r
    NewMessagePtr += StrLength;\r
\r
    StrLength = sizeof("\r\n") - 1;\r
    CopyMem (NewMessagePtr, "\r\n", StrLength);\r
    NewMessagePtr += StrLength;\r
  }\r
  StrLength = sizeof("\r\n") - 1;\r
  CopyMem (NewMessagePtr, "\r\n", StrLength);\r
  NewMessagePtr += StrLength;\r
\r
  ASSERT (*NewMessageSize == (UINTN)NewMessagePtr - (UINTN)(*NewMessage));\r
\r
  //\r
  // Free allocated buffer\r
  //\r
ON_EXIT:\r
  if (SeedHeaderFields != NULL) {\r
    HttpFreeHeaderFields(SeedHeaderFields, SeedFieldCount);\r
  }\r
\r
  if (TempHeaderFields != NULL) {\r
    HttpFreeHeaderFields(TempHeaderFields, TempFieldCount);\r
  }\r
\r
  if (NewHeaderFields != NULL) {\r
    HttpFreeHeaderFields(NewHeaderFields, NewFieldCount);\r
  }\r
\r
  return Status;\r
}\r
\r
\r
/**\r
  Parses HTTP header and produces an array of key/value pairs.\r
\r
  The Parse() function is used to transform data stored in HttpHeader\r
  into a list of fields paired with their corresponding values.\r
\r
  @param[in]  This                Pointer to EFI_HTTP_UTILITIES_PROTOCOL instance.\r
  @param[in]  HttpMessage         Contains raw unformatted HTTP header string.\r
  @param[in]  HttpMessageSize     Size of HTTP header.\r
  @param[out] HeaderFields        Array of key/value header pairs.\r
  @param[out] FieldCount          Number of headers in HeaderFields.\r
\r
  @retval EFI_SUCCESS             Allocation succeeded.\r
  @retval EFI_NOT_STARTED         This EFI HTTP Protocol instance has not been\r
                                  initialized.\r
  @retval EFI_INVALID_PARAMETER   One or more of the following conditions is TRUE:\r
                                  This is NULL.\r
                                  HttpMessage is NULL.\r
                                  HeaderFields is NULL.\r
                                  FieldCount is NULL.\r
**/\r
EFI_STATUS\r
EFIAPI\r
HttpUtilitiesParse (\r
  IN  EFI_HTTP_UTILITIES_PROTOCOL  *This,\r
  IN  CHAR8                        *HttpMessage,\r
  IN  UINTN                        HttpMessageSize,\r
  OUT EFI_HTTP_HEADER              **HeaderFields,\r
  OUT UINTN                        *FieldCount\r
  )\r
{\r
  EFI_STATUS                Status;\r
  CHAR8                     *TempHttpMessage;\r
  CHAR8                     *Token;\r
  CHAR8                     *NextToken;\r
  CHAR8                     *FieldName;\r
  CHAR8                     *FieldValue;\r
  UINTN                     Index;\r
  UINTN                     HttpBufferSize;\r
\r
  Status          = EFI_SUCCESS;\r
  TempHttpMessage = NULL;\r
  Token           = NULL;\r
  NextToken       = NULL;\r
  FieldName       = NULL;\r
  FieldValue      = NULL;\r
  Index           = 0;\r
\r
  if (This == NULL || HttpMessage == NULL || HeaderFields == NULL || FieldCount == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Append the http response string along with a Null-terminator.\r
  //\r
  HttpBufferSize = HttpMessageSize + 1;\r
  TempHttpMessage = AllocatePool (HttpBufferSize);\r
  if (TempHttpMessage == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  CopyMem (TempHttpMessage, HttpMessage, HttpMessageSize);\r
  *(TempHttpMessage + HttpMessageSize) = '\0';\r
\r
  //\r
  // Get header number\r
  //\r
  *FieldCount = 0;\r
  Token = TempHttpMessage;\r
  while (TRUE) {\r
    FieldName     = NULL;\r
    FieldValue    = NULL;\r
    NextToken = HttpGetFieldNameAndValue (Token, &FieldName, &FieldValue);\r
    Token     = NextToken;\r
    if (FieldName == NULL || FieldValue == NULL) {\r
      break;\r
    }\r
\r
    (*FieldCount)++;\r
  }\r
\r
  if (*FieldCount == 0) {\r
    Status =  EFI_INVALID_PARAMETER;\r
    goto ON_EXIT;\r
  }\r
\r
  //\r
  // Allocate buffer for header\r
  //\r
  *HeaderFields = AllocateZeroPool ((*FieldCount) * sizeof(EFI_HTTP_HEADER));\r
  if (*HeaderFields == NULL) {\r
    *FieldCount = 0;\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto ON_EXIT;\r
  }\r
\r
  CopyMem (TempHttpMessage, HttpMessage, HttpMessageSize);\r
\r
  //\r
  // Set Field and Value to each header\r
  //\r
  Token = TempHttpMessage;\r
  while (Index < *FieldCount) {\r
    FieldName     = NULL;\r
    FieldValue    = NULL;\r
    NextToken = HttpGetFieldNameAndValue (Token, &FieldName, &FieldValue);\r
    Token     = NextToken;\r
    if (FieldName == NULL || FieldValue == NULL) {\r
      break;\r
    }\r
\r
    Status = HttpSetFieldNameAndValue (&(*HeaderFields)[Index], FieldName, FieldValue);\r
    if (EFI_ERROR (Status)) {\r
      *FieldCount = 0;\r
      HttpFreeHeaderFields (*HeaderFields, Index);\r
      goto ON_EXIT;\r
    }\r
\r
    Index++;\r
  }\r
\r
  //\r
  // Free allocated buffer\r
  //\r
ON_EXIT:\r
  if (TempHttpMessage != NULL) {\r
    FreePool (TempHttpMessage);\r
  }\r
\r
  return Status;\r
}\r
Commit	Line	Data
	1	/** @file\r
	2	Implementation of EFI_HTTP_PROTOCOL protocol interfaces.\r
	3	\r
	4	Copyright (c) 2015 - 2018, Intel Corporation. All rights reserved.<BR>\r
	5	(C) Copyright 2016 Hewlett Packard Enterprise Development LP<BR>\r
	6	SPDX-License-Identifier: BSD-2-Clause-Patent\r
	7	\r
	8	**/\r
	9	\r
	10	#include "HttpUtilitiesDxe.h"\r
	11	\r
	12	EFI_HTTP_UTILITIES_PROTOCOL mHttpUtilitiesProtocol = {\r
	13	HttpUtilitiesBuild,\r
	14	HttpUtilitiesParse\r
	15	};\r
	16	\r
	17	\r
	18	/**\r
	19	Create HTTP header based on a combination of seed header, fields\r
	20	to delete, and fields to append.\r
	21	\r
	22	The Build() function is used to manage the headers portion of an\r
	23	HTTP message by providing the ability to add, remove, or replace\r
	24	HTTP headers.\r
	25	\r
	26	@param[in] This Pointer to EFI_HTTP_UTILITIES_PROTOCOL instance.\r
	27	@param[in] SeedMessageSize Size of the initial HTTP header. This can be zero.\r
	28	@param[in] SeedMessage Initial HTTP header to be used as a base for\r
	29	building a new HTTP header. If NULL,\r
	30	SeedMessageSize is ignored.\r
	31	@param[in] DeleteCount Number of null-terminated HTTP header field names\r
	32	in DeleteList.\r
	33	@param[in] DeleteList List of null-terminated HTTP header field names to\r
	34	remove from SeedMessage. Only the field names are\r
	35	in this list because the field values are irrelevant\r
	36	to this operation.\r
	37	@param[in] AppendCount Number of header fields in AppendList.\r
	38	@param[in] AppendList List of HTTP headers to populate NewMessage with.\r
	39	If SeedMessage is not NULL, AppendList will be\r
	40	appended to the existing list from SeedMessage in\r
	41	NewMessage.\r
	42	@param[out] NewMessageSize Pointer to number of header fields in NewMessage.\r
	43	@param[out] NewMessage Pointer to a new list of HTTP headers based on.\r
	44	\r
	45	@retval EFI_SUCCESS Add, remove, and replace operations succeeded.\r
	46	@retval EFI_OUT_OF_RESOURCES Could not allocate memory for NewMessage.\r
	47	@retval EFI_INVALID_PARAMETER One or more of the following conditions is TRUE:\r
	48	This is NULL.\r
	49	**/\r
	50	EFI_STATUS\r
	51	EFIAPI\r
	52	HttpUtilitiesBuild (\r
	53	IN EFI_HTTP_UTILITIES_PROTOCOL *This,\r
	54	IN UINTN SeedMessageSize,\r
	55	IN VOID *SeedMessage, OPTIONAL\r
	56	IN UINTN DeleteCount,\r
	57	IN CHAR8 *DeleteList[], OPTIONAL\r
	58	IN UINTN AppendCount,\r
	59	IN EFI_HTTP_HEADER *AppendList[], OPTIONAL\r
	60	OUT UINTN *NewMessageSize,\r
	61	OUT VOID **NewMessage\r
	62	)\r
	63	{\r
	64	EFI_STATUS Status;\r
	65	EFI_HTTP_HEADER *SeedHeaderFields;\r
	66	UINTN SeedFieldCount;\r
	67	UINTN Index;\r
	68	EFI_HTTP_HEADER *TempHeaderFields;\r
	69	UINTN TempFieldCount;\r
	70	EFI_HTTP_HEADER *NewHeaderFields;\r
	71	UINTN NewFieldCount;\r
	72	EFI_HTTP_HEADER *HttpHeader;\r
	73	UINTN StrLength;\r
	74	UINT8 *NewMessagePtr;\r
	75	\r
	76	SeedHeaderFields = NULL;\r
	77	SeedFieldCount = 0;\r
	78	TempHeaderFields = NULL;\r
	79	TempFieldCount = 0;\r
	80	NewHeaderFields = NULL;\r
	81	NewFieldCount = 0;\r
	82	\r
	83	HttpHeader = NULL;\r
	84	StrLength = 0;\r
	85	NewMessagePtr = NULL;\r
	86	*NewMessageSize = 0;\r
	87	Status = EFI_SUCCESS;\r
	88	\r
	89	if (This == NULL) {\r
	90	return EFI_INVALID_PARAMETER;\r
	91	}\r
	92	\r
	93	if (SeedMessage != NULL) {\r
	94	Status = This->Parse (\r
	95	This,\r
	96	SeedMessage,\r
	97	SeedMessageSize,\r
	98	&SeedHeaderFields,\r
	99	&SeedFieldCount\r
	100	);\r
	101	if (EFI_ERROR (Status)) {\r
	102	goto ON_EXIT;\r
	103	}\r
	104	}\r
	105	\r
	106	//\r
	107	// Handle DeleteList\r
	108	//\r
	109	if (SeedFieldCount != 0 && DeleteCount != 0) {\r
	110	TempHeaderFields = AllocateZeroPool (SeedFieldCount * sizeof(EFI_HTTP_HEADER));\r
	111	if (TempHeaderFields == NULL) {\r
	112	Status = EFI_OUT_OF_RESOURCES;\r
	113	goto ON_EXIT;\r
	114	}\r
	115	\r
	116	for (Index = 0, TempFieldCount = 0; Index < SeedFieldCount; Index++) {\r
	117	//\r
	118	// Check whether each SeedHeaderFields member is in DeleteList\r
	119	//\r
	120	if (HttpIsValidHttpHeader( DeleteList, DeleteCount, SeedHeaderFields[Index].FieldName)) {\r
	121	Status = HttpSetFieldNameAndValue (\r
	122	&TempHeaderFields[TempFieldCount],\r
	123	SeedHeaderFields[Index].FieldName,\r
	124	SeedHeaderFields[Index].FieldValue\r
	125	);\r
	126	if (EFI_ERROR (Status)) {\r
	127	goto ON_EXIT;\r
	128	}\r
	129	TempFieldCount++;\r
	130	}\r
	131	}\r
	132	} else {\r
	133	TempHeaderFields = SeedHeaderFields;\r
	134	TempFieldCount = SeedFieldCount;\r
	135	}\r
	136	\r
	137	//\r
	138	// Handle AppendList\r
	139	//\r
	140	NewHeaderFields = AllocateZeroPool ((TempFieldCount + AppendCount) * sizeof (EFI_HTTP_HEADER));\r
	141	if (NewHeaderFields == NULL) {\r
	142	Status = EFI_OUT_OF_RESOURCES;\r
	143	goto ON_EXIT;\r
	144	}\r
	145	\r
	146	for (Index = 0; Index < TempFieldCount; Index++) {\r
	147	Status = HttpSetFieldNameAndValue (\r
	148	&NewHeaderFields[Index],\r
	149	TempHeaderFields[Index].FieldName,\r
	150	TempHeaderFields[Index].FieldValue\r
	151	);\r
	152	if (EFI_ERROR (Status)) {\r
	153	goto ON_EXIT;\r
	154	}\r
	155	}\r
	156	\r
	157	NewFieldCount = TempFieldCount;\r
	158	\r
	159	for (Index = 0; Index < AppendCount; Index++) {\r
	160	HttpHeader = HttpFindHeader (NewFieldCount, NewHeaderFields, AppendList[Index]->FieldName);\r
	161	if (HttpHeader != NULL) {\r
	162	Status = HttpSetFieldNameAndValue (\r
	163	HttpHeader,\r
	164	AppendList[Index]->FieldName,\r
	165	AppendList[Index]->FieldValue\r
	166	);\r
	167	if (EFI_ERROR (Status)) {\r
	168	goto ON_EXIT;\r
	169	}\r
	170	} else {\r
	171	Status = HttpSetFieldNameAndValue (\r
	172	&NewHeaderFields[NewFieldCount],\r
	173	AppendList[Index]->FieldName,\r
	174	AppendList[Index]->FieldValue\r
	175	);\r
	176	if (EFI_ERROR (Status)) {\r
	177	goto ON_EXIT;\r
	178	}\r
	179	NewFieldCount++;\r
	180	}\r
	181	}\r
	182	\r
	183	//\r
	184	// Calculate NewMessageSize, then build NewMessage\r
	185	//\r
	186	for (Index = 0; Index < NewFieldCount; Index++) {\r
	187	HttpHeader = &NewHeaderFields[Index];\r
	188	\r
	189	StrLength = AsciiStrLen (HttpHeader->FieldName);\r
	190	*NewMessageSize += StrLength;\r
	191	\r
	192	StrLength = sizeof(": ") - 1;\r
	193	*NewMessageSize += StrLength;\r
	194	\r
	195	StrLength = AsciiStrLen (HttpHeader->FieldValue);\r
	196	*NewMessageSize += StrLength;\r
	197	\r
	198	StrLength = sizeof("\r\n") - 1;\r
	199	*NewMessageSize += StrLength;\r
	200	}\r
	201	StrLength = sizeof("\r\n") - 1;\r
	202	*NewMessageSize += StrLength;\r
	203	\r
	204	NewMessage = AllocateZeroPool (NewMessageSize);\r
	205	if (*NewMessage == NULL) {\r
	206	Status = EFI_OUT_OF_RESOURCES;\r
	207	goto ON_EXIT;\r
	208	}\r
	209	\r
	210	NewMessagePtr = (UINT8 )(NewMessage);\r
	211	\r
	212	for (Index = 0; Index < NewFieldCount; Index++) {\r
	213	HttpHeader = &NewHeaderFields[Index];\r
	214	\r
	215	StrLength = AsciiStrLen (HttpHeader->FieldName);\r
	216	CopyMem (NewMessagePtr, HttpHeader->FieldName, StrLength);\r
	217	NewMessagePtr += StrLength;\r
	218	\r
	219	StrLength = sizeof(": ") - 1;\r
	220	CopyMem (NewMessagePtr, ": ", StrLength);\r
	221	NewMessagePtr += StrLength;\r
	222	\r
	223	StrLength = AsciiStrLen (HttpHeader->FieldValue);\r
	224	CopyMem (NewMessagePtr, HttpHeader->FieldValue, StrLength);\r
	225	NewMessagePtr += StrLength;\r
	226	\r
	227	StrLength = sizeof("\r\n") - 1;\r
	228	CopyMem (NewMessagePtr, "\r\n", StrLength);\r
	229	NewMessagePtr += StrLength;\r
	230	}\r
	231	StrLength = sizeof("\r\n") - 1;\r
	232	CopyMem (NewMessagePtr, "\r\n", StrLength);\r
	233	NewMessagePtr += StrLength;\r
	234	\r
	235	ASSERT (NewMessageSize == (UINTN)NewMessagePtr - (UINTN)(NewMessage));\r
	236	\r
	237	//\r
	238	// Free allocated buffer\r
	239	//\r
	240	ON_EXIT:\r
	241	if (SeedHeaderFields != NULL) {\r
	242	HttpFreeHeaderFields(SeedHeaderFields, SeedFieldCount);\r
	243	}\r
	244	\r
	245	if (TempHeaderFields != NULL) {\r
	246	HttpFreeHeaderFields(TempHeaderFields, TempFieldCount);\r
	247	}\r
	248	\r
	249	if (NewHeaderFields != NULL) {\r
	250	HttpFreeHeaderFields(NewHeaderFields, NewFieldCount);\r
	251	}\r
	252	\r
	253	return Status;\r
	254	}\r
	255	\r
	256	\r
	257	/**\r
	258	Parses HTTP header and produces an array of key/value pairs.\r
	259	\r
	260	The Parse() function is used to transform data stored in HttpHeader\r
	261	into a list of fields paired with their corresponding values.\r
	262	\r
	263	@param[in] This Pointer to EFI_HTTP_UTILITIES_PROTOCOL instance.\r
	264	@param[in] HttpMessage Contains raw unformatted HTTP header string.\r
	265	@param[in] HttpMessageSize Size of HTTP header.\r
	266	@param[out] HeaderFields Array of key/value header pairs.\r
	267	@param[out] FieldCount Number of headers in HeaderFields.\r
	268	\r
	269	@retval EFI_SUCCESS Allocation succeeded.\r
	270	@retval EFI_NOT_STARTED This EFI HTTP Protocol instance has not been\r
	271	initialized.\r
	272	@retval EFI_INVALID_PARAMETER One or more of the following conditions is TRUE:\r
	273	This is NULL.\r
	274	HttpMessage is NULL.\r
	275	HeaderFields is NULL.\r
	276	FieldCount is NULL.\r
	277	**/\r
	278	EFI_STATUS\r
	279	EFIAPI\r
	280	HttpUtilitiesParse (\r
	281	IN EFI_HTTP_UTILITIES_PROTOCOL *This,\r
	282	IN CHAR8 *HttpMessage,\r
	283	IN UINTN HttpMessageSize,\r
	284	OUT EFI_HTTP_HEADER **HeaderFields,\r
	285	OUT UINTN *FieldCount\r
	286	)\r
	287	{\r
	288	EFI_STATUS Status;\r
	289	CHAR8 *TempHttpMessage;\r
	290	CHAR8 *Token;\r
	291	CHAR8 *NextToken;\r
	292	CHAR8 *FieldName;\r
	293	CHAR8 *FieldValue;\r
	294	UINTN Index;\r
	295	UINTN HttpBufferSize;\r
	296	\r
	297	Status = EFI_SUCCESS;\r
	298	TempHttpMessage = NULL;\r
	299	Token = NULL;\r
	300	NextToken = NULL;\r
	301	FieldName = NULL;\r
	302	FieldValue = NULL;\r
	303	Index = 0;\r
	304	\r
	305	if (This == NULL \|\| HttpMessage == NULL \|\| HeaderFields == NULL \|\| FieldCount == NULL) {\r
	306	return EFI_INVALID_PARAMETER;\r
	307	}\r
	308	\r
	309	//\r
	310	// Append the http response string along with a Null-terminator.\r
	311	//\r
	312	HttpBufferSize = HttpMessageSize + 1;\r
	313	TempHttpMessage = AllocatePool (HttpBufferSize);\r
	314	if (TempHttpMessage == NULL) {\r
	315	return EFI_OUT_OF_RESOURCES;\r
	316	}\r
	317	\r
	318	CopyMem (TempHttpMessage, HttpMessage, HttpMessageSize);\r
	319	*(TempHttpMessage + HttpMessageSize) = '\0';\r
	320	\r
	321	//\r
	322	// Get header number\r
	323	//\r
	324	*FieldCount = 0;\r
	325	Token = TempHttpMessage;\r
	326	while (TRUE) {\r
	327	FieldName = NULL;\r
	328	FieldValue = NULL;\r
	329	NextToken = HttpGetFieldNameAndValue (Token, &FieldName, &FieldValue);\r
	330	Token = NextToken;\r
	331	if (FieldName == NULL \|\| FieldValue == NULL) {\r
	332	break;\r
	333	}\r
	334	\r
	335	(*FieldCount)++;\r
	336	}\r
	337	\r
	338	if (*FieldCount == 0) {\r
	339	Status = EFI_INVALID_PARAMETER;\r
	340	goto ON_EXIT;\r
	341	}\r
	342	\r
	343	//\r
	344	// Allocate buffer for header\r
	345	//\r
	346	HeaderFields = AllocateZeroPool ((FieldCount) * sizeof(EFI_HTTP_HEADER));\r
	347	if (*HeaderFields == NULL) {\r
	348	*FieldCount = 0;\r
	349	Status = EFI_OUT_OF_RESOURCES;\r
	350	goto ON_EXIT;\r
	351	}\r
	352	\r
	353	CopyMem (TempHttpMessage, HttpMessage, HttpMessageSize);\r
	354	\r
	355	//\r
	356	// Set Field and Value to each header\r
	357	//\r
	358	Token = TempHttpMessage;\r
	359	while (Index < *FieldCount) {\r
	360	FieldName = NULL;\r
	361	FieldValue = NULL;\r
	362	NextToken = HttpGetFieldNameAndValue (Token, &FieldName, &FieldValue);\r
	363	Token = NextToken;\r
	364	if (FieldName == NULL \|\| FieldValue == NULL) {\r
	365	break;\r
	366	}\r
	367	\r
	368	Status = HttpSetFieldNameAndValue (&(*HeaderFields)[Index], FieldName, FieldValue);\r
	369	if (EFI_ERROR (Status)) {\r
	370	*FieldCount = 0;\r
	371	HttpFreeHeaderFields (*HeaderFields, Index);\r
	372	goto ON_EXIT;\r
	373	}\r
	374	\r
	375	Index++;\r
	376	}\r
	377	\r
	378	//\r
	379	// Free allocated buffer\r
	380	//\r
	381	ON_EXIT:\r
	382	if (TempHttpMessage != NULL) {\r
	383	FreePool (TempHttpMessage);\r
	384	}\r
	385	\r
	386	return Status;\r
	387	}\r