1*** Settings ***
2Documentation    Vmi network utilities keywords.
3
4Resource         ../../lib/resource.robot
5Resource         ../../lib/bmc_redfish_resource.robot
6Resource         ../../lib/openbmc_ffdc.robot
7Resource         ../../lib/bmc_redfish_utils.robot
8Resource         ../../lib/state_manager.robot
9Resource         ../../lib/bmc_network_utils.robot
10Library          ../../lib/bmc_network_utils.py
11
12*** Variables ***
13
14&{DHCP_ENABLED}           DHCPEnabled=${True}
15&{DHCP_DISABLED}          DHCPEnabled=${False}
16
17&{ENABLE_DHCP}            DHCPv4=&{DHCP_ENABLED}
18&{DISABLE_DHCP}           DHCPv4=&{DHCP_DISABLED}
19
20${wait_time}              10s
21${ethernet_interface}     eth0
22
23*** Keywords ***
24
25Set Static IPv4 Address To VMI And Verify
26    [Documentation]  Set static IPv4 address to VMI.
27    [Arguments]  ${ip}  ${gateway}  ${netmask}  ${valid_status_code}=${HTTP_ACCEPTED}
28    ...  ${interface}=${ethernet_interface}
29
30    # Description of argument(s):
31    # ip                 VMI IPv4 address.
32    # gateway            Gateway for VMI IP.
33    # netmask            Subnetmask for VMI IP.
34    # valid_status_code  Expected valid status code from GET request. Default is HTTP_ACCEPTED.
35    # interface          VMI interface (eg. eth0 or eth1).
36
37    ${ip_details}=  Create dictionary  Address=${ip}  SubnetMask=${netmask}  Gateway=${gateway}
38    ${ip_data}=  Create List  ${ip_details}
39    ${resp}=  Redfish.Patch  /redfish/v1/Systems/hypervisor/EthernetInterfaces/${interface}
40    ...  body={'IPv4StaticAddresses':${ip_data}}  valid_status_codes=[${valid_status_code}]
41
42    # Wait few seconds for new configuration to get populated on runtime.
43    Sleep  ${wait_time}
44
45    Return From Keyword If  ${valid_status_code} != ${HTTP_ACCEPTED}
46    ${host_power_state}  ${host_state}=   Redfish Get Host State
47    Run Keyword If  '${host_power_state}' == 'On' and '${host_state}' == 'Enabled'
48    ...  Verify VMI Network Interface Details  ${ip}  Static  ${gateway}  ${netmask}  ${interface}
49
50Verify VMI Network Interface Details
51    [Documentation]  Verify VMI network interface details.
52    [Arguments]  ${ip}  ${origin}  ${gateway}  ${netmask}
53    ...  ${interface}=${ethernet_interface}  ${valid_status_code}=${HTTP_OK}
54
55    # Description of argument(s):
56    # ip                 VMI IPv4 address.
57    # origin             Origin of IPv4 address eg. Static or DHCP.
58    # gateway            Gateway for VMI IP.
59    # netmask            Subnetmask for VMI IP.
60    # interface          VMI interface (eg. eth0 or eth1).
61    # valid_status_code  Expected valid status code from GET request. Default is HTTP_OK.
62
63    ${vmi_ip}=  Get VMI Network Interface Details  ${interface}  ${valid_status_code}
64    Should Be Equal As Strings  ${origin}  ${vmi_ip["IPv4_AddressOrigin"]}
65    Should Be Equal As Strings  ${gateway}  ${vmi_ip["IPv4_Gateway"]}
66    Should Be Equal As Strings  ${netmask}  ${vmi_ip["IPv4_SubnetMask"]}
67    Should Be Equal As Strings  ${ip}  ${vmi_ip["IPv4_Address"]}
68
69Delete VMI IPv4 Address
70    [Documentation]  Delete VMI IPv4 address.
71    [Arguments]  ${delete_param}=IPv4StaticAddresses  ${valid_status_code}=${HTTP_ACCEPTED}
72    ...  ${interface}=${ethernet_interface}
73
74    # Description of argument(s):
75    # delete_param       Parameter to be deleted eg. IPv4StaticAddresses or IPv4Addresses.
76    #                    Default is IPv4StaticAddresses.
77    # valid_status_code  Expected valid status code from PATCH request. Default is HTTP_OK.
78    # interface          VMI interface (eg. eth0 or eth1).
79
80    ${data}=  Set Variable  {"${delete_param}": [${Null}]}
81    ${resp}=  Redfish.Patch
82    ...  /redfish/v1/Systems/hypervisor/EthernetInterfaces/${interface}
83    ...  body=${data}  valid_status_codes=[${valid_status_code}]
84
85    Return From Keyword If  ${valid_status_code} != ${HTTP_ACCEPTED}
86
87    # Wait few seconds for configuration to get effective.
88    Sleep  ${wait_time}
89    ${vmi_ip}=  Get VMI Network Interface Details  ${interface}
90    ${default}=  Set Variable  0.0.0.0
91    Verify VMI Network Interface Details  ${default}  Static  ${default}  ${default}
92
93Set VMI IPv4 Origin
94    [Documentation]  Set VMI IPv4 origin.
95    [Arguments]  ${dhcp_enabled}=${False}  ${valid_status_code}=${HTTP_ACCEPTED}
96    ...  ${interface}=${ethernet_interface}
97
98    # Description of argument(s):
99    # dhcp_enabled       True if user wants to enable DHCP. Default is Static, hence value is set to False.
100    # valid_status_code  Expected valid status code from PATCH request. Default is HTTP_OK.
101    # interface          VMI interface (eg. eth0 or eth1).
102
103    ${data}=  Set Variable If  ${dhcp_enabled} == ${False}  ${DISABLE_DHCP}  ${ENABLE_DHCP}
104    ${resp}=  Redfish.Patch
105    ...  /redfish/v1/Systems/hypervisor/EthernetInterfaces/${interface}
106    ...  body=${data}  valid_status_codes=[${valid_status_code}]
107
108    Sleep  ${wait_time}
109    Return From Keyword If  ${valid_status_code} != ${HTTP_ACCEPTED}
110    ${resp}=  Redfish.Get
111    ...  /redfish/v1/Systems/hypervisor/EthernetInterfaces/${interface}
112    Should Be Equal  ${resp.dict["DHCPv4"]["DHCPEnabled"]}  ${dhcp_enabled}
113
114Get VMI Network Interface Details
115    [Documentation]  Get VMI network interface details.
116    [Arguments]  ${interface}=${ethernet_interface}  ${valid_status_code}=${HTTP_OK}
117
118    # Description of argument(s):
119    # interface          VMI interface (eg. eth0 or eth1).
120    # valid_status_code  Expected valid status code from GET request.
121
122    # Note: It returns a dictionary of VMI ethernet interface parameters.
123
124    ${resp}=  Redfish.Get
125    ...  /redfish/v1/Systems/hypervisor/EthernetInterfaces/${interface}
126    ...  valid_status_codes=[${valid_status_code}]
127
128    ${ip_resp}=  Evaluate  json.loads(r'''${resp.text}''')  json
129
130    ${ip_exists}=  Set Variable If  ${ip_resp["IPv4Addresses"]} == @{empty}  ${False}  ${True}
131    ${static_exists}=  Set Variable If  ${ip_resp["IPv4StaticAddresses"]} == @{empty}  ${False}  ${True}
132
133    ${vmi_ip}=  Run Keyword If   ${ip_exists} == ${True}
134    ...  Create Dictionary  DHCPv4=${${ip_resp["DHCPv4"]["DHCPEnabled"]}}  Id=${ip_resp["Id"]}
135    ...  Description=${ip_resp["Description"]}  IPv4_Address=${ip_resp["IPv4Addresses"][0]["Address"]}
136    ...  IPv4_AddressOrigin=${ip_resp["IPv4Addresses"][0]["AddressOrigin"]}  Name=${ip_resp["Name"]}
137    ...  IPv4_Gateway=${ip_resp["IPv4Addresses"][0]["Gateway"]}
138    ...  InterfaceEnabled=${${ip_resp["InterfaceEnabled"]}}
139    ...  IPv4_SubnetMask=${ip_resp["IPv4Addresses"][0]["SubnetMask"]}
140    ...  IPv4StaticAddresses=${${static_exists}}
141    ...  ELSE
142    ...  Create Dictionary  DHCPv4=${${ip_resp["DHCPv4"]["DHCPEnabled"]}}  Id=${ip_resp["Id"]}
143    ...  Description=${ip_resp["Description"]}  IPv4StaticAddresses=${ip_resp["IPv4StaticAddresses"]}
144    ...  IPv4_Address=${ip_resp["IPv4Addresses"]}  Name=${ip_resp["Name"]}
145    ...  InterfaceEnabled=${${ip_resp["InterfaceEnabled"]}}
146
147    [Return]  &{vmi_ip}
148
149
150Get VMI Interfaces
151    [Documentation]  Get VMI network interface.
152    [Arguments]  ${valid_status_code}=${HTTP_OK}
153
154    # Description of argument(s):
155    # valid_status_code  Expected valid status code from GET request.
156    #                    By default set to ${HTTP_OK}.
157
158    ${resp}=  Redfish.Get  /redfish/v1/Systems/hypervisor/EthernetInterfaces
159    ...  valid_status_codes=[${valid_status_code}]
160
161    ${resp}=  Evaluate  json.loads(r'''${resp.text}''')  json
162    ${interfaces_uri}=  Set Variable  ${resp["Members"]}
163    ${interface_list}=  Create List
164    ${number_of_interfaces}=  Get Length  ${interfaces_uri}
165    FOR  ${interface}  IN RANGE  ${number_of_interfaces}
166        ${_}  ${interface_value}=  Split String From Right  ${interfaces_uri[${interface}]}[@odata.id]  /  1
167        Append To List  ${interface_list}  ${interface_value}
168    END
169
170   [Return]  @{interface_list}
171
172
173Verify VMI EthernetInterfaces
174    [Documentation]  Verify VMI ethernet interfaces.
175    [Arguments]  ${valid_status_code}=${HTTP_OK}
176
177    # Description of argument(s):
178    # valid_status_code  Expected valid status code from GET request.
179
180    ${resp}=  Redfish.Get  /redfish/v1/Systems/hypervisor/EthernetInterfaces
181    ...  valid_status_codes=[${valid_status_code}]
182
183    ${resp}=  Evaluate  json.loads(r'''${resp.text}''')  json
184    ${interfaces}=  Set Variable  ${resp["Members"]}
185
186    ${number_of_interfaces}=  Get Length  ${interfaces}
187    FOR  ${i}  IN RANGE  ${number_of_interfaces}
188        Should Be Equal As Strings  ${interfaces[${i}]}[@odata.id]
189        ...  /redfish/v1/Systems/hypervisor/EthernetInterfaces/eth${i}
190    END
191    Should Be Equal  ${resp["Members@odata.count"]}  ${number_of_interfaces}
192
193Get And Set Static VMI IP
194    [Documentation]  Get a suitable VMI IP and set it.
195    [Arguments]   ${host}=${OPENBMC_HOST}  ${network_active_channel}=${CHANNEL_NUMBER}
196    ...  ${interface}=eth0  ${valid_status_code}=${HTTP_ACCEPTED}
197
198    # Description of argument(s):
199    # host                    BMC host name or IP address.
200    # network_active_channel  Ethernet channel number (e.g.1 or 2).
201    # interface               VMI interface (eg. eth0 or eth1).
202    # valid_status_code       Expected valid status code from PATCH request. Default is HTTP_ACCEPTED.
203
204    ${vmi_ip}=  Get First Non Pingable IP From Subnet  ${host}
205    ${bmc_ip_data}=  Get Network Configuration  ${network_active_channel}
206
207    Set Static IPv4 Address To VMI And Verify  ${vmi_ip}  ${bmc_ip_data[0]['Gateway']}
208    ...  ${bmc_ip_data[0]['SubnetMask']}  ${valid_status_code}  ${interface}
209
210    [Return]   ${vmi_ip}  ${bmc_ip_data}
211