efi_loader: efi_net: add efi_net_set_addr, efi_net_get_addr

Add the functions efi_net_set_addr and  efi_net_get_addr to set
and get the ip address from efi code in a network agnostic way.
This could also go in net_common, or be compiled conditionally
for each network stack.

Signed-off-by: Adriano Cordova <adrianox@gmail.com>
Reviewed-by: Heinrich Schuchardt <xypron.glpk@gmx.de>
This commit is contained in:
Adriano Cordova
2024-12-04 00:05:21 -03:00
committed by Heinrich Schuchardt
parent b98d232350
commit b20f497f22
2 changed files with 141 additions and 0 deletions

View File

@@ -17,6 +17,7 @@
#include <efi_loader.h>
#include <malloc.h>
#include <vsprintf.h>
#include <net.h>
static const efi_guid_t efi_net_guid = EFI_SIMPLE_NETWORK_PROTOCOL_GUID;
@@ -997,3 +998,127 @@ out_of_resources:
printf("ERROR: Out of memory\n");
return EFI_OUT_OF_RESOURCES;
}
/**
* efi_net_get_addr() - get IP address information
*
* Copy the current IP address, mask, and gateway into the
* efi_ipv4_address structs pointed to by ip, mask and gw,
* respectively.
*
* @ip: pointer to an efi_ipv4_address struct to
* be filled with the current IP address
* @mask: pointer to an efi_ipv4_address struct to
* be filled with the current network mask
* @gw: pointer to an efi_ipv4_address struct to be
* filled with the current network gateway
*/
void efi_net_get_addr(struct efi_ipv4_address *ip,
struct efi_ipv4_address *mask,
struct efi_ipv4_address *gw)
{
#ifdef CONFIG_NET_LWIP
char ipstr[] = "ipaddr\0\0";
char maskstr[] = "netmask\0\0";
char gwstr[] = "gatewayip\0\0";
int idx;
struct in_addr tmp;
char *env;
idx = dev_seq(eth_get_dev());
if (idx < 0 || idx > 99) {
log_err("unexpected idx %d\n", idx);
return;
}
if (idx) {
sprintf(ipstr, "ipaddr%d", idx);
sprintf(maskstr, "netmask%d", idx);
sprintf(gwstr, "gatewayip%d", idx);
}
env = env_get(ipstr);
if (env && ip) {
tmp = string_to_ip(env);
memcpy(ip, &tmp, sizeof(tmp));
}
env = env_get(maskstr);
if (env && mask) {
tmp = string_to_ip(env);
memcpy(mask, &tmp, sizeof(tmp));
}
env = env_get(gwstr);
if (env && gw) {
tmp = string_to_ip(env);
memcpy(gw, &tmp, sizeof(tmp));
}
#else
if (ip)
memcpy(ip, &net_ip, sizeof(net_ip));
if (mask)
memcpy(mask, &net_netmask, sizeof(net_netmask));
#endif
}
/**
* efi_net_set_addr() - set IP address information
*
* Set the current IP address, mask, and gateway to the
* efi_ipv4_address structs pointed to by ip, mask and gw,
* respectively.
*
* @ip: pointer to new IP address
* @mask: pointer to new network mask to set
* @gw: pointer to new network gateway
*/
void efi_net_set_addr(struct efi_ipv4_address *ip,
struct efi_ipv4_address *mask,
struct efi_ipv4_address *gw)
{
#ifdef CONFIG_NET_LWIP
char ipstr[] = "ipaddr\0\0";
char maskstr[] = "netmask\0\0";
char gwstr[] = "gatewayip\0\0";
int idx;
struct in_addr *addr;
char tmp[46];
idx = dev_seq(eth_get_dev());
if (idx < 0 || idx > 99) {
log_err("unexpected idx %d\n", idx);
return;
}
if (idx) {
sprintf(ipstr, "ipaddr%d", idx);
sprintf(maskstr, "netmask%d", idx);
sprintf(gwstr, "gatewayip%d", idx);
}
if (ip) {
addr = (struct in_addr *)ip;
ip_to_string(*addr, tmp);
env_set(ipstr, tmp);
}
if (mask) {
addr = (struct in_addr *)mask;
ip_to_string(*addr, tmp);
env_set(maskstr, tmp);
}
if (gw) {
addr = (struct in_addr *)gw;
ip_to_string(*addr, tmp);
env_set(gwstr, tmp);
}
#else
if (ip)
memcpy(&net_ip, ip, sizeof(*ip));
if (mask)
memcpy(&net_netmask, mask, sizeof(*mask));
#endif
}