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