[mirror_edk2.git] / MdeModulePkg / Universal / Network / Ip4ConfigDxe / NicIp4Variable.c

/** @file\r
  Routines used to operate the Ip4 configure variable.\r
\r
Copyright (c) 2006 - 2009, 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<BR>\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 "NicIp4Variable.h"\r
\r
\r
/**\r
  Check whether the configure parameter is valid.\r
\r
  @param  NicConfig    The configure parameter to check\r
\r
  @return TRUE if the parameter is valid for the interface, otherwise FALSE.\r
\r
**/\r
BOOLEAN\r
Ip4ConfigIsValid (\r
  IN NIC_IP4_CONFIG_INFO    *NicConfig\r
  )\r
{\r
  EFI_IP4_IPCONFIG_DATA     *IpConfig;\r
  IP4_ADDR                  Station;\r
  IP4_ADDR                  Netmask;\r
  IP4_ADDR                  Gateway;\r
  UINT32                    Index;\r
\r
  IpConfig = &NicConfig->Ip4Info;\r
\r
  if (NicConfig->Source == IP4_CONFIG_SOURCE_STATIC) {\r
    //\r
    // Validate that the addresses are unicast and mask\r
    // is properly formated\r
    //\r
    Station = EFI_NTOHL (IpConfig->StationAddress);\r
    Netmask = EFI_NTOHL (IpConfig->SubnetMask);\r
\r
    if ((Netmask == 0) || !IP4_IS_VALID_NETMASK (Netmask) ||\r
        (Station == 0) || !NetIp4IsUnicast (Station, Netmask)) {\r
      return FALSE;\r
    }\r
\r
    //\r
    // Validate that the next hops are on the connected network\r
    // or that is a direct route (Gateway == 0).\r
    //\r
    for (Index = 0; Index < IpConfig->RouteTableSize; Index++) {\r
      Gateway = EFI_NTOHL (IpConfig->RouteTable[Index].GatewayAddress);\r
\r
      if ((Gateway != 0) && (!IP4_NET_EQUAL (Station, Gateway, Netmask) ||\r
          !NetIp4IsUnicast (Gateway, Netmask))) {\r
        return FALSE;\r
      }\r
    }\r
\r
    return TRUE;\r
  }\r
\r
  //\r
  // return false if it is an unkown configure source. Valid\r
  // sources are static and dhcp.\r
  //\r
  return (BOOLEAN) (NicConfig->Source == IP4_CONFIG_SOURCE_DHCP);\r
}\r
\r
\r
\r
/**\r
  Read the ip4 configure variable from the EFI variable.\r
\r
  None\r
\r
  @return The IP4 configure read if it is there and is valid, otherwise NULL\r
\r
**/\r
IP4_CONFIG_VARIABLE *\r
Ip4ConfigReadVariable (\r
  VOID\r
  )\r
{\r
  IP4_CONFIG_VARIABLE       *Variable;\r
  EFI_STATUS                Status;\r
  UINTN                     Size;\r
  UINT16                    CheckSum;\r
\r
  //\r
  // Get the size of variable, then allocate a buffer to read the variable.\r
  //\r
  Size     = 0;\r
  Variable = NULL;\r
  Status   = gRT->GetVariable (\r
                    EFI_NIC_IP4_CONFIG_VARIABLE,\r
                    &gEfiNicIp4ConfigVariableGuid,\r
                    NULL,\r
                    &Size,\r
                    NULL\r
                    );\r
\r
  if (Status != EFI_BUFFER_TOO_SMALL) {\r
    return NULL;\r
  }\r
\r
  if (Size < sizeof (IP4_CONFIG_VARIABLE)) {\r
    goto REMOVE_VARIABLE;\r
  }\r
\r
  Variable = AllocatePool (Size);\r
\r
  if (Variable == NULL) {\r
    return NULL;\r
  }\r
\r
  Status = gRT->GetVariable (\r
                  EFI_NIC_IP4_CONFIG_VARIABLE,\r
                  &gEfiNicIp4ConfigVariableGuid,\r
                  NULL,\r
                  &Size,\r
                  Variable\r
                  );\r
\r
  if (EFI_ERROR (Status)) {\r
    goto ON_ERROR;\r
  }\r
\r
  //\r
  // Verify the checksum, variable size and count\r
  //\r
  CheckSum = (UINT16) (~NetblockChecksum ((UINT8 *) Variable, (UINT32)Size));\r
\r
  if ((CheckSum != 0) || (Size != Variable->Len)) {\r
    goto REMOVE_VARIABLE;\r
  }\r
\r
  if ((Variable->Count < 1) || (Variable->Count > MAX_IP4_CONFIG_IN_VARIABLE)) {\r
    goto REMOVE_VARIABLE;\r
  }\r
\r
  return Variable;\r
\r
REMOVE_VARIABLE:\r
  Ip4ConfigWriteVariable (NULL);\r
\r
ON_ERROR:\r
  if (Variable != NULL) {\r
    FreePool (Variable);\r
  }\r
\r
  return NULL;\r
}\r
\r
\r
/**\r
  Write the IP4 configure variable to the NVRAM. If Config\r
  is NULL, remove the variable.\r
\r
  @param  Config       The IP4 configure data to write\r
\r
  @retval EFI_SUCCESS  The variable is written to the NVRam\r
  @retval Others       Failed to write the variable.\r
\r
**/\r
EFI_STATUS\r
Ip4ConfigWriteVariable (\r
  IN IP4_CONFIG_VARIABLE    *Config        OPTIONAL\r
  )\r
{\r
  EFI_STATUS                Status;\r
\r
  Status = gRT->SetVariable (\r
                  EFI_NIC_IP4_CONFIG_VARIABLE,\r
                  &gEfiNicIp4ConfigVariableGuid,\r
                  IP4_CONFIG_VARIABLE_ATTRIBUTES,\r
                  (Config == NULL) ? 0 : Config->Len,\r
                  Config\r
                  );\r
\r
  return Status;\r
}\r
\r
\r
/**\r
  Locate the IP4 configure parameters from the variable.If a\r
  configuration is found, copy it to a newly allocated block\r
  of memory to avoid the alignment problem. Caller should\r
  release the memory after use.\r
\r
  @param  Variable     The IP4 configure variable to search in\r
  @param  NicAddr      The interface address to check\r
\r
  @return The point to the NIC's IP4 configure info if it is found\r
          in the IP4 variable, otherwise NULL.\r
\r
**/\r
NIC_IP4_CONFIG_INFO *\r
Ip4ConfigFindNicVariable (\r
  IN IP4_CONFIG_VARIABLE    *Variable,\r
  IN NIC_ADDR               *NicAddr\r
  )\r
{\r
  NIC_IP4_CONFIG_INFO       Temp;\r
  NIC_IP4_CONFIG_INFO       *Config;\r
  UINT32                    Index;\r
  UINT8                     *Cur;\r
  UINT32                    Len;\r
\r
  Cur = (UINT8*)&Variable->ConfigInfo;\r
\r
  for (Index = 0; Index < Variable->Count; Index++) {\r
    //\r
    // Copy the data to Temp to avoid the alignment problems\r
    //\r
    CopyMem (&Temp, Cur, sizeof (NIC_IP4_CONFIG_INFO));\r
    Len = SIZEOF_NIC_IP4_CONFIG_INFO (&Temp);\r
\r
    //\r
    // Found the matching configuration parameters, allocate\r
    // a block of memory then copy it out.\r
    //\r
    if (NIC_ADDR_EQUAL (&Temp.NicAddr, NicAddr)) {\r
      Config = AllocatePool (Len);\r
\r
      if (Config == NULL) {\r
        return NULL;\r
      }\r
\r
      CopyMem (Config, Cur, Len);\r
      Ip4ConfigFixRouteTablePointer (&Config->Ip4Info);\r
      return Config;\r
    }\r
\r
    Cur += Len;\r
  }\r
\r
  return NULL;\r
}\r
\r
\r
/**\r
  Modify the configuration parameter for the NIC in the variable.\r
  If Config is NULL, old configuration will be remove from the new\r
  variable. Otherwise, append it or replace the old one.\r
\r
  @param  Variable     The IP4 variable to change\r
  @param  NicAddr      The interface to search\r
  @param  Config       The new configuration parameter (NULL to remove the old)\r
\r
  @return The new IP4_CONFIG_VARIABLE variable if the new variable has at\r
          least one NIC configure and no EFI_OUT_OF_RESOURCES failure.\r
          Return NULL either because failed to locate memory for new variable\r
          or the only NIC configure is removed from the Variable.\r
\r
**/\r
IP4_CONFIG_VARIABLE *\r
Ip4ConfigModifyVariable (\r
  IN IP4_CONFIG_VARIABLE    *Variable     OPTIONAL,\r
  IN NIC_ADDR               *NicAddr,\r
  IN NIC_IP4_CONFIG_INFO    *Config       OPTIONAL\r
  )\r
{\r
  NIC_IP4_CONFIG_INFO       Temp;\r
  NIC_IP4_CONFIG_INFO       *Old;\r
  IP4_CONFIG_VARIABLE       *NewVar;\r
  UINT32                    Len;\r
  UINT32                    TotalLen;\r
  UINT32                    Count;\r
  UINT8                     *Next;\r
  UINT8                     *Cur;\r
  UINT32                    Index;\r
\r
  ASSERT ((Variable != NULL) || (Config != NULL));\r
\r
  //\r
  // Compute the total length\r
  //\r
  if (Variable != NULL) {\r
    //\r
    // Variable != NULL, then Config can be NULL or not.  and so is\r
    // the Old. If old configure exists, it is removed from the\r
    // Variable. New configure is append to the variable.\r
    //\r
    //\r
    Count     = Variable->Count;\r
    Cur       = (UINT8 *)&Variable->ConfigInfo;\r
    TotalLen  = Variable->Len;\r
\r
    Old       = Ip4ConfigFindNicVariable (Variable, NicAddr);\r
\r
    if (Old != NULL) {\r
      TotalLen -= SIZEOF_NIC_IP4_CONFIG_INFO (Old);\r
      FreePool (Old);\r
    }\r
\r
    if (Config != NULL) {\r
      TotalLen += SIZEOF_NIC_IP4_CONFIG_INFO (Config);\r
    }\r
\r
    //\r
    // Return NULL if the only NIC_IP4_CONFIG_INFO is being removed.\r
    //\r
    if (TotalLen < sizeof (IP4_CONFIG_VARIABLE)) {\r
      return NULL;\r
    }\r
\r
  } else {\r
    //\r
    // Variable == NULL and Config != NULL, Create a new variable with\r
    // this NIC configure.\r
    //\r
    Count     = 0;\r
    Cur       = NULL;\r
    TotalLen  = sizeof (IP4_CONFIG_VARIABLE) - sizeof (NIC_IP4_CONFIG_INFO)\r
                + SIZEOF_NIC_IP4_CONFIG_INFO (Config);\r
  }\r
\r
  ASSERT (TotalLen >= sizeof (IP4_CONFIG_VARIABLE));\r
\r
  NewVar = AllocateZeroPool (TotalLen);\r
\r
  if (NewVar == NULL) {\r
    return NULL;\r
  }\r
\r
  NewVar->Len = TotalLen;\r
\r
  //\r
  // Copy the other configure parameters from the old variable\r
  //\r
  Next = (UINT8 *)&NewVar->ConfigInfo;\r
\r
  for (Index = 0; Index < Count; Index++) {\r
    CopyMem (&Temp, Cur, sizeof (NIC_IP4_CONFIG_INFO));\r
    Len = SIZEOF_NIC_IP4_CONFIG_INFO (&Temp);\r
\r
    if (!NIC_ADDR_EQUAL (&Temp.NicAddr, NicAddr)) {\r
      CopyMem (Next, Cur, Len);\r
      Next += Len;\r
      NewVar->Count++;\r
    }\r
\r
    Cur += Len;\r
  }\r
\r
  //\r
  // Append the new configure if it isn't NULL.\r
  //\r
  Len = 0;\r
\r
  if (Config != NULL) {\r
    Len = SIZEOF_NIC_IP4_CONFIG_INFO (Config);\r
\r
    CopyMem (Next, Config, Len);\r
    NewVar->Count++;\r
  }\r
\r
  ASSERT (Next + Len == (UINT8 *) NewVar + TotalLen);\r
\r
  NewVar->CheckSum = (UINT16) (~NetblockChecksum ((UINT8 *) NewVar, TotalLen));\r
  return NewVar;\r
}\r
\r
/**\r
  Fix the RouteTable pointer in an EFI_IP4_IPCONFIG_DATA structure. \r
  \r
  The pointer is set to be immediately follow the ConfigData if there're entries\r
  in the RouteTable. Otherwise it is set to NULL.\r
  \r
  @param  ConfigData     The IP4 IP configure data.\r
\r
**/\r
VOID\r
Ip4ConfigFixRouteTablePointer (\r
  IN OUT EFI_IP4_IPCONFIG_DATA  *ConfigData\r
  )\r
{\r
  //\r
  // The memory used for route table entries must immediately follow \r
  // the ConfigData and be not packed.\r
  //\r
  if (ConfigData->RouteTableSize > 0) {\r
    ConfigData->RouteTable = (EFI_IP4_ROUTE_TABLE *) (ConfigData + 1);\r
  } else {\r
    ConfigData->RouteTable = NULL;\r
  }\r
}\r
\r
Commit	Line	Data
b2570da8	1	/** @file\r
3e8c18da	2	Routines used to operate the Ip4 configure variable.\r
b2570da8	3	\r
e5eed7d3 HT	4	Copyright (c) 2006 - 2009, Intel Corporation. All rights reserved.<BR>\r
e5eed7d3 HT	5	This program and the accompanying materials\r
b2570da8	6	are licensed and made available under the terms and conditions of the BSD License\r
402fa70f	7	which accompanies this distribution. The full text of the license may be found at<BR>\r
b2570da8	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
b2570da8	13	**/\r
b2570da8	14	\r
b2570da8	15	#include "NicIp4Variable.h"\r
	16	\r
	17	\r
	18	/**\r
	19	Check whether the configure parameter is valid.\r
	20	\r
	21	@param NicConfig The configure parameter to check\r
	22	\r
	23	@return TRUE if the parameter is valid for the interface, otherwise FALSE.\r
	24	\r
	25	**/\r
	26	BOOLEAN\r
	27	Ip4ConfigIsValid (\r
	28	IN NIC_IP4_CONFIG_INFO *NicConfig\r
	29	)\r
	30	{\r
	31	EFI_IP4_IPCONFIG_DATA *IpConfig;\r
	32	IP4_ADDR Station;\r
	33	IP4_ADDR Netmask;\r
	34	IP4_ADDR Gateway;\r
	35	UINT32 Index;\r
	36	\r
	37	IpConfig = &NicConfig->Ip4Info;\r
	38	\r
	39	if (NicConfig->Source == IP4_CONFIG_SOURCE_STATIC) {\r
	40	//\r
	41	// Validate that the addresses are unicast and mask\r
	42	// is properly formated\r
	43	//\r
	44	Station = EFI_NTOHL (IpConfig->StationAddress);\r
	45	Netmask = EFI_NTOHL (IpConfig->SubnetMask);\r
	46	\r
	47	if ((Netmask == 0) \|\| !IP4_IS_VALID_NETMASK (Netmask) \|\|\r
f6b7393c	48	(Station == 0) \|\| !NetIp4IsUnicast (Station, Netmask)) {\r
b2570da8	49	return FALSE;\r
	50	}\r
	51	\r
	52	//\r
	53	// Validate that the next hops are on the connected network\r
	54	// or that is a direct route (Gateway == 0).\r
	55	//\r
	56	for (Index = 0; Index < IpConfig->RouteTableSize; Index++) {\r
	57	Gateway = EFI_NTOHL (IpConfig->RouteTable[Index].GatewayAddress);\r
	58	\r
	59	if ((Gateway != 0) && (!IP4_NET_EQUAL (Station, Gateway, Netmask) \|\|\r
f6b7393c	60	!NetIp4IsUnicast (Gateway, Netmask))) {\r
b2570da8	61	return FALSE;\r
	62	}\r
	63	}\r
	64	\r
	65	return TRUE;\r
	66	}\r
	67	\r
	68	//\r
	69	// return false if it is an unkown configure source. Valid\r
	70	// sources are static and dhcp.\r
	71	//\r
	72	return (BOOLEAN) (NicConfig->Source == IP4_CONFIG_SOURCE_DHCP);\r
	73	}\r
	74	\r
	75	\r
	76	\r
	77	/**\r
7bce0c5a	78	Read the ip4 configure variable from the EFI variable.\r
b2570da8	79	\r
	80	None\r
	81	\r
	82	@return The IP4 configure read if it is there and is valid, otherwise NULL\r
	83	\r
	84	**/\r
	85	IP4_CONFIG_VARIABLE *\r
	86	Ip4ConfigReadVariable (\r
	87	VOID\r
	88	)\r
	89	{\r
	90	IP4_CONFIG_VARIABLE *Variable;\r
	91	EFI_STATUS Status;\r
	92	UINTN Size;\r
	93	UINT16 CheckSum;\r
	94	\r
	95	//\r
	96	// Get the size of variable, then allocate a buffer to read the variable.\r
	97	//\r
	98	Size = 0;\r
	99	Variable = NULL;\r
	100	Status = gRT->GetVariable (\r
	101	EFI_NIC_IP4_CONFIG_VARIABLE,\r
	102	&gEfiNicIp4ConfigVariableGuid,\r
	103	NULL,\r
	104	&Size,\r
	105	NULL\r
	106	);\r
	107	\r
	108	if (Status != EFI_BUFFER_TOO_SMALL) {\r
	109	return NULL;\r
	110	}\r
	111	\r
	112	if (Size < sizeof (IP4_CONFIG_VARIABLE)) {\r
	113	goto REMOVE_VARIABLE;\r
	114	}\r
	115	\r
e48e37fc	116	Variable = AllocatePool (Size);\r
b2570da8	117	\r
	118	if (Variable == NULL) {\r
	119	return NULL;\r
	120	}\r
	121	\r
	122	Status = gRT->GetVariable (\r
	123	EFI_NIC_IP4_CONFIG_VARIABLE,\r
	124	&gEfiNicIp4ConfigVariableGuid,\r
	125	NULL,\r
	126	&Size,\r
	127	Variable\r
	128	);\r
	129	\r
	130	if (EFI_ERROR (Status)) {\r
	131	goto ON_ERROR;\r
	132	}\r
	133	\r
	134	//\r
	135	// Verify the checksum, variable size and count\r
	136	//\r
4eb65aff	137	CheckSum = (UINT16) (~NetblockChecksum ((UINT8 *) Variable, (UINT32)Size));\r
b2570da8	138	\r
	139	if ((CheckSum != 0) \|\| (Size != Variable->Len)) {\r
	140	goto REMOVE_VARIABLE;\r
	141	}\r
	142	\r
	143	if ((Variable->Count < 1) \|\| (Variable->Count > MAX_IP4_CONFIG_IN_VARIABLE)) {\r
	144	goto REMOVE_VARIABLE;\r
	145	}\r
	146	\r
	147	return Variable;\r
	148	\r
	149	REMOVE_VARIABLE:\r
	150	Ip4ConfigWriteVariable (NULL);\r
	151	\r
	152	ON_ERROR:\r
	153	if (Variable != NULL) {\r
766c7483	154	FreePool (Variable);\r
b2570da8	155	}\r
	156	\r
	157	return NULL;\r
	158	}\r
	159	\r
	160	\r
	161	/**\r
	162	Write the IP4 configure variable to the NVRAM. If Config\r
	163	is NULL, remove the variable.\r
	164	\r
	165	@param Config The IP4 configure data to write\r
	166	\r
	167	@retval EFI_SUCCESS The variable is written to the NVRam\r
	168	@retval Others Failed to write the variable.\r
	169	\r
	170	**/\r
	171	EFI_STATUS\r
	172	Ip4ConfigWriteVariable (\r
7bce0c5a	173	IN IP4_CONFIG_VARIABLE *Config OPTIONAL\r
b2570da8	174	)\r
	175	{\r
	176	EFI_STATUS Status;\r
	177	\r
	178	Status = gRT->SetVariable (\r
	179	EFI_NIC_IP4_CONFIG_VARIABLE,\r
	180	&gEfiNicIp4ConfigVariableGuid,\r
	181	IP4_CONFIG_VARIABLE_ATTRIBUTES,\r
	182	(Config == NULL) ? 0 : Config->Len,\r
	183	Config\r
	184	);\r
	185	\r
	186	return Status;\r
	187	}\r
	188	\r
	189	\r
	190	/**\r
	191	Locate the IP4 configure parameters from the variable.If a\r
	192	configuration is found, copy it to a newly allocated block\r
	193	of memory to avoid the alignment problem. Caller should\r
	194	release the memory after use.\r
	195	\r
	196	@param Variable The IP4 configure variable to search in\r
	197	@param NicAddr The interface address to check\r
	198	\r
	199	@return The point to the NIC's IP4 configure info if it is found\r
c8d8f1e3	200	in the IP4 variable, otherwise NULL.\r
b2570da8	201	\r
	202	**/\r
	203	NIC_IP4_CONFIG_INFO *\r
	204	Ip4ConfigFindNicVariable (\r
	205	IN IP4_CONFIG_VARIABLE *Variable,\r
	206	IN NIC_ADDR *NicAddr\r
	207	)\r
	208	{\r
	209	NIC_IP4_CONFIG_INFO Temp;\r
	210	NIC_IP4_CONFIG_INFO *Config;\r
	211	UINT32 Index;\r
	212	UINT8 *Cur;\r
	213	UINT32 Len;\r
	214	\r
	215	Cur = (UINT8*)&Variable->ConfigInfo;\r
	216	\r
	217	for (Index = 0; Index < Variable->Count; Index++) {\r
	218	//\r
	219	// Copy the data to Temp to avoid the alignment problems\r
	220	//\r
e48e37fc	221	CopyMem (&Temp, Cur, sizeof (NIC_IP4_CONFIG_INFO));\r
b2570da8	222	Len = SIZEOF_NIC_IP4_CONFIG_INFO (&Temp);\r
	223	\r
	224	//\r
	225	// Found the matching configuration parameters, allocate\r
	226	// a block of memory then copy it out.\r
	227	//\r
	228	if (NIC_ADDR_EQUAL (&Temp.NicAddr, NicAddr)) {\r
e48e37fc	229	Config = AllocatePool (Len);\r
b2570da8	230	\r
	231	if (Config == NULL) {\r
	232	return NULL;\r
	233	}\r
	234	\r
e48e37fc	235	CopyMem (Config, Cur, Len);\r
7659d0c9	236	Ip4ConfigFixRouteTablePointer (&Config->Ip4Info);\r
b2570da8	237	return Config;\r
	238	}\r
	239	\r
	240	Cur += Len;\r
	241	}\r
	242	\r
	243	return NULL;\r
	244	}\r
	245	\r
	246	\r
	247	/**\r
	248	Modify the configuration parameter for the NIC in the variable.\r
	249	If Config is NULL, old configuration will be remove from the new\r
	250	variable. Otherwise, append it or replace the old one.\r
	251	\r
	252	@param Variable The IP4 variable to change\r
	253	@param NicAddr The interface to search\r
	254	@param Config The new configuration parameter (NULL to remove the old)\r
	255	\r
	256	@return The new IP4_CONFIG_VARIABLE variable if the new variable has at\r
c8d8f1e3	257	least one NIC configure and no EFI_OUT_OF_RESOURCES failure.\r
	258	Return NULL either because failed to locate memory for new variable\r
	259	or the only NIC configure is removed from the Variable.\r
b2570da8	260	\r
	261	**/\r
	262	IP4_CONFIG_VARIABLE *\r
	263	Ip4ConfigModifyVariable (\r
3e8c18da	264	IN IP4_CONFIG_VARIABLE *Variable OPTIONAL,\r
b2570da8	265	IN NIC_ADDR *NicAddr,\r
	266	IN NIC_IP4_CONFIG_INFO *Config OPTIONAL\r
	267	)\r
	268	{\r
	269	NIC_IP4_CONFIG_INFO Temp;\r
	270	NIC_IP4_CONFIG_INFO *Old;\r
	271	IP4_CONFIG_VARIABLE *NewVar;\r
	272	UINT32 Len;\r
	273	UINT32 TotalLen;\r
	274	UINT32 Count;\r
	275	UINT8 *Next;\r
	276	UINT8 *Cur;\r
	277	UINT32 Index;\r
	278	\r
	279	ASSERT ((Variable != NULL) \|\| (Config != NULL));\r
	280	\r
	281	//\r
	282	// Compute the total length\r
	283	//\r
	284	if (Variable != NULL) {\r
	285	//\r
	286	// Variable != NULL, then Config can be NULL or not. and so is\r
	287	// the Old. If old configure exists, it is removed from the\r
	288	// Variable. New configure is append to the variable.\r
	289	//\r
	290	//\r
	291	Count = Variable->Count;\r
	292	Cur = (UINT8 *)&Variable->ConfigInfo;\r
	293	TotalLen = Variable->Len;\r
	294	\r
	295	Old = Ip4ConfigFindNicVariable (Variable, NicAddr);\r
	296	\r
	297	if (Old != NULL) {\r
	298	TotalLen -= SIZEOF_NIC_IP4_CONFIG_INFO (Old);\r
766c7483	299	FreePool (Old);\r
b2570da8	300	}\r
	301	\r
	302	if (Config != NULL) {\r
	303	TotalLen += SIZEOF_NIC_IP4_CONFIG_INFO (Config);\r
	304	}\r
	305	\r
	306	//\r
	307	// Return NULL if the only NIC_IP4_CONFIG_INFO is being removed.\r
	308	//\r
	309	if (TotalLen < sizeof (IP4_CONFIG_VARIABLE)) {\r
	310	return NULL;\r
	311	}\r
	312	\r
	313	} else {\r
	314	//\r
	315	// Variable == NULL and Config != NULL, Create a new variable with\r
	316	// this NIC configure.\r
	317	//\r
	318	Count = 0;\r
	319	Cur = NULL;\r
	320	TotalLen = sizeof (IP4_CONFIG_VARIABLE) - sizeof (NIC_IP4_CONFIG_INFO)\r
	321	+ SIZEOF_NIC_IP4_CONFIG_INFO (Config);\r
	322	}\r
	323	\r
	324	ASSERT (TotalLen >= sizeof (IP4_CONFIG_VARIABLE));\r
	325	\r
e48e37fc	326	NewVar = AllocateZeroPool (TotalLen);\r
b2570da8	327	\r
	328	if (NewVar == NULL) {\r
	329	return NULL;\r
	330	}\r
	331	\r
	332	NewVar->Len = TotalLen;\r
	333	\r
	334	//\r
	335	// Copy the other configure parameters from the old variable\r
	336	//\r
	337	Next = (UINT8 *)&NewVar->ConfigInfo;\r
	338	\r
	339	for (Index = 0; Index < Count; Index++) {\r
e48e37fc	340	CopyMem (&Temp, Cur, sizeof (NIC_IP4_CONFIG_INFO));\r
b2570da8	341	Len = SIZEOF_NIC_IP4_CONFIG_INFO (&Temp);\r
	342	\r
	343	if (!NIC_ADDR_EQUAL (&Temp.NicAddr, NicAddr)) {\r
e48e37fc	344	CopyMem (Next, Cur, Len);\r
b2570da8	345	Next += Len;\r
	346	NewVar->Count++;\r
	347	}\r
	348	\r
	349	Cur += Len;\r
	350	}\r
	351	\r
	352	//\r
	353	// Append the new configure if it isn't NULL.\r
	354	//\r
	355	Len = 0;\r
	356	\r
	357	if (Config != NULL) {\r
	358	Len = SIZEOF_NIC_IP4_CONFIG_INFO (Config);\r
	359	\r
e48e37fc	360	CopyMem (Next, Config, Len);\r
b2570da8	361	NewVar->Count++;\r
	362	}\r
	363	\r
	364	ASSERT (Next + Len == (UINT8 *) NewVar + TotalLen);\r
	365	\r
4eb65aff	366	NewVar->CheckSum = (UINT16) (~NetblockChecksum ((UINT8 *) NewVar, TotalLen));\r
b2570da8	367	return NewVar;\r
b2570da8	368	}\r
7659d0c9	369	\r
402fa70f	370	/**\r
	371	Fix the RouteTable pointer in an EFI_IP4_IPCONFIG_DATA structure. \r
	372	\r
	373	The pointer is set to be immediately follow the ConfigData if there're entries\r
	374	in the RouteTable. Otherwise it is set to NULL.\r
	375	\r
	376	@param ConfigData The IP4 IP configure data.\r
	377	\r
	378	**/\r
7659d0c9	379	VOID\r
7659d0c9	380	Ip4ConfigFixRouteTablePointer (\r
c8d8f1e3	381	IN OUT EFI_IP4_IPCONFIG_DATA *ConfigData\r
7659d0c9	382	)\r
	383	{\r
	384	//\r
	385	// The memory used for route table entries must immediately follow \r
	386	// the ConfigData and be not packed.\r
	387	//\r
	388	if (ConfigData->RouteTableSize > 0) {\r
	389	ConfigData->RouteTable = (EFI_IP4_ROUTE_TABLE *) (ConfigData + 1);\r
	390	} else {\r
	391	ConfigData->RouteTable = NULL;\r
	392	}\r
	393	}\r
	394	\r