]> git.proxmox.com Git - mirror_edk2.git/blame - NetworkPkg/Application/IpsecConfig/PolicyEntryOperation.c
projects / mirror_edk2.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
New Shell binaries based on EDKII core r12898 and Shell r57.
[mirror_edk2.git] / NetworkPkg / Application / IpsecConfig / PolicyEntryOperation.c
CommitLineData
a3bcde70
HT		 1/** @file\r
2 The implementation of policy entry operation function in IpSecConfig application.\r
3\r
686d4d4a 4 Copyright (c) 2009 - 2011, Intel Corporation. All rights reserved.<BR>\r
a3bcde70
HT		 5\r
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 "IpSecConfig.h"\r
17#include "Indexer.h"\r
18#include "Match.h"\r
19#include "Helper.h"\r
20#include "ForEach.h"\r
21#include "PolicyEntryOperation.h"\r
22\r
23/**\r
24 Fill in EFI_IPSEC_SPD_SELECTOR through ParamPackage list.\r
25\r
26 @param[out] Selector The pointer to the EFI_IPSEC_SPD_SELECTOR structure.\r
27 @param[in] ParamPackage The pointer to the ParamPackage list.\r
76389e18 28 @param[in, out] Mask The pointer to the Mask.\r
a3bcde70
HT		 29\r
30 @retval EFI_SUCCESS Fill in EFI_IPSEC_SPD_SELECTOR successfully.\r
31 @retval EFI_INVALID_PARAMETER Invalid user input parameter.\r
32\r
33**/\r
34EFI_STATUS\r
35CreateSpdSelector (\r
36 OUT EFI_IPSEC_SPD_SELECTOR *Selector,\r
37 IN LIST_ENTRY *ParamPackage,\r
38 IN OUT UINT32 *Mask\r
39 )\r
40{\r
41 EFI_STATUS Status;\r
42 EFI_STATUS ReturnStatus;\r
43 CONST CHAR16 *ValueStr;\r
44\r
45 Status = EFI_SUCCESS;\r
46 ReturnStatus = EFI_SUCCESS;\r
47\r
48 //\r
49 // Convert user imput from string to integer, and fill in the member in EFI_IPSEC_SPD_SELECTOR.\r
50 //\r
51 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--local");\r
52 if (ValueStr != NULL) {\r
53 Selector->LocalAddressCount = 1;\r
54 Status = EfiInetAddrRange ((CHAR16 *) ValueStr, Selector->LocalAddress);\r
55 if (EFI_ERROR (Status)) {\r
56 ShellPrintHiiEx (\r
57 -1,\r
58 -1,\r
59 NULL,\r
60 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
61 mHiiHandle,\r
62 mAppName,\r
63 L"--local",\r
64 ValueStr\r
65 );\r
66 ReturnStatus = EFI_INVALID_PARAMETER;\r
67 } else {\r
68 *Mask |= LOCAL;\r
69 }\r
70 }\r
71\r
72 //\r
73 // Convert user imput from string to integer, and fill in the member in EFI_IPSEC_SPD_SELECTOR.\r
74 //\r
75 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--remote");\r
76 if (ValueStr != NULL) {\r
77 Selector->RemoteAddressCount = 1;\r
78 Status = EfiInetAddrRange ((CHAR16 *) ValueStr, Selector->RemoteAddress);\r
79 if (EFI_ERROR (Status)) {\r
80 ShellPrintHiiEx (\r
81 -1,\r
82 -1,\r
83 NULL,\r
84 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
85 mHiiHandle,\r
86 mAppName,\r
87 L"--remote",\r
88 ValueStr\r
89 );\r
90 ReturnStatus = EFI_INVALID_PARAMETER;\r
91 } else {\r
92 *Mask |= REMOTE;\r
93 }\r
94 }\r
95\r
96 Selector->NextLayerProtocol = EFI_IPSEC_ANY_PROTOCOL;\r
97\r
98 //\r
99 // Convert user imput from string to integer, and fill in the member in EFI_IPSEC_SPD_SELECTOR.\r
100 //\r
101 Status = GetNumber (\r
102 L"--proto",\r
103 (UINT16) -1,\r
104 &Selector->NextLayerProtocol,\r
105 sizeof (UINT16),\r
106 mMapIpProtocol,\r
107 ParamPackage,\r
108 FORMAT_NUMBER | FORMAT_STRING\r
109 );\r
110 if (!EFI_ERROR (Status)) {\r
111 *Mask |= PROTO;\r
112 }\r
113\r
114 if (Status == EFI_INVALID_PARAMETER) {\r
115 ReturnStatus = EFI_INVALID_PARAMETER;\r
116 }\r
117\r
118 Selector->LocalPort = EFI_IPSEC_ANY_PORT;\r
119 Selector->RemotePort = EFI_IPSEC_ANY_PORT;\r
120\r
121 //\r
122 // Convert user imput from string to integer, and fill in the member in EFI_IPSEC_SPD_SELECTOR.\r
123 //\r
124 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--local-port");\r
125 if (ValueStr != NULL) {\r
126 Status = EfiInetPortRange ((CHAR16 *) ValueStr, &Selector->LocalPort, &Selector->LocalPortRange);\r
127 if (EFI_ERROR (Status)) {\r
128 ShellPrintHiiEx (\r
129 -1,\r
130 -1,\r
131 NULL,\r
132 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
133 mHiiHandle,\r
134 mAppName,\r
135 L"--local-port",\r
136 ValueStr\r
137 );\r
138 ReturnStatus = EFI_INVALID_PARAMETER;\r
139 } else {\r
140 *Mask |= LOCAL_PORT;\r
141 }\r
142 }\r
143\r
144 //\r
145 // Convert user imput from string to integer, and fill in the member in EFI_IPSEC_SPD_SELECTOR.\r
146 //\r
147 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--remote-port");\r
148 if (ValueStr != NULL) {\r
149 Status = EfiInetPortRange ((CHAR16 *) ValueStr, &Selector->RemotePort, &Selector->RemotePortRange);\r
150 if (EFI_ERROR (Status)) {\r
151 ShellPrintHiiEx (\r
152 -1,\r
153 -1,\r
154 NULL,\r
155 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
156 mHiiHandle,\r
157 mAppName,\r
158 L"--remote-port",\r
159 ValueStr\r
160 );\r
161 ReturnStatus = EFI_INVALID_PARAMETER;\r
162 } else {\r
163 *Mask |= REMOTE_PORT;\r
164 }\r
165 }\r
166\r
167 //\r
168 // Convert user imput from string to integer, and fill in the member in EFI_IPSEC_SPD_SELECTOR.\r
169 //\r
170 Status = GetNumber (\r
171 L"--icmp-type",\r
172 (UINT8) -1,\r
173 &Selector->LocalPort,\r
174 sizeof (UINT16),\r
175 NULL,\r
176 ParamPackage,\r
177 FORMAT_NUMBER\r
178 );\r
179 if (!EFI_ERROR (Status)) {\r
180 *Mask |= ICMP_TYPE;\r
181 }\r
182\r
183 if (Status == EFI_INVALID_PARAMETER) {\r
184 ReturnStatus = EFI_INVALID_PARAMETER;\r
185 }\r
186\r
187 //\r
188 // Convert user imput from string to integer, and fill in the member in EFI_IPSEC_SPD_SELECTOR.\r
189 //\r
190 Status = GetNumber (\r
191 L"--icmp-code",\r
192 (UINT8) -1,\r
193 &Selector->RemotePort,\r
194 sizeof (UINT16),\r
195 NULL,\r
196 ParamPackage,\r
197 FORMAT_NUMBER\r
198 );\r
199 if (!EFI_ERROR (Status)) {\r
200 *Mask |= ICMP_CODE;\r
201 }\r
202\r
203 if (Status == EFI_INVALID_PARAMETER) {\r
204 ReturnStatus = EFI_INVALID_PARAMETER;\r
205 }\r
206\r
207 return ReturnStatus;\r
208}\r
209\r
210/**\r
211 Fill in EFI_IPSEC_SPD_SELECTOR and EFI_IPSEC_SPD_DATA through ParamPackage list.\r
212\r
213 @param[out] Selector The pointer to the EFI_IPSEC_SPD_SELECTOR structure.\r
214 @param[out] Data The pointer to the EFI_IPSEC_SPD_DATA structure.\r
215 @param[in] ParamPackage The pointer to the ParamPackage list.\r
216 @param[out] Mask The pointer to the Mask.\r
217 @param[in] CreateNew The switch to create new.\r
218\r
219 @retval EFI_SUCCESS Fill in EFI_IPSEC_SPD_SELECTOR and EFI_IPSEC_SPD_DATA successfully.\r
220 @retval EFI_INVALID_PARAMETER Invalid user input parameter.\r
221\r
222**/\r
223EFI_STATUS\r
224CreateSpdEntry (\r
225 OUT EFI_IPSEC_SPD_SELECTOR **Selector,\r
226 OUT EFI_IPSEC_SPD_DATA **Data,\r
227 IN LIST_ENTRY *ParamPackage,\r
228 OUT UINT32 *Mask,\r
229 IN BOOLEAN CreateNew\r
230 )\r
231{\r
232 EFI_STATUS Status;\r
233 EFI_STATUS ReturnStatus;\r
234 CONST CHAR16 *ValueStr;\r
235 UINTN DataSize;\r
236\r
237 Status = EFI_SUCCESS;\r
238 *Mask = 0;\r
239\r
240 *Selector = AllocateZeroPool (sizeof (EFI_IPSEC_SPD_SELECTOR) + 2 * sizeof (EFI_IP_ADDRESS_INFO));\r
241 ASSERT (*Selector != NULL);\r
242\r
243 (*Selector)->LocalAddress = (EFI_IP_ADDRESS_INFO *) (*Selector + 1);\r
244 (*Selector)->RemoteAddress = (*Selector)->LocalAddress + 1;\r
245\r
246 ReturnStatus = CreateSpdSelector (*Selector, ParamPackage, Mask);\r
247\r
248 //\r
249 // SPD DATA\r
250 // NOTE: Allocate enough memory and add padding for different arch.\r
251 //\r
252 DataSize = ALIGN_VARIABLE (sizeof (EFI_IPSEC_SPD_DATA));\r
253 DataSize = ALIGN_VARIABLE (DataSize + sizeof (EFI_IPSEC_PROCESS_POLICY));\r
254 DataSize += sizeof (EFI_IPSEC_TUNNEL_OPTION);\r
255\r
256 *Data = AllocateZeroPool (DataSize);\r
257 ASSERT (*Data != NULL);\r
258\r
259 (*Data)->ProcessingPolicy = (EFI_IPSEC_PROCESS_POLICY *) ALIGN_POINTER (\r
260 (*Data + 1),\r
261 sizeof (UINTN)\r
262 );\r
263 (*Data)->ProcessingPolicy->TunnelOption = (EFI_IPSEC_TUNNEL_OPTION *) ALIGN_POINTER (\r
264 ((*Data)->ProcessingPolicy + 1),\r
265 sizeof (UINTN)\r
266 );\r
267\r
268\r
269 //\r
270 // Convert user imput from string to integer, and fill in the Name in EFI_IPSEC_SPD_DATA.\r
271 //\r
272 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--name");\r
273 if (ValueStr != NULL) {\r
274 UnicodeStrToAsciiStr (ValueStr, (CHAR8 *) (*Data)->Name);\r
275 *Mask |= NAME;\r
276 }\r
277\r
278 //\r
279 // Convert user imput from string to integer, and fill in the PackageFlag in EFI_IPSEC_SPD_DATA.\r
280 //\r
281 Status = GetNumber (\r
282 L"--packet-flag",\r
283 (UINT8) -1,\r
284 &(*Data)->PackageFlag,\r
285 sizeof (UINT32),\r
286 NULL,\r
287 ParamPackage,\r
288 FORMAT_NUMBER\r
289 );\r
290 if (!EFI_ERROR (Status)) {\r
291 *Mask |= PACKET_FLAG;\r
292 }\r
293\r
294 if (Status == EFI_INVALID_PARAMETER) {\r
295 ReturnStatus = EFI_INVALID_PARAMETER;\r
296 }\r
297\r
298 //\r
299 // Convert user imput from string to integer, and fill in the Action in EFI_IPSEC_SPD_DATA.\r
300 //\r
301 Status = GetNumber (\r
302 L"--action",\r
303 (UINT8) -1,\r
304 &(*Data)->Action,\r
305 sizeof (UINT32),\r
306 mMapIpSecAction,\r
307 ParamPackage,\r
308 FORMAT_STRING\r
309 );\r
310 if (!EFI_ERROR (Status)) {\r
311 *Mask |= ACTION;\r
312 }\r
313\r
314 if (Status == EFI_INVALID_PARAMETER) {\r
315 ReturnStatus = EFI_INVALID_PARAMETER;\r
316 }\r
317\r
318 //\r
319 // Convert user imput from string to integer, and fill in the ExtSeqNum in EFI_IPSEC_SPD_DATA.\r
320 //\r
321 if (ShellCommandLineGetFlag (ParamPackage, L"--ext-sequence")) {\r
322 (*Data)->ProcessingPolicy->ExtSeqNum = TRUE;\r
323 *Mask |= EXT_SEQUENCE;\r
324 } else if (ShellCommandLineGetFlag (ParamPackage, L"--ext-sequence-")) {\r
325 (*Data)->ProcessingPolicy->ExtSeqNum = FALSE;\r
326 *Mask |= EXT_SEQUENCE;\r
327 }\r
328\r
329 //\r
330 // Convert user imput from string to integer, and fill in the SeqOverflow in EFI_IPSEC_SPD_DATA.\r
331 //\r
332 if (ShellCommandLineGetFlag (ParamPackage, L"--sequence-overflow")) {\r
333 (*Data)->ProcessingPolicy->SeqOverflow = TRUE;\r
334 *Mask |= SEQUENCE_OVERFLOW;\r
335 } else if (ShellCommandLineGetFlag (ParamPackage, L"--sequence-overflow-")) {\r
336 (*Data)->ProcessingPolicy->SeqOverflow = FALSE;\r
337 *Mask |= SEQUENCE_OVERFLOW;\r
338 }\r
339\r
340 //\r
341 // Convert user imput from string to integer, and fill in the FragCheck in EFI_IPSEC_SPD_DATA.\r
342 //\r
343 if (ShellCommandLineGetFlag (ParamPackage, L"--fragment-check")) {\r
344 (*Data)->ProcessingPolicy->FragCheck = TRUE;\r
345 *Mask |= FRAGMENT_CHECK;\r
346 } else if (ShellCommandLineGetFlag (ParamPackage, L"--fragment-check-")) {\r
347 (*Data)->ProcessingPolicy->FragCheck = FALSE;\r
348 *Mask |= FRAGMENT_CHECK;\r
349 }\r
350\r
351 //\r
352 // Convert user imput from string to integer, and fill in the ProcessingPolicy in EFI_IPSEC_SPD_DATA.\r
353 //\r
354 Status = GetNumber (\r
355 L"--lifebyte",\r
356 (UINT64) -1,\r
357 &(*Data)->ProcessingPolicy->SaLifetime.ByteCount,\r
358 sizeof (UINT64),\r
359 NULL,\r
360 ParamPackage,\r
361 FORMAT_NUMBER\r
362 );\r
363 if (!EFI_ERROR (Status)) {\r
364 *Mask |= LIFEBYTE;\r
365 }\r
366\r
367 if (Status == EFI_INVALID_PARAMETER) {\r
368 ReturnStatus = EFI_INVALID_PARAMETER;\r
369 }\r
370\r
371 Status = GetNumber (\r
372 L"--lifetime",\r
373 (UINT64) -1,\r
374 &(*Data)->ProcessingPolicy->SaLifetime.HardLifetime,\r
375 sizeof (UINT64),\r
376 NULL,\r
377 ParamPackage,\r
378 FORMAT_NUMBER\r
379 );\r
380 if (!EFI_ERROR (Status)) {\r
381 *Mask |= LIFETIME;\r
382 }\r
383 if (Status == EFI_INVALID_PARAMETER) {\r
384 ReturnStatus = EFI_INVALID_PARAMETER;\r
385 }\r
386\r
387 Status = GetNumber (\r
388 L"--lifetime-soft",\r
389 (UINT64) -1,\r
390 &(*Data)->ProcessingPolicy->SaLifetime.SoftLifetime,\r
391 sizeof (UINT64),\r
392 NULL,\r
393 ParamPackage,\r
394 FORMAT_NUMBER\r
395 );\r
396 if (!EFI_ERROR (Status)) {\r
397 *Mask |= LIFETIME_SOFT;\r
398 }\r
399\r
400 if (Status == EFI_INVALID_PARAMETER) {\r
401 ReturnStatus = EFI_INVALID_PARAMETER;\r
402 }\r
403\r
404 (*Data)->ProcessingPolicy->Mode = EfiIPsecTransport;\r
405 Status = GetNumber (\r
406 L"--mode",\r
407 0,\r
408 &(*Data)->ProcessingPolicy->Mode,\r
409 sizeof (UINT32),\r
410 mMapIpSecMode,\r
411 ParamPackage,\r
412 FORMAT_STRING\r
413 );\r
414 if (!EFI_ERROR (Status)) {\r
415 *Mask |= MODE;\r
416 }\r
417\r
418 if (Status == EFI_INVALID_PARAMETER) {\r
419 ReturnStatus = EFI_INVALID_PARAMETER;\r
420 }\r
421\r
422 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--tunnel-local");\r
423 if (ValueStr != NULL) {\r
424 Status = EfiInetAddr2 ((CHAR16 *) ValueStr, &(*Data)->ProcessingPolicy->TunnelOption->LocalTunnelAddress);\r
425 if (EFI_ERROR (Status)) {\r
426 ShellPrintHiiEx (\r
427 -1,\r
428 -1,\r
429 NULL,\r
430 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
431 mHiiHandle,\r
432 mAppName,\r
433 L"--tunnel-local",\r
434 ValueStr\r
435 );\r
436 ReturnStatus = EFI_INVALID_PARAMETER;\r
437 } else {\r
438 *Mask |= TUNNEL_LOCAL;\r
439 }\r
440 }\r
441\r
442 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--tunnel-remote");\r
443 if (ValueStr != NULL) {\r
444 Status = EfiInetAddr2 ((CHAR16 *) ValueStr, &(*Data)->ProcessingPolicy->TunnelOption->RemoteTunnelAddress);\r
445 if (EFI_ERROR (Status)) {\r
446 ShellPrintHiiEx (\r
447 -1,\r
448 -1,\r
449 NULL,\r
450 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
451 mHiiHandle,\r
452 mAppName,\r
453 L"--tunnel-remote",\r
454 ValueStr\r
455 );\r
456 ReturnStatus = EFI_INVALID_PARAMETER;\r
457 } else {\r
458 *Mask |= TUNNEL_REMOTE;\r
459 }\r
460 }\r
461\r
462 (*Data)->ProcessingPolicy->TunnelOption->DF = EfiIPsecTunnelCopyDf;\r
463 Status = GetNumber (\r
464 L"--dont-fragment",\r
465 0,\r
466 &(*Data)->ProcessingPolicy->TunnelOption->DF,\r
467 sizeof (UINT32),\r
468 mMapDfOption,\r
469 ParamPackage,\r
470 FORMAT_STRING\r
471 );\r
472 if (!EFI_ERROR (Status)) {\r
473 *Mask |= DONT_FRAGMENT;\r
474 }\r
475\r
476 if (Status == EFI_INVALID_PARAMETER) {\r
477 ReturnStatus = EFI_INVALID_PARAMETER;\r
478 }\r
479\r
480 (*Data)->ProcessingPolicy->Proto = EfiIPsecESP;\r
481 Status = GetNumber (\r
482 L"--ipsec-proto",\r
483 0,\r
484 &(*Data)->ProcessingPolicy->Proto,\r
485 sizeof (UINT32),\r
486 mMapIpSecProtocol,\r
487 ParamPackage,\r
488 FORMAT_STRING\r
489 );\r
490 if (!EFI_ERROR (Status)) {\r
491 *Mask |= IPSEC_PROTO;\r
492 }\r
493\r
494 if (Status == EFI_INVALID_PARAMETER) {\r
495 ReturnStatus = EFI_INVALID_PARAMETER;\r
496 }\r
497\r
498 Status = GetNumber (\r
499 L"--encrypt-algo",\r
500 0,\r
501 &(*Data)->ProcessingPolicy->EncAlgoId,\r
502 sizeof (UINT8),\r
503 mMapEncAlgo,\r
504 ParamPackage,\r
505 FORMAT_STRING\r
506 );\r
507 if (!EFI_ERROR (Status)) {\r
508 *Mask |= ENCRYPT_ALGO;\r
509 }\r
510\r
511 if (Status == EFI_INVALID_PARAMETER) {\r
512 ReturnStatus = EFI_INVALID_PARAMETER;\r
513 }\r
514\r
515 Status = GetNumber (\r
516 L"--auth-algo",\r
517 0,\r
518 &(*Data)->ProcessingPolicy->AuthAlgoId,\r
519 sizeof (UINT8),\r
520 mMapAuthAlgo,\r
521 ParamPackage,\r
522 FORMAT_STRING\r
523 );\r
524 if (!EFI_ERROR (Status)) {\r
525 *Mask |= AUTH_ALGO;\r
526 }\r
527\r
528 if (Status == EFI_INVALID_PARAMETER) {\r
529 ReturnStatus = EFI_INVALID_PARAMETER;\r
530 }\r
531\r
532 //\r
533 // Cannot check Mode against EfiIPsecTunnel, because user may want to change tunnel_remote only so the Mode is not set.\r
534 //\r
535 if ((*Mask & (TUNNEL_LOCAL | TUNNEL_REMOTE | DONT_FRAGMENT)) == 0) {\r
536 (*Data)->ProcessingPolicy->TunnelOption = NULL;\r
537 }\r
538\r
539 if ((*Mask & (EXT_SEQUENCE | SEQUENCE_OVERFLOW | FRAGMENT_CHECK | LIFEBYTE |\r
540 LIFETIME_SOFT | LIFETIME | MODE | TUNNEL_LOCAL | TUNNEL_REMOTE |\r
541 DONT_FRAGMENT | IPSEC_PROTO | AUTH_ALGO | ENCRYPT_ALGO)) == 0) {\r
542 if ((*Data)->Action != EfiIPsecActionProtect) {\r
543 //\r
544 // User may not provide additional parameter for Protect action, so we cannot simply set ProcessingPolicy to NULL.\r
545 //\r
546 (*Data)->ProcessingPolicy = NULL;\r
547 }\r
548 }\r
549\r
550 if (CreateNew) {\r
551 if ((*Mask & (LOCAL | REMOTE | PROTO | ACTION)) != (LOCAL | REMOTE | PROTO | ACTION)) {\r
552 ShellPrintHiiEx (\r
553 -1,\r
554 -1,\r
555 NULL,\r
556 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
557 mHiiHandle,\r
558 mAppName,\r
559 L"--local --remote --proto --action"\r
560 );\r
561 ReturnStatus = EFI_INVALID_PARAMETER;\r
562 } else if (((*Data)->Action == EfiIPsecActionProtect) &&\r
563 ((*Data)->ProcessingPolicy->Mode == EfiIPsecTunnel) &&\r
564 ((*Mask & (TUNNEL_LOCAL | TUNNEL_REMOTE)) != (TUNNEL_LOCAL | TUNNEL_REMOTE))) {\r
565 ShellPrintHiiEx (\r
566 -1,\r
567 -1,\r
568 NULL,\r
569 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
570 mHiiHandle,\r
571 mAppName,\r
572 L"--tunnel-local --tunnel-remote"\r
573 );\r
574 ReturnStatus = EFI_INVALID_PARAMETER;\r
575 }\r
576 }\r
577\r
578 return ReturnStatus;\r
579}\r
580\r
581/**\r
64b2d0e5 582 Fill in EFI_IPSEC_SA_ID and EFI_IPSEC_SA_DATA2 through ParamPackage list.\r
a3bcde70
HT		 583\r
584 @param[out] SaId The pointer to the EFI_IPSEC_SA_ID structure.\r
64b2d0e5 585 @param[out] Data The pointer to the EFI_IPSEC_SA_DATA2 structure.\r
a3bcde70
HT		 586 @param[in] ParamPackage The pointer to the ParamPackage list.\r
587 @param[out] Mask The pointer to the Mask.\r
588 @param[in] CreateNew The switch to create new.\r
589\r
64b2d0e5 590 @retval EFI_SUCCESS Fill in EFI_IPSEC_SA_ID and EFI_IPSEC_SA_DATA2 successfully.\r
a3bcde70
HT		 591 @retval EFI_INVALID_PARAMETER Invalid user input parameter.\r
592\r
593**/\r
594EFI_STATUS\r
595CreateSadEntry (\r
596 OUT EFI_IPSEC_SA_ID **SaId,\r
64b2d0e5 597 OUT EFI_IPSEC_SA_DATA2 **Data,\r
a3bcde70
HT		 598 IN LIST_ENTRY *ParamPackage,\r
599 OUT UINT32 *Mask,\r
600 IN BOOLEAN CreateNew\r
601 )\r
602{\r
603 EFI_STATUS Status;\r
604 EFI_STATUS ReturnStatus;\r
605 UINTN AuthKeyLength;\r
606 UINTN EncKeyLength;\r
607 CONST CHAR16 *ValueStr;\r
64b2d0e5 608 CHAR8 *AsciiStr;\r
a3bcde70
HT		 609 UINTN DataSize;\r
610\r
611 Status = EFI_SUCCESS;\r
612 ReturnStatus = EFI_SUCCESS;\r
613 *Mask = 0;\r
614 AuthKeyLength = 0;\r
615 EncKeyLength = 0;\r
616\r
617 *SaId = AllocateZeroPool (sizeof (EFI_IPSEC_SA_ID));\r
618 ASSERT (*SaId != NULL);\r
619\r
620 //\r
621 // Convert user imput from string to integer, and fill in the Spi in EFI_IPSEC_SA_ID.\r
622 //\r
623 Status = GetNumber (L"--spi", (UINT32) -1, &(*SaId)->Spi, sizeof (UINT32), NULL, ParamPackage, FORMAT_NUMBER);\r
624 if (!EFI_ERROR (Status)) {\r
625 *Mask |= SPI;\r
626 }\r
627\r
628 if (Status == EFI_INVALID_PARAMETER) {\r
629 ReturnStatus = EFI_INVALID_PARAMETER;\r
630 }\r
631\r
632 //\r
633 // Convert user imput from string to integer, and fill in the Proto in EFI_IPSEC_SA_ID.\r
634 //\r
635 Status = GetNumber (\r
636 L"--ipsec-proto",\r
637 0,\r
638 &(*SaId)->Proto,\r
639 sizeof (EFI_IPSEC_PROTOCOL_TYPE),\r
640 mMapIpSecProtocol,\r
641 ParamPackage,\r
642 FORMAT_STRING\r
643 );\r
644 if (!EFI_ERROR (Status)) {\r
645 *Mask |= IPSEC_PROTO;\r
646 }\r
647\r
648 if (Status == EFI_INVALID_PARAMETER) {\r
649 ReturnStatus = EFI_INVALID_PARAMETER;\r
650 }\r
651\r
652 //\r
64b2d0e5 653 // Convert user imput from string to integer, and fill in EFI_IPSEC_SA_DATA2.\r
a3bcde70
HT		 654 //\r
655 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--auth-key");\r
656 if (ValueStr != NULL) {\r
64b2d0e5 657 AuthKeyLength = StrLen (ValueStr);\r
a3bcde70
HT		 658 }\r
659\r
660 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--encrypt-key");\r
661 if (ValueStr != NULL) {\r
64b2d0e5 662 EncKeyLength = StrLen (ValueStr);\r
a3bcde70
HT		 663 }\r
664\r
665 //\r
64b2d0e5 666 // EFI_IPSEC_SA_DATA2:\r
a3bcde70 667 // +------------\r
64b2d0e5 668 // | EFI_IPSEC_SA_DATA2\r
a3bcde70
HT		 669 // +-----------------------\r
670 // | AuthKey\r
671 // +-------------------------\r
672 // | EncKey\r
673 // +-------------------------\r
674 // | SpdSelector\r
675 //\r
676 // Notes: To make sure the address alignment add padding after each data if needed.\r
677 //\r
64b2d0e5 678 DataSize = ALIGN_VARIABLE (sizeof (EFI_IPSEC_SA_DATA2));\r
a3bcde70
HT		 679 DataSize = ALIGN_VARIABLE (DataSize + AuthKeyLength);\r
680 DataSize = ALIGN_VARIABLE (DataSize + EncKeyLength);\r
681 DataSize = ALIGN_VARIABLE (DataSize + sizeof (EFI_IPSEC_SPD_SELECTOR));\r
682 DataSize = ALIGN_VARIABLE (DataSize + sizeof (EFI_IP_ADDRESS_INFO));\r
683 DataSize += sizeof (EFI_IP_ADDRESS_INFO);\r
684\r
685\r
686\r
687 *Data = AllocateZeroPool (DataSize);\r
688 ASSERT (*Data != NULL);\r
689\r
690 (*Data)->ManualSet = TRUE;\r
691 (*Data)->AlgoInfo.EspAlgoInfo.AuthKey = (VOID *) ALIGN_POINTER (((*Data) + 1), sizeof (UINTN));\r
692 (*Data)->AlgoInfo.EspAlgoInfo.EncKey = (VOID *) ALIGN_POINTER (\r
693 ((UINT8 *) (*Data)->AlgoInfo.EspAlgoInfo.AuthKey + AuthKeyLength),\r
694 sizeof (UINTN)\r
695 );\r
696 (*Data)->SpdSelector = (EFI_IPSEC_SPD_SELECTOR *) ALIGN_POINTER (\r
697 ((UINT8 *) (*Data)->AlgoInfo.EspAlgoInfo.EncKey + EncKeyLength),\r
698 sizeof (UINTN)\r
699 );\r
700 (*Data)->SpdSelector->LocalAddress = (EFI_IP_ADDRESS_INFO *) ALIGN_POINTER (\r
701 ((UINT8 *) (*Data)->SpdSelector + sizeof (EFI_IPSEC_SPD_SELECTOR)),\r
702 sizeof (UINTN));\r
703 (*Data)->SpdSelector->RemoteAddress = (EFI_IP_ADDRESS_INFO *) ALIGN_POINTER (\r
704 (*Data)->SpdSelector->LocalAddress + 1,\r
705 sizeof (UINTN)\r
706 );\r
707\r
708 (*Data)->Mode = EfiIPsecTransport;\r
709 Status = GetNumber (\r
710 L"--mode",\r
711 0,\r
712 &(*Data)->Mode,\r
713 sizeof (EFI_IPSEC_MODE),\r
714 mMapIpSecMode,\r
715 ParamPackage,\r
716 FORMAT_STRING\r
717 );\r
718 if (!EFI_ERROR (Status)) {\r
719 *Mask |= MODE;\r
720 }\r
721\r
722 if (Status == EFI_INVALID_PARAMETER) {\r
723 ReturnStatus = EFI_INVALID_PARAMETER;\r
724 }\r
725\r
726 //\r
727 // According to RFC 4303-3.3.3. The first packet sent using a given SA\r
728 // will contain a sequence number of 1.\r
729 //\r
730 (*Data)->SNCount = 1;\r
731 Status = GetNumber (\r
732 L"--sequence-number",\r
733 (UINT64) -1,\r
734 &(*Data)->SNCount,\r
735 sizeof (UINT64),\r
736 NULL,\r
737 ParamPackage,\r
738 FORMAT_NUMBER\r
739 );\r
740 if (!EFI_ERROR (Status)) {\r
741 *Mask |= SEQUENCE_NUMBER;\r
742 }\r
743\r
744 if (Status == EFI_INVALID_PARAMETER) {\r
745 ReturnStatus = EFI_INVALID_PARAMETER;\r
746 }\r
747\r
748 (*Data)->AntiReplayWindows = 0;\r
749 Status = GetNumber (\r
750 L"--antireplay-window",\r
751 (UINT8) -1,\r
752 &(*Data)->AntiReplayWindows,\r
753 sizeof (UINT8),\r
754 NULL,\r
755 ParamPackage,\r
756 FORMAT_NUMBER\r
757 );\r
758 if (!EFI_ERROR (Status)) {\r
759 *Mask |= SEQUENCE_NUMBER;\r
760 }\r
761\r
762 if (Status == EFI_INVALID_PARAMETER) {\r
763 ReturnStatus = EFI_INVALID_PARAMETER;\r
764 }\r
765\r
766 Status = GetNumber (\r
767 L"--encrypt-algo",\r
768 0,\r
769 &(*Data)->AlgoInfo.EspAlgoInfo.EncAlgoId,\r
770 sizeof (UINT8),\r
771 mMapEncAlgo,\r
772 ParamPackage,\r
773 FORMAT_STRING\r
774 );\r
775 if (!EFI_ERROR (Status)) {\r
776 *Mask |= ENCRYPT_ALGO;\r
777 }\r
778\r
779 if (Status == EFI_INVALID_PARAMETER) {\r
780 ReturnStatus = EFI_INVALID_PARAMETER;\r
781 }\r
782\r
783 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--encrypt-key");\r
784 if (ValueStr != NULL ) {\r
785 (*Data)->AlgoInfo.EspAlgoInfo.EncKeyLength = EncKeyLength;\r
64b2d0e5 786 AsciiStr = AllocateZeroPool (EncKeyLength + 1);\r
787 UnicodeStrToAsciiStr (ValueStr, AsciiStr);\r
788 CopyMem ((*Data)->AlgoInfo.EspAlgoInfo.EncKey, AsciiStr, EncKeyLength);\r
789 FreePool (AsciiStr);\r
a3bcde70
HT		 790 *Mask |= ENCRYPT_KEY;\r
791 } else {\r
792 (*Data)->AlgoInfo.EspAlgoInfo.EncKey = NULL;\r
793 }\r
794\r
795 Status = GetNumber (\r
796 L"--auth-algo",\r
797 0,\r
798 &(*Data)->AlgoInfo.EspAlgoInfo.AuthAlgoId,\r
799 sizeof (UINT8),\r
800 mMapAuthAlgo,\r
801 ParamPackage,\r
802 FORMAT_STRING\r
803 );\r
804 if (!EFI_ERROR (Status)) {\r
805 *Mask |= AUTH_ALGO;\r
806 }\r
807\r
808 if (Status == EFI_INVALID_PARAMETER) {\r
809 ReturnStatus = EFI_INVALID_PARAMETER;\r
810 }\r
811\r
812 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--auth-key");\r
813 if (ValueStr != NULL) {\r
814 (*Data)->AlgoInfo.EspAlgoInfo.AuthKeyLength = AuthKeyLength;\r
64b2d0e5 815 AsciiStr = AllocateZeroPool (AuthKeyLength + 1);\r
816 UnicodeStrToAsciiStr (ValueStr, AsciiStr);\r
817 CopyMem ((*Data)->AlgoInfo.EspAlgoInfo.AuthKey, AsciiStr, AuthKeyLength);\r
818 FreePool (AsciiStr);\r
a3bcde70
HT		 819 *Mask |= AUTH_KEY;\r
820 } else {\r
821 (*Data)->AlgoInfo.EspAlgoInfo.AuthKey = NULL;\r
822 }\r
823\r
824 Status = GetNumber (\r
825 L"--lifebyte",\r
826 (UINT64) -1,\r
827 &(*Data)->SaLifetime.ByteCount,\r
828 sizeof (UINT64),\r
829 NULL,\r
830 ParamPackage,\r
831 FORMAT_NUMBER\r
832 );\r
833 if (!EFI_ERROR (Status)) {\r
834 *Mask |= LIFEBYTE;\r
835 }\r
836\r
837 if (Status == EFI_INVALID_PARAMETER) {\r
838 ReturnStatus = EFI_INVALID_PARAMETER;\r
839 }\r
840\r
841 Status = GetNumber (\r
842 L"--lifetime",\r
843 (UINT64) -1,\r
844 &(*Data)->SaLifetime.HardLifetime,\r
845 sizeof (UINT64),\r
846 NULL,\r
847 ParamPackage,\r
848 FORMAT_NUMBER\r
849 );\r
850 if (!EFI_ERROR (Status)) {\r
851 *Mask |= LIFETIME;\r
852 }\r
853\r
854 if (Status == EFI_INVALID_PARAMETER) {\r
855 ReturnStatus = EFI_INVALID_PARAMETER;\r
856 }\r
857\r
858 Status = GetNumber (\r
859 L"--lifetime-soft",\r
860 (UINT64) -1,\r
861 &(*Data)->SaLifetime.SoftLifetime,\r
862 sizeof (UINT64),\r
863 NULL,\r
864 ParamPackage,\r
865 FORMAT_NUMBER\r
866 );\r
867 if (!EFI_ERROR (Status)) {\r
868 *Mask |= LIFETIME_SOFT;\r
869 }\r
870\r
871 if (Status == EFI_INVALID_PARAMETER) {\r
872 ReturnStatus = EFI_INVALID_PARAMETER;\r
873 }\r
874\r
875 Status = GetNumber (\r
876 L"--path-mtu",\r
877 (UINT32) -1,\r
878 &(*Data)->PathMTU,\r
879 sizeof (UINT32),\r
880 NULL,\r
881 ParamPackage,\r
882 FORMAT_NUMBER\r
883 );\r
884 if (!EFI_ERROR (Status)) {\r
885 *Mask |= PATH_MTU;\r
886 }\r
887\r
888 if (Status == EFI_INVALID_PARAMETER) {\r
889 ReturnStatus = EFI_INVALID_PARAMETER;\r
890 }\r
891\r
64b2d0e5 892 //\r
893 // Convert user imput from string to integer, and fill in the DestAddress in EFI_IPSEC_SA_ID.\r
894 //\r
895 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--tunnel-dest");\r
896 if (ValueStr != NULL) {\r
897 Status = EfiInetAddr2 ((CHAR16 *) ValueStr, &(*Data)->TunnelDestinationAddress);\r
898 if (EFI_ERROR (Status)) {\r
899 ShellPrintHiiEx (\r
900 -1,\r
901 -1,\r
902 NULL,\r
903 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
904 mHiiHandle,\r
905 mAppName,\r
906 L"--tunnel-dest",\r
907 ValueStr\r
908 );\r
909 ReturnStatus = EFI_INVALID_PARAMETER;\r
910 } else {\r
911 *Mask |= DEST;\r
912 }\r
913 }\r
914\r
915 //\r
da7c529c 916 // Convert user input from string to integer, and fill in the DestAddress in EFI_IPSEC_SA_ID.\r
64b2d0e5 917 //\r
918 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--tunnel-source");\r
919 if (ValueStr != NULL) {\r
920 Status = EfiInetAddr2 ((CHAR16 *) ValueStr, &(*Data)->TunnelSourceAddress);\r
921 if (EFI_ERROR (Status)) {\r
922 ShellPrintHiiEx (\r
923 -1,\r
924 -1,\r
925 NULL,\r
926 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
927 mHiiHandle,\r
928 mAppName,\r
929 L"--tunnel-source",\r
930 ValueStr\r
931 );\r
932 ReturnStatus = EFI_INVALID_PARAMETER;\r
933 } else {\r
934 *Mask |= SOURCE;\r
935 }\r
936 }\r
da7c529c 937\r
938 //\r
939 // If it is TunnelMode, then check if the tunnel-source and --tunnel-dest are set\r
940 //\r
941 if ((*Data)->Mode == EfiIPsecTunnel) {\r
942 if ((*Mask & (DEST|SOURCE)) != (DEST|SOURCE)) {\r
943 ShellPrintHiiEx (\r
944 -1,\r
945 -1,\r
946 NULL,\r
947 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
948 mHiiHandle,\r
949 mAppName,\r
950 L"--tunnel-source --tunnel-dest"\r
951 );\r
952 ReturnStatus = EFI_INVALID_PARAMETER;\r
953 }\r
954 }\r
a3bcde70
HT		 955 ReturnStatus = CreateSpdSelector ((*Data)->SpdSelector, ParamPackage, Mask);\r
956\r
957 if (CreateNew) {\r
da7c529c 958 if ((*Mask & (SPI|IPSEC_PROTO|LOCAL|REMOTE)) != (SPI|IPSEC_PROTO|LOCAL|REMOTE)) {\r
a3bcde70
HT		 959 ShellPrintHiiEx (\r
960 -1,\r
961 -1,\r
962 NULL,\r
963 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
964 mHiiHandle,\r
965 mAppName,\r
da7c529c 966 L"--spi --ipsec-proto --local --remote"\r
a3bcde70
HT		 967 );\r
968 ReturnStatus = EFI_INVALID_PARAMETER;\r
969 } else {\r
970 if ((*SaId)->Proto == EfiIPsecAH) {\r
971 if ((*Mask & AUTH_ALGO) == 0) {\r
972 ShellPrintHiiEx (\r
973 -1,\r
974 -1,\r
975 NULL,\r
976 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_PARAMETER),\r
977 mHiiHandle,\r
978 mAppName,\r
979 L"--auth-algo"\r
980 );\r
981 ReturnStatus = EFI_INVALID_PARAMETER;\r
780847d1 982 } else if ((*Data)->AlgoInfo.EspAlgoInfo.AuthAlgoId != IPSEC_AALG_NONE && (*Mask & AUTH_KEY) == 0) {\r
a3bcde70
HT		 983 ShellPrintHiiEx (\r
984 -1,\r
985 -1,\r
986 NULL,\r
987 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_PARAMETER),\r
988 mHiiHandle,\r
989 mAppName,\r
990 L"--auth-key"\r
991 );\r
992 ReturnStatus = EFI_INVALID_PARAMETER;\r
993 }\r
994 } else {\r
da7c529c 995 if ((*Mask & (ENCRYPT_ALGO|AUTH_ALGO)) != (ENCRYPT_ALGO|AUTH_ALGO) ) {\r
a3bcde70
HT		 996 ShellPrintHiiEx (\r
997 -1,\r
998 -1,\r
999 NULL,\r
1000 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_PARAMETER),\r
1001 mHiiHandle,\r
1002 mAppName,\r
da7c529c 1003 L"--encrypt-algo --auth-algo"\r
a3bcde70
HT		 1004 );\r
1005 ReturnStatus = EFI_INVALID_PARAMETER;\r
780847d1 1006 } else if ((*Data)->AlgoInfo.EspAlgoInfo.EncAlgoId != IPSEC_EALG_NONE && (*Mask & ENCRYPT_KEY) == 0) {\r
a3bcde70
HT		 1007 ShellPrintHiiEx (\r
1008 -1,\r
1009 -1,\r
1010 NULL,\r
1011 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_PARAMETER),\r
1012 mHiiHandle,\r
1013 mAppName,\r
1014 L"--encrypt-key"\r
1015 );\r
1016 ReturnStatus = EFI_INVALID_PARAMETER;\r
da7c529c 1017 } else if ((*Data)->AlgoInfo.EspAlgoInfo.AuthAlgoId != IPSEC_AALG_NONE && (*Mask & AUTH_KEY) == 0) {\r
1018 ShellPrintHiiEx (\r
1019 -1,\r
1020 -1,\r
1021 NULL,\r
1022 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_PARAMETER),\r
1023 mHiiHandle,\r
1024 mAppName,\r
1025 L"--auth-key"\r
1026 );\r
1027 ReturnStatus = EFI_INVALID_PARAMETER;\r
a3bcde70
HT		 1028 }\r
1029 }\r
1030 }\r
1031 }\r
1032\r
1033 return ReturnStatus;\r
1034}\r
1035\r
1036/**\r
1037 Fill in EFI_IPSEC_PAD_ID and EFI_IPSEC_PAD_DATA through ParamPackage list.\r
1038\r
1039 @param[out] PadId The pointer to the EFI_IPSEC_PAD_ID structure.\r
1040 @param[out] Data The pointer to the EFI_IPSEC_PAD_DATA structure.\r
1041 @param[in] ParamPackage The pointer to the ParamPackage list.\r
1042 @param[out] Mask The pointer to the Mask.\r
1043 @param[in] CreateNew The switch to create new.\r
1044\r
1045 @retval EFI_SUCCESS Fill in EFI_IPSEC_PAD_ID and EFI_IPSEC_PAD_DATA successfully.\r
1046 @retval EFI_INVALID_PARAMETER Invalid user input parameter.\r
1047\r
1048**/\r
1049EFI_STATUS\r
1050CreatePadEntry (\r
1051 OUT EFI_IPSEC_PAD_ID **PadId,\r
1052 OUT EFI_IPSEC_PAD_DATA **Data,\r
1053 IN LIST_ENTRY *ParamPackage,\r
1054 OUT UINT32 *Mask,\r
1055 IN BOOLEAN CreateNew\r
1056 )\r
1057{\r
1058 EFI_STATUS Status;\r
1059 EFI_STATUS ReturnStatus;\r
780847d1 1060 SHELL_FILE_HANDLE FileHandle;\r
a3bcde70
HT		 1061 UINT64 FileSize;\r
1062 UINTN AuthDataLength;\r
1063 UINTN RevocationDataLength;\r
1064 UINTN DataLength;\r
1065 UINTN Index;\r
1066 CONST CHAR16 *ValueStr;\r
1067 UINTN DataSize;\r
1068\r
1069 Status = EFI_SUCCESS;\r
1070 ReturnStatus = EFI_SUCCESS;\r
1071 *Mask = 0;\r
1072 AuthDataLength = 0;\r
1073 RevocationDataLength = 0;\r
1074\r
1075 *PadId = AllocateZeroPool (sizeof (EFI_IPSEC_PAD_ID));\r
1076 ASSERT (*PadId != NULL);\r
1077\r
1078 //\r
1079 // Convert user imput from string to integer, and fill in EFI_IPSEC_PAD_ID.\r
1080 //\r
1081 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--peer-address");\r
1082 if (ValueStr != NULL) {\r
1083 (*PadId)->PeerIdValid = FALSE;\r
1084 Status = EfiInetAddrRange ((CHAR16 *) ValueStr, &(*PadId)->Id.IpAddress);\r
1085 if (EFI_ERROR (Status)) {\r
1086 ShellPrintHiiEx (\r
1087 -1,\r
1088 -1,\r
1089 NULL,\r
1090 STRING_TOKEN (STR_IPSEC_CONFIG_INCORRECT_PARAMETER_VALUE),\r
1091 mHiiHandle,\r
1092 mAppName,\r
1093 L"--peer-address",\r
1094 ValueStr\r
1095 );\r
1096 ReturnStatus = EFI_INVALID_PARAMETER;\r
1097 } else {\r
1098 *Mask |= PEER_ADDRESS;\r
1099 }\r
1100 }\r
1101\r
1102 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--peer-id");\r
1103 if (ValueStr != NULL) {\r
1104 (*PadId)->PeerIdValid = TRUE;\r
1105 StrnCpy ((CHAR16 *) (*PadId)->Id.PeerId, ValueStr, ARRAY_SIZE ((*PadId)->Id.PeerId) - 1);\r
1106 *Mask |= PEER_ID;\r
1107 }\r
1108\r
1109 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--auth-data");\r
1110 if (ValueStr != NULL) {\r
1111 if (ValueStr[0] == L'@') {\r
1112 //\r
1113 // Input is a file: --auth-data "@fs1:\My Certificates\tom.dat"\r
1114 //\r
1115 Status = ShellOpenFileByName (&ValueStr[1], &FileHandle, EFI_FILE_MODE_READ, 0);\r
1116 if (EFI_ERROR (Status)) {\r
1117 ShellPrintHiiEx (\r
1118 -1,\r
1119 -1,\r
1120 NULL,\r
1121 STRING_TOKEN (STR_IPSEC_CONFIG_FILE_OPEN_FAILED),\r
1122 mHiiHandle,\r
1123 mAppName,\r
1124 &ValueStr[1]\r
1125 );\r
1126 ReturnStatus = EFI_INVALID_PARAMETER;\r
1127 } else {\r
1128 Status = ShellGetFileSize (FileHandle, &FileSize);\r
1129 ShellCloseFile (&FileHandle);\r
1130 if (EFI_ERROR (Status)) {\r
1131 ShellPrintHiiEx (\r
1132 -1,\r
1133 -1,\r
1134 NULL,\r
1135 STRING_TOKEN (STR_IPSEC_CONFIG_FILE_OPEN_FAILED),\r
1136 mHiiHandle,\r
1137 mAppName,\r
1138 &ValueStr[1]\r
1139 );\r
1140 ReturnStatus = EFI_INVALID_PARAMETER;\r
1141 } else {\r
1142 AuthDataLength = (UINTN) FileSize;\r
1143 }\r
1144 }\r
1145 } else {\r
1146 AuthDataLength = StrLen (ValueStr);\r
1147 }\r
1148 }\r
1149\r
1150 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--revocation-data");\r
1151 if (ValueStr != NULL) {\r
1152 RevocationDataLength = (StrLen (ValueStr) + 1) * sizeof (CHAR16);\r
1153 }\r
1154\r
1155 //\r
1156 // Allocate Buffer for Data. Add padding after each struct to make sure the alignment\r
1157 // in different Arch.\r
1158 //\r
1159 DataSize = ALIGN_VARIABLE (sizeof (EFI_IPSEC_PAD_DATA));\r
1160 DataSize = ALIGN_VARIABLE (DataSize + AuthDataLength);\r
1161 DataSize += RevocationDataLength;\r
1162\r
1163 *Data = AllocateZeroPool (DataSize);\r
1164 ASSERT (*Data != NULL);\r
1165\r
1166 (*Data)->AuthData = (VOID *) ALIGN_POINTER ((*Data + 1), sizeof (UINTN));\r
1167 (*Data)->RevocationData = (VOID *) ALIGN_POINTER (((UINT8 *) (*Data + 1) + AuthDataLength), sizeof (UINTN));\r
1168 (*Data)->AuthProtocol = EfiIPsecAuthProtocolIKEv1;\r
1169\r
1170 //\r
1171 // Convert user imput from string to integer, and fill in EFI_IPSEC_PAD_DATA.\r
1172 //\r
1173 Status = GetNumber (\r
1174 L"--auth-proto",\r
1175 0,\r
1176 &(*Data)->AuthProtocol,\r
1177 sizeof (EFI_IPSEC_AUTH_PROTOCOL_TYPE),\r
1178 mMapAuthProto,\r
1179 ParamPackage,\r
1180 FORMAT_STRING\r
1181 );\r
1182 if (!EFI_ERROR (Status)) {\r
1183 *Mask |= AUTH_PROTO;\r
1184 }\r
1185\r
1186 if (Status == EFI_INVALID_PARAMETER) {\r
1187 ReturnStatus = EFI_INVALID_PARAMETER;\r
1188 }\r
1189\r
1190 Status = GetNumber (\r
1191 L"--auth-method",\r
1192 0,\r
1193 &(*Data)->AuthMethod,\r
1194 sizeof (EFI_IPSEC_AUTH_METHOD),\r
1195 mMapAuthMethod,\r
1196 ParamPackage,\r
1197 FORMAT_STRING\r
1198 );\r
1199 if (!EFI_ERROR (Status)) {\r
1200 *Mask |= AUTH_METHOD;\r
1201 }\r
1202\r
1203 if (Status == EFI_INVALID_PARAMETER) {\r
1204 ReturnStatus = EFI_INVALID_PARAMETER;\r
1205 }\r
1206\r
1207 if (ShellCommandLineGetFlag (ParamPackage, L"--ike-id")) {\r
1208 (*Data)->IkeIdFlag = TRUE;\r
1209 *Mask |= IKE_ID;\r
1210 }\r
1211\r
1212 if (ShellCommandLineGetFlag (ParamPackage, L"--ike-id-")) {\r
1213 (*Data)->IkeIdFlag = FALSE;\r
1214 *Mask |= IKE_ID;\r
1215 }\r
1216\r
1217 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--auth-data");\r
1218 if (ValueStr != NULL) {\r
1219 if (ValueStr[0] == L'@') {\r
1220 //\r
1221 // Input is a file: --auth-data "@fs1:\My Certificates\tom.dat"\r
1222 //\r
1223\r
1224 Status = ShellOpenFileByName (&ValueStr[1], &FileHandle, EFI_FILE_MODE_READ, 0);\r
1225 if (EFI_ERROR (Status)) {\r
1226 ShellPrintHiiEx (\r
1227 -1,\r
1228 -1,\r
1229 NULL,\r
1230 STRING_TOKEN (STR_IPSEC_CONFIG_FILE_OPEN_FAILED),\r
1231 mHiiHandle,\r
1232 mAppName,\r
1233 &ValueStr[1]\r
1234 );\r
1235 ReturnStatus = EFI_INVALID_PARAMETER;\r
1236 (*Data)->AuthData = NULL;\r
1237 } else {\r
1238 DataLength = AuthDataLength;\r
64b2d0e5 1239 Status = ShellReadFile (FileHandle, &DataLength, (*Data)->AuthData);\r
a3bcde70
HT		 1240 ShellCloseFile (&FileHandle);\r
1241 if (EFI_ERROR (Status)) {\r
1242 ShellPrintHiiEx (\r
1243 -1,\r
1244 -1,\r
1245 NULL,\r
1246 STRING_TOKEN (STR_IPSEC_CONFIG_FILE_OPEN_FAILED),\r
1247 mHiiHandle,\r
1248 mAppName,\r
1249 &ValueStr[1]\r
1250 );\r
1251 ReturnStatus = EFI_INVALID_PARAMETER;\r
1252 (*Data)->AuthData = NULL;\r
1253 } else {\r
1254 ASSERT (DataLength == AuthDataLength);\r
1255 *Mask |= AUTH_DATA;\r
1256 }\r
1257 }\r
1258 } else {\r
1259 for (Index = 0; Index < AuthDataLength; Index++) {\r
1260 ((CHAR8 *) (*Data)->AuthData)[Index] = (CHAR8) ValueStr[Index];\r
1261 }\r
1262 (*Data)->AuthDataSize = AuthDataLength;\r
1263 *Mask |= AUTH_DATA;\r
1264 }\r
1265 }\r
1266\r
1267 ValueStr = ShellCommandLineGetValue (ParamPackage, L"--revocation-data");\r
1268 if (ValueStr != NULL) {\r
1269 CopyMem ((*Data)->RevocationData, ValueStr, RevocationDataLength);\r
1270 (*Data)->RevocationDataSize = RevocationDataLength;\r
1271 *Mask |= REVOCATION_DATA;\r
1272 } else {\r
1273 (*Data)->RevocationData = NULL;\r
1274 }\r
1275\r
1276 if (CreateNew) {\r
1277 if ((*Mask & (PEER_ID | PEER_ADDRESS)) == 0) {\r
1278 ShellPrintHiiEx (\r
1279 -1,\r
1280 -1,\r
1281 NULL,\r
1282 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
1283 mHiiHandle,\r
1284 mAppName,\r
1285 L"--peer-id --peer-address"\r
1286 );\r
1287 ReturnStatus = EFI_INVALID_PARAMETER;\r
1288 } else if ((*Mask & (AUTH_METHOD | AUTH_DATA)) != (AUTH_METHOD | AUTH_DATA)) {\r
1289 ShellPrintHiiEx (\r
1290 -1,\r
1291 -1,\r
1292 NULL,\r
1293 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
1294 mHiiHandle,\r
1295 mAppName,\r
1296 L"--auth-method --auth-data"\r
1297 );\r
1298 ReturnStatus = EFI_INVALID_PARAMETER;\r
1299 }\r
1300 }\r
1301\r
1302 return ReturnStatus;\r
1303}\r
1304\r
1305CREATE_POLICY_ENTRY mCreatePolicyEntry[] = {\r
1306 (CREATE_POLICY_ENTRY) CreateSpdEntry,\r
1307 (CREATE_POLICY_ENTRY) CreateSadEntry,\r
1308 (CREATE_POLICY_ENTRY) CreatePadEntry\r
1309};\r
1310\r
1311/**\r
1312 Combine old SPD entry with new SPD entry.\r
1313\r
1314 @param[in, out] OldSelector The pointer to the EFI_IPSEC_SPD_SELECTOR structure.\r
1315 @param[in, out] OldData The pointer to the EFI_IPSEC_SPD_DATA structure.\r
1316 @param[in] NewSelector The pointer to the EFI_IPSEC_SPD_SELECTOR structure.\r
1317 @param[in] NewData The pointer to the EFI_IPSEC_SPD_DATA structure.\r
1318 @param[in] Mask The pointer to the Mask.\r
1319 @param[out] CreateNew The switch to create new.\r
1320\r
1321 @retval EFI_SUCCESS Combined successfully.\r
1322 @retval EFI_INVALID_PARAMETER Invalid user input parameter.\r
1323\r
1324**/\r
1325EFI_STATUS\r
1326CombineSpdEntry (\r
1327 IN OUT EFI_IPSEC_SPD_SELECTOR *OldSelector,\r
1328 IN OUT EFI_IPSEC_SPD_DATA *OldData,\r
1329 IN EFI_IPSEC_SPD_SELECTOR *NewSelector,\r
1330 IN EFI_IPSEC_SPD_DATA *NewData,\r
1331 IN UINT32 Mask,\r
1332 OUT BOOLEAN *CreateNew\r
1333 )\r
1334{\r
1335\r
1336 //\r
1337 // Process Selector\r
1338 //\r
1339 *CreateNew = FALSE;\r
1340 if ((Mask & LOCAL) == 0) {\r
1341 NewSelector->LocalAddressCount = OldSelector->LocalAddressCount;\r
1342 NewSelector->LocalAddress = OldSelector->LocalAddress;\r
1343 } else if ((NewSelector->LocalAddressCount != OldSelector->LocalAddressCount) ||\r
1344 (CompareMem (NewSelector->LocalAddress, OldSelector->LocalAddress, NewSelector->LocalAddressCount * sizeof (EFI_IP_ADDRESS_INFO)) != 0)) {\r
1345 *CreateNew = TRUE;\r
1346 }\r
1347\r
1348 if ((Mask & REMOTE) == 0) {\r
1349 NewSelector->RemoteAddressCount = OldSelector->RemoteAddressCount;\r
1350 NewSelector->RemoteAddress = OldSelector->RemoteAddress;\r
1351 } else if ((NewSelector->RemoteAddressCount != OldSelector->RemoteAddressCount) ||\r
1352 (CompareMem (NewSelector->RemoteAddress, OldSelector->RemoteAddress, NewSelector->RemoteAddressCount * sizeof (EFI_IP_ADDRESS_INFO)) != 0)) {\r
1353 *CreateNew = TRUE;\r
1354 }\r
1355\r
1356 if ((Mask & PROTO) == 0) {\r
1357 NewSelector->NextLayerProtocol = OldSelector->NextLayerProtocol;\r
1358 } else if (NewSelector->NextLayerProtocol != OldSelector->NextLayerProtocol) {\r
1359 *CreateNew = TRUE;\r
1360 }\r
1361\r
1362 switch (NewSelector->NextLayerProtocol) {\r
1363 case EFI_IP4_PROTO_TCP:\r
1364 case EFI_IP4_PROTO_UDP:\r
1365 if ((Mask & LOCAL_PORT) == 0) {\r
1366 NewSelector->LocalPort = OldSelector->LocalPort;\r
1367 NewSelector->LocalPortRange = OldSelector->LocalPortRange;\r
1368 } else if ((NewSelector->LocalPort != OldSelector->LocalPort) ||\r
1369 (NewSelector->LocalPortRange != OldSelector->LocalPortRange)) {\r
1370 *CreateNew = TRUE;\r
1371 }\r
1372\r
1373 if ((Mask & REMOTE_PORT) == 0) {\r
1374 NewSelector->RemotePort = OldSelector->RemotePort;\r
1375 NewSelector->RemotePortRange = OldSelector->RemotePortRange;\r
1376 } else if ((NewSelector->RemotePort != OldSelector->RemotePort) ||\r
1377 (NewSelector->RemotePortRange != OldSelector->RemotePortRange)) {\r
1378 *CreateNew = TRUE;\r
1379 }\r
1380 break;\r
1381\r
1382 case EFI_IP4_PROTO_ICMP:\r
1383 if ((Mask & ICMP_TYPE) == 0) {\r
1384 NewSelector->LocalPort = OldSelector->LocalPort;\r
1385 } else if (NewSelector->LocalPort != OldSelector->LocalPort) {\r
1386 *CreateNew = TRUE;\r
1387 }\r
1388\r
1389 if ((Mask & ICMP_CODE) == 0) {\r
1390 NewSelector->RemotePort = OldSelector->RemotePort;\r
1391 } else if (NewSelector->RemotePort != OldSelector->RemotePort) {\r
1392 *CreateNew = TRUE;\r
1393 }\r
1394 break;\r
1395 }\r
1396 //\r
1397 // Process Data\r
1398 //\r
1399 if ((Mask & NAME) != 0) {\r
1400 AsciiStrCpy ((CHAR8 *) OldData->Name, (CHAR8 *) NewData->Name);\r
1401 }\r
1402\r
1403 if ((Mask & PACKET_FLAG) != 0) {\r
1404 OldData->PackageFlag = NewData->PackageFlag;\r
1405 }\r
1406\r
1407 if ((Mask & ACTION) != 0) {\r
1408 OldData->Action = NewData->Action;\r
1409 }\r
1410\r
1411 if (OldData->Action != EfiIPsecActionProtect) {\r
1412 OldData->ProcessingPolicy = NULL;\r
1413 } else {\r
1414 //\r
1415 // Protect\r
1416 //\r
1417 if (OldData->ProcessingPolicy == NULL) {\r
1418 //\r
1419 // Just point to new data if originally NULL.\r
1420 //\r
1421 OldData->ProcessingPolicy = NewData->ProcessingPolicy;\r
1422 if (OldData->ProcessingPolicy->Mode == EfiIPsecTunnel &&\r
1423 (Mask & (TUNNEL_LOCAL | TUNNEL_REMOTE)) != (TUNNEL_LOCAL | TUNNEL_REMOTE)\r
1424 ) {\r
1425 //\r
1426 // Change to Protect action and Tunnel mode, but without providing local/remote tunnel address.\r
1427 //\r
1428 ShellPrintHiiEx (\r
1429 -1,\r
1430 -1,\r
1431 NULL,\r
1432 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
1433 mHiiHandle,\r
1434 mAppName,\r
1435 L"--tunnel-local --tunnel-remote"\r
1436 );\r
1437 return EFI_INVALID_PARAMETER;\r
1438 }\r
1439 } else {\r
1440 //\r
1441 // Modify some of the data.\r
1442 //\r
1443 if ((Mask & EXT_SEQUENCE) != 0) {\r
1444 OldData->ProcessingPolicy->ExtSeqNum = NewData->ProcessingPolicy->ExtSeqNum;\r
1445 }\r
1446\r
1447 if ((Mask & SEQUENCE_OVERFLOW) != 0) {\r
1448 OldData->ProcessingPolicy->SeqOverflow = NewData->ProcessingPolicy->SeqOverflow;\r
1449 }\r
1450\r
1451 if ((Mask & FRAGMENT_CHECK) != 0) {\r
1452 OldData->ProcessingPolicy->FragCheck = NewData->ProcessingPolicy->FragCheck;\r
1453 }\r
1454\r
1455 if ((Mask & LIFEBYTE) != 0) {\r
1456 OldData->ProcessingPolicy->SaLifetime.ByteCount = NewData->ProcessingPolicy->SaLifetime.ByteCount;\r
1457 }\r
1458\r
1459 if ((Mask & LIFETIME_SOFT) != 0) {\r
1460 OldData->ProcessingPolicy->SaLifetime.SoftLifetime = NewData->ProcessingPolicy->SaLifetime.SoftLifetime;\r
1461 }\r
1462\r
1463 if ((Mask & LIFETIME) != 0) {\r
1464 OldData->ProcessingPolicy->SaLifetime.HardLifetime = NewData->ProcessingPolicy->SaLifetime.HardLifetime;\r
1465 }\r
1466\r
1467 if ((Mask & MODE) != 0) {\r
1468 OldData->ProcessingPolicy->Mode = NewData->ProcessingPolicy->Mode;\r
1469 }\r
1470\r
1471 if ((Mask & IPSEC_PROTO) != 0) {\r
1472 OldData->ProcessingPolicy->Proto = NewData->ProcessingPolicy->Proto;\r
1473 }\r
1474\r
1475 if ((Mask & AUTH_ALGO) != 0) {\r
1476 OldData->ProcessingPolicy->AuthAlgoId = NewData->ProcessingPolicy->AuthAlgoId;\r
1477 }\r
1478\r
1479 if ((Mask & ENCRYPT_ALGO) != 0) {\r
1480 OldData->ProcessingPolicy->EncAlgoId = NewData->ProcessingPolicy->EncAlgoId;\r
1481 }\r
1482\r
1483 if (OldData->ProcessingPolicy->Mode != EfiIPsecTunnel) {\r
1484 OldData->ProcessingPolicy->TunnelOption = NULL;\r
1485 } else {\r
1486 if (OldData->ProcessingPolicy->TunnelOption == NULL) {\r
1487 //\r
1488 // Set from Transport mode to Tunnel mode, should ensure TUNNEL_LOCAL & TUNNEL_REMOTE both exists.\r
1489 //\r
1490 if ((Mask & (TUNNEL_LOCAL | TUNNEL_REMOTE)) != (TUNNEL_LOCAL | TUNNEL_REMOTE)) {\r
1491 ShellPrintHiiEx (\r
1492 -1,\r
1493 -1,\r
1494 NULL,\r
1495 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
1496 mHiiHandle,\r
1497 mAppName,\r
1498 L"--tunnel-local --tunnel-remote"\r
1499 );\r
1500 return EFI_INVALID_PARAMETER;\r
1501 }\r
1502\r
1503 OldData->ProcessingPolicy->TunnelOption = NewData->ProcessingPolicy->TunnelOption;\r
1504 } else {\r
1505 if ((Mask & TUNNEL_LOCAL) != 0) {\r
1506 CopyMem (\r
1507 &OldData->ProcessingPolicy->TunnelOption->LocalTunnelAddress,\r
1508 &NewData->ProcessingPolicy->TunnelOption->LocalTunnelAddress,\r
1509 sizeof (EFI_IP_ADDRESS)\r
1510 );\r
1511 }\r
1512\r
1513 if ((Mask & TUNNEL_REMOTE) != 0) {\r
1514 CopyMem (\r
1515 &OldData->ProcessingPolicy->TunnelOption->RemoteTunnelAddress,\r
1516 &NewData->ProcessingPolicy->TunnelOption->RemoteTunnelAddress,\r
1517 sizeof (EFI_IP_ADDRESS)\r
1518 );\r
1519 }\r
1520\r
1521 if ((Mask & DONT_FRAGMENT) != 0) {\r
1522 OldData->ProcessingPolicy->TunnelOption->DF = NewData->ProcessingPolicy->TunnelOption->DF;\r
1523 }\r
1524 }\r
1525 }\r
1526 }\r
1527 }\r
1528\r
1529 return EFI_SUCCESS;\r
1530}\r
1531\r
1532/**\r
1533 Combine old SAD entry with new SAD entry.\r
1534\r
1535 @param[in, out] OldSaId The pointer to the EFI_IPSEC_SA_ID structure.\r
64b2d0e5 1536 @param[in, out] OldData The pointer to the EFI_IPSEC_SA_DATA2 structure.\r
a3bcde70 1537 @param[in] NewSaId The pointer to the EFI_IPSEC_SA_ID structure.\r
64b2d0e5 1538 @param[in] NewData The pointer to the EFI_IPSEC_SA_DATA2 structure.\r
a3bcde70
HT		 1539 @param[in] Mask The pointer to the Mask.\r
1540 @param[out] CreateNew The switch to create new.\r
1541\r
1542 @retval EFI_SUCCESS Combined successfully.\r
1543 @retval EFI_INVALID_PARAMETER Invalid user input parameter.\r
1544\r
1545**/\r
1546EFI_STATUS\r
1547CombineSadEntry (\r
1548 IN OUT EFI_IPSEC_SA_ID *OldSaId,\r
686d4d4a 1549 IN OUT EFI_IPSEC_SA_DATA2 *OldData,\r
a3bcde70 1550 IN EFI_IPSEC_SA_ID *NewSaId,\r
686d4d4a 1551 IN EFI_IPSEC_SA_DATA2 *NewData,\r
a3bcde70
HT		 1552 IN UINT32 Mask,\r
1553 OUT BOOLEAN *CreateNew\r
1554 )\r
1555{\r
1556\r
1557 *CreateNew = FALSE;\r
1558\r
1559 if ((Mask & SPI) == 0) {\r
1560 NewSaId->Spi = OldSaId->Spi;\r
1561 } else if (NewSaId->Spi != OldSaId->Spi) {\r
1562 *CreateNew = TRUE;\r
1563 }\r
1564\r
1565 if ((Mask & IPSEC_PROTO) == 0) {\r
1566 NewSaId->Proto = OldSaId->Proto;\r
1567 } else if (NewSaId->Proto != OldSaId->Proto) {\r
1568 *CreateNew = TRUE;\r
1569 }\r
1570\r
1571 if ((Mask & DEST) == 0) {\r
64b2d0e5 1572 CopyMem (&NewData->TunnelDestinationAddress, &OldData->TunnelDestinationAddress, sizeof (EFI_IP_ADDRESS));\r
1573 } else if (CompareMem (&NewData->TunnelDestinationAddress, &OldData->TunnelDestinationAddress, sizeof (EFI_IP_ADDRESS)) != 0) {\r
a3bcde70
HT		 1574 *CreateNew = TRUE;\r
1575 }\r
1576\r
64b2d0e5 1577 if ((Mask & SOURCE) == 0) {\r
1578 CopyMem (&NewData->TunnelSourceAddress, &OldData->TunnelSourceAddress, sizeof (EFI_IP_ADDRESS));\r
1579 } else if (CompareMem (&NewData->TunnelSourceAddress, &OldData->TunnelSourceAddress, sizeof (EFI_IP_ADDRESS)) != 0) {\r
1580 *CreateNew = TRUE;\r
1581 }\r
a3bcde70
HT		 1582 //\r
1583 // Process SA_DATA.\r
1584 //\r
1585 if ((Mask & MODE) != 0) {\r
1586 OldData->Mode = NewData->Mode;\r
1587 }\r
1588\r
1589 if ((Mask & SEQUENCE_NUMBER) != 0) {\r
1590 OldData->SNCount = NewData->SNCount;\r
1591 }\r
1592\r
1593 if ((Mask & ANTIREPLAY_WINDOW) != 0) {\r
1594 OldData->AntiReplayWindows = NewData->AntiReplayWindows;\r
1595 }\r
1596\r
1597 if ((Mask & AUTH_ALGO) != 0) {\r
1598 OldData->AlgoInfo.EspAlgoInfo.AuthAlgoId = NewData->AlgoInfo.EspAlgoInfo.AuthAlgoId;\r
1599 }\r
1600\r
1601 if ((Mask & AUTH_KEY) != 0) {\r
1602 OldData->AlgoInfo.EspAlgoInfo.AuthKey = NewData->AlgoInfo.EspAlgoInfo.AuthKey;\r
1603 OldData->AlgoInfo.EspAlgoInfo.AuthKeyLength = NewData->AlgoInfo.EspAlgoInfo.AuthKeyLength;\r
1604 }\r
1605\r
1606 if ((Mask & ENCRYPT_ALGO) != 0) {\r
1607 OldData->AlgoInfo.EspAlgoInfo.EncAlgoId = NewData->AlgoInfo.EspAlgoInfo.EncAlgoId;\r
1608 }\r
1609\r
1610 if ((Mask & ENCRYPT_KEY) != 0) {\r
1611 OldData->AlgoInfo.EspAlgoInfo.EncKey = NewData->AlgoInfo.EspAlgoInfo.EncKey;\r
1612 OldData->AlgoInfo.EspAlgoInfo.EncKeyLength = NewData->AlgoInfo.EspAlgoInfo.EncKeyLength;\r
1613 }\r
1614\r
1615 if (NewSaId->Proto == EfiIPsecAH) {\r
1616 if ((Mask & (ENCRYPT_ALGO | ENCRYPT_KEY)) != 0) {\r
1617 //\r
1618 // Should not provide encrypt_* if AH.\r
1619 //\r
1620 ShellPrintHiiEx (\r
1621 -1,\r
1622 -1,\r
1623 NULL,\r
1624 STRING_TOKEN (STR_IPSEC_CONFIG_UNWANTED_PARAMETER),\r
1625 mHiiHandle,\r
1626 mAppName,\r
1627 L"--encrypt-algo --encrypt-key"\r
1628 );\r
1629 return EFI_INVALID_PARAMETER;\r
1630 }\r
1631 }\r
1632\r
1633 if (NewSaId->Proto == EfiIPsecESP && OldSaId->Proto == EfiIPsecAH) {\r
1634 //\r
1635 // AH -> ESP\r
1636 // Should provide encrypt_algo at least.\r
1637 //\r
1638 if ((Mask & ENCRYPT_ALGO) == 0) {\r
1639 ShellPrintHiiEx (\r
1640 -1,\r
1641 -1,\r
1642 NULL,\r
1643 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_PARAMETER),\r
1644 mHiiHandle,\r
1645 mAppName,\r
1646 L"--encrypt-algo"\r
1647 );\r
1648 return EFI_INVALID_PARAMETER;\r
1649 }\r
1650\r
1651 //\r
1652 // Encrypt_key should be provided if algorithm is not NONE.\r
1653 //\r
780847d1 1654 if (NewData->AlgoInfo.EspAlgoInfo.EncAlgoId != IPSEC_EALG_NONE && (Mask & ENCRYPT_KEY) == 0) {\r
a3bcde70
HT		 1655 ShellPrintHiiEx (\r
1656 -1,\r
1657 -1,\r
1658 NULL,\r
1659 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_PARAMETER),\r
1660 mHiiHandle,\r
1661 mAppName,\r
1662 L"--encrypt-algo"\r
1663 );\r
1664 return EFI_INVALID_PARAMETER;\r
1665 }\r
1666 }\r
1667\r
1668 if ((Mask & LIFEBYTE) != 0) {\r
1669 OldData->SaLifetime.ByteCount = NewData->SaLifetime.ByteCount;\r
1670 }\r
1671\r
1672 if ((Mask & LIFETIME_SOFT) != 0) {\r
1673 OldData->SaLifetime.SoftLifetime = NewData->SaLifetime.SoftLifetime;\r
1674 }\r
1675\r
1676 if ((Mask & LIFETIME) != 0) {\r
1677 OldData->SaLifetime.HardLifetime = NewData->SaLifetime.HardLifetime;\r
1678 }\r
1679\r
1680 if ((Mask & PATH_MTU) != 0) {\r
1681 OldData->PathMTU = NewData->PathMTU;\r
1682 }\r
1683 //\r
1684 // Process SpdSelector.\r
1685 //\r
1686 if (OldData->SpdSelector == NULL) {\r
1687 if ((Mask & (LOCAL | REMOTE | PROTO | LOCAL_PORT | REMOTE_PORT | ICMP_TYPE | ICMP_CODE)) != 0) {\r
1688 if ((Mask & (LOCAL | REMOTE | PROTO)) != (LOCAL | REMOTE | PROTO)) {\r
1689 ShellPrintHiiEx (\r
1690 -1,\r
1691 -1,\r
1692 NULL,\r
1693 STRING_TOKEN (STR_IPSEC_CONFIG_MISSING_ONE_OF_PARAMETERS),\r
1694 mHiiHandle,\r
1695 mAppName,\r
1696 L"--local --remote --proto"\r
1697 );\r
1698 return EFI_INVALID_PARAMETER;\r
1699 }\r
1700\r
1701 OldData->SpdSelector = NewData->SpdSelector;\r
1702 }\r
1703 } else {\r
1704 if ((Mask & LOCAL) != 0) {\r
1705 OldData->SpdSelector->LocalAddressCount = NewData->SpdSelector->LocalAddressCount;\r
1706 OldData->SpdSelector->LocalAddress = NewData->SpdSelector->LocalAddress;\r
1707 }\r
1708\r
1709 if ((Mask & REMOTE) != 0) {\r
1710 OldData->SpdSelector->RemoteAddressCount = NewData->SpdSelector->RemoteAddressCount;\r
1711 OldData->SpdSelector->RemoteAddress = NewData->SpdSelector->RemoteAddress;\r
1712 }\r
1713\r
1714 if ((Mask & PROTO) != 0) {\r
1715 OldData->SpdSelector->NextLayerProtocol = NewData->SpdSelector->NextLayerProtocol;\r
1716 }\r
1717\r
1718 if (OldData->SpdSelector != NULL) {\r
1719 switch (OldData->SpdSelector->NextLayerProtocol) {\r
1720 case EFI_IP4_PROTO_TCP:\r
1721 case EFI_IP4_PROTO_UDP:\r
1722 if ((Mask & LOCAL_PORT) != 0) {\r
1723 OldData->SpdSelector->LocalPort = NewData->SpdSelector->LocalPort;\r
1724 }\r
1725\r
1726 if ((Mask & REMOTE_PORT) != 0) {\r
1727 OldData->SpdSelector->RemotePort = NewData->SpdSelector->RemotePort;\r
1728 }\r
1729 break;\r
1730\r
1731 case EFI_IP4_PROTO_ICMP:\r
1732 if ((Mask & ICMP_TYPE) != 0) {\r
1733 OldData->SpdSelector->LocalPort = (UINT8) NewData->SpdSelector->LocalPort;\r
1734 }\r
1735\r
1736 if ((Mask & ICMP_CODE) != 0) {\r
1737 OldData->SpdSelector->RemotePort = (UINT8) NewData->SpdSelector->RemotePort;\r
1738 }\r
1739 break;\r
1740 }\r
1741 }\r
1742 }\r
1743\r
1744 return EFI_SUCCESS;\r
1745}\r
1746\r
1747/**\r
1748 Combine old PAD entry with new PAD entry.\r
1749\r
1750 @param[in, out] OldPadId The pointer to the EFI_IPSEC_PAD_ID structure.\r
1751 @param[in, out] OldData The pointer to the EFI_IPSEC_PAD_DATA structure.\r
1752 @param[in] NewPadId The pointer to the EFI_IPSEC_PAD_ID structure.\r
1753 @param[in] NewData The pointer to the EFI_IPSEC_PAD_DATA structure.\r
1754 @param[in] Mask The pointer to the Mask.\r
1755 @param[out] CreateNew The switch to create new.\r
1756\r
1757 @retval EFI_SUCCESS Combined successfully.\r
1758 @retval EFI_INVALID_PARAMETER Invalid user input parameter.\r
1759\r
1760**/\r
1761EFI_STATUS\r
1762CombinePadEntry (\r
1763 IN OUT EFI_IPSEC_PAD_ID *OldPadId,\r
1764 IN OUT EFI_IPSEC_PAD_DATA *OldData,\r
1765 IN EFI_IPSEC_PAD_ID *NewPadId,\r
1766 IN EFI_IPSEC_PAD_DATA *NewData,\r
1767 IN UINT32 Mask,\r
1768 OUT BOOLEAN *CreateNew\r
1769 )\r
1770{\r
1771\r
1772 *CreateNew = FALSE;\r
1773\r
1774 if ((Mask & (PEER_ID | PEER_ADDRESS)) == 0) {\r
1775 CopyMem (NewPadId, OldPadId, sizeof (EFI_IPSEC_PAD_ID));\r
1776 } else {\r
1777 if ((Mask & PEER_ID) != 0) {\r
1778 if (OldPadId->PeerIdValid) {\r
1779 if (StrCmp ((CONST CHAR16 *) OldPadId->Id.PeerId, (CONST CHAR16 *) NewPadId->Id.PeerId) != 0) {\r
1780 *CreateNew = TRUE;\r
1781 }\r
1782 } else {\r
1783 *CreateNew = TRUE;\r
1784 }\r
1785 } else {\r
1786 //\r
1787 // MASK & PEER_ADDRESS\r
1788 //\r
1789 if (OldPadId->PeerIdValid) {\r
1790 *CreateNew = TRUE;\r
1791 } else {\r
1792 if ((CompareMem (&OldPadId->Id.IpAddress.Address, &NewPadId->Id.IpAddress.Address, sizeof (EFI_IP_ADDRESS)) != 0) ||\r
1793 (OldPadId->Id.IpAddress.PrefixLength != NewPadId->Id.IpAddress.PrefixLength)) {\r
1794 *CreateNew = TRUE;\r
1795 }\r
1796 }\r
1797 }\r
1798 }\r
1799\r
1800 if ((Mask & AUTH_PROTO) != 0) {\r
1801 OldData->AuthProtocol = NewData->AuthProtocol;\r
1802 }\r
1803\r
1804 if ((Mask & AUTH_METHOD) != 0) {\r
1805 OldData->AuthMethod = NewData->AuthMethod;\r
1806 }\r
1807\r
1808 if ((Mask & IKE_ID) != 0) {\r
1809 OldData->IkeIdFlag = NewData->IkeIdFlag;\r
1810 }\r
1811\r
1812 if ((Mask & AUTH_DATA) != 0) {\r
1813 OldData->AuthDataSize = NewData->AuthDataSize;\r
1814 OldData->AuthData = NewData->AuthData;\r
1815 }\r
1816\r
1817 if ((Mask & REVOCATION_DATA) != 0) {\r
1818 OldData->RevocationDataSize = NewData->RevocationDataSize;\r
1819 OldData->RevocationData = NewData->RevocationData;\r
1820 }\r
1821\r
1822 return EFI_SUCCESS;\r
1823}\r
1824\r
1825COMBINE_POLICY_ENTRY mCombinePolicyEntry[] = {\r
1826 (COMBINE_POLICY_ENTRY) CombineSpdEntry,\r
1827 (COMBINE_POLICY_ENTRY) CombineSadEntry,\r
1828 (COMBINE_POLICY_ENTRY) CombinePadEntry\r
1829};\r
1830\r
1831/**\r
1832 Edit entry information in the database.\r
1833\r
1834 @param[in] Selector The pointer to the EFI_IPSEC_CONFIG_SELECTOR structure.\r
1835 @param[in] Data The pointer to the data.\r
1836 @param[in] Context The pointer to the INSERT_POLICY_ENTRY_CONTEXT structure.\r
1837\r
1838 @retval EFI_SUCCESS Continue the iteration.\r
1839 @retval EFI_ABORTED Abort the iteration.\r
1840**/\r
1841EFI_STATUS\r
1842EditOperatePolicyEntry (\r
1843 IN EFI_IPSEC_CONFIG_SELECTOR *Selector,\r
1844 IN VOID *Data,\r
1845 IN EDIT_POLICY_ENTRY_CONTEXT *Context\r
1846 )\r
1847{\r
1848 EFI_STATUS Status;\r
1849 BOOLEAN CreateNew;\r
1850\r
1851 if (mMatchPolicyEntry[Context->DataType] (Selector, Data, &Context->Indexer)) {\r
1852 ASSERT (Context->DataType < 3);\r
1853\r
1854 Status = mCombinePolicyEntry[Context->DataType] (\r
1855 Selector,\r
1856 Data,\r
1857 Context->Selector,\r
1858 Context->Data,\r
1859 Context->Mask,\r
1860 &CreateNew\r
1861 );\r
1862 if (!EFI_ERROR (Status)) {\r
1863 if (CreateNew) {\r
1864 //\r
1865 // Insert new entry before old entry\r
1866 //\r
1867 Status = mIpSecConfig->SetData (\r
1868 mIpSecConfig,\r
1869 Context->DataType,\r
1870 Context->Selector,\r
1871 Data,\r
1872 Selector\r
1873 );\r
1874 ASSERT_EFI_ERROR (Status);\r
1875 //\r
1876 // Delete old entry\r
1877 //\r
1878 Status = mIpSecConfig->SetData (\r
1879 mIpSecConfig,\r
1880 Context->DataType,\r
1881 Selector,\r
1882 NULL,\r
1883 NULL\r
1884 );\r
1885 ASSERT_EFI_ERROR (Status);\r
1886 } else {\r
1887 Status = mIpSecConfig->SetData (\r
1888 mIpSecConfig,\r
1889 Context->DataType,\r
1890 Context->Selector,\r
1891 Data,\r
1892 NULL\r
1893 );\r
1894 }\r
1895 }\r
1896\r
1897 Context->Status = Status;\r
1898 return EFI_ABORTED;\r
1899 }\r
1900\r
1901 return EFI_SUCCESS;\r
1902}\r
1903\r
1904/**\r
1905 Edit entry information in database according to datatype.\r
1906\r
1907 @param[in] DataType The value of EFI_IPSEC_CONFIG_DATA_TYPE.\r
1908 @param[in] ParamPackage The pointer to the ParamPackage list.\r
1909\r
1910 @retval EFI_SUCCESS Edit entry information successfully.\r
1911 @retval EFI_NOT_FOUND Can't find the specified entry.\r
1912 @retval Others Some mistaken case.\r
1913**/\r
1914EFI_STATUS\r
1915EditPolicyEntry (\r
1916 IN EFI_IPSEC_CONFIG_DATA_TYPE DataType,\r
1917 IN LIST_ENTRY *ParamPackage\r
1918 )\r
1919{\r
1920 EFI_STATUS Status;\r
1921 EDIT_POLICY_ENTRY_CONTEXT Context;\r
1922 CONST CHAR16 *ValueStr;\r
1923\r
1924 ValueStr = ShellCommandLineGetValue (ParamPackage, L"-e");\r
1925 if (ValueStr == NULL) {\r
1926 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IPSEC_CONFIG_INDEX_NOT_SPECIFIED), mHiiHandle, mAppName, ValueStr);\r
1927 return EFI_NOT_FOUND;\r
1928 }\r
1929\r
1930 Status = mConstructPolicyEntryIndexer[DataType] (&Context.Indexer, ParamPackage);\r
1931 if (!EFI_ERROR (Status)) {\r
1932 Context.DataType = DataType;\r
1933 Context.Status = EFI_NOT_FOUND;\r
1934 Status = mCreatePolicyEntry[DataType] (&Context.Selector, &Context.Data, ParamPackage, &Context.Mask, FALSE);\r
1935 if (!EFI_ERROR (Status)) {\r
1936 ForeachPolicyEntry (DataType, (VISIT_POLICY_ENTRY) EditOperatePolicyEntry, &Context);\r
1937 Status = Context.Status;\r
1938 }\r
1939\r
1940 if (Context.Selector != NULL) {\r
1941 gBS->FreePool (Context.Selector);\r
1942 }\r
1943\r
1944 if (Context.Data != NULL) {\r
1945 gBS->FreePool (Context.Data);\r
1946 }\r
1947 }\r
1948\r
1949 if (Status == EFI_NOT_FOUND) {\r
1950 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IPSEC_CONFIG_INDEX_NOT_FOUND), mHiiHandle, mAppName, ValueStr);\r
1951 } else if (EFI_ERROR (Status)) {\r
1952 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IPSEC_CONFIG_EDIT_FAILED), mHiiHandle, mAppName);\r
1953 }\r
1954\r
1955 return Status;\r
1956\r
1957}\r
1958\r
1959/**\r
1960 Insert entry information in database.\r
1961\r
1962 @param[in] Selector The pointer to the EFI_IPSEC_CONFIG_SELECTOR structure.\r
1963 @param[in] Data The pointer to the data.\r
1964 @param[in] Context The pointer to the INSERT_POLICY_ENTRY_CONTEXT structure.\r
1965\r
1966 @retval EFI_SUCCESS Continue the iteration.\r
1967 @retval EFI_ABORTED Abort the iteration.\r
1968**/\r
1969EFI_STATUS\r
1970InsertPolicyEntry (\r
1971 IN EFI_IPSEC_CONFIG_SELECTOR *Selector,\r
1972 IN VOID *Data,\r
1973 IN INSERT_POLICY_ENTRY_CONTEXT *Context\r
1974 )\r
1975{\r
1976 //\r
1977 // Found the entry which we want to insert before.\r
1978 //\r
1979 if (mMatchPolicyEntry[Context->DataType] (Selector, Data, &Context->Indexer)) {\r
1980\r
1981 Context->Status = mIpSecConfig->SetData (\r
1982 mIpSecConfig,\r
1983 Context->DataType,\r
1984 Context->Selector,\r
1985 Context->Data,\r
1986 Selector\r
1987 );\r
1988 //\r
1989 // Abort the iteration after the insertion.\r
1990 //\r
1991 return EFI_ABORTED;\r
1992 }\r
1993\r
1994 return EFI_SUCCESS;\r
1995}\r
1996\r
1997/**\r
1998 Insert or add entry information in database according to datatype.\r
1999\r
2000 @param[in] DataType The value of EFI_IPSEC_CONFIG_DATA_TYPE.\r
2001 @param[in] ParamPackage The pointer to the ParamPackage list.\r
2002\r
2003 @retval EFI_SUCCESS Insert or add entry information successfully.\r
2004 @retval EFI_NOT_FOUND Can't find the specified entry.\r
2005 @retval EFI_BUFFER_TOO_SMALL The entry already existed.\r
2006 @retval EFI_UNSUPPORTED The operation is not supported.\r
2007 @retval Others Some mistaken case.\r
2008**/\r
2009EFI_STATUS\r
2010AddOrInsertPolicyEntry (\r
2011 IN EFI_IPSEC_CONFIG_DATA_TYPE DataType,\r
2012 IN LIST_ENTRY *ParamPackage\r
2013 )\r
2014{\r
2015 EFI_STATUS Status;\r
2016 EFI_IPSEC_CONFIG_SELECTOR *Selector;\r
2017 VOID *Data;\r
2018 INSERT_POLICY_ENTRY_CONTEXT Context;\r
2019 UINT32 Mask;\r
2020 UINTN DataSize;\r
2021 CONST CHAR16 *ValueStr;\r
2022\r
2023 Status = mCreatePolicyEntry[DataType] (&Selector, &Data, ParamPackage, &Mask, TRUE);\r
2024 if (!EFI_ERROR (Status)) {\r
2025 //\r
2026 // Find if the Selector to be inserted already exists.\r
2027 //\r
2028 DataSize = 0;\r
2029 Status = mIpSecConfig->GetData (\r
2030 mIpSecConfig,\r
2031 DataType,\r
2032 Selector,\r
2033 &DataSize,\r
2034 NULL\r
2035 );\r
2036 if (Status == EFI_BUFFER_TOO_SMALL) {\r
2037 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IPSEC_CONFIG_ALREADY_EXISTS), mHiiHandle, mAppName);\r
2038 } else if (ShellCommandLineGetFlag (ParamPackage, L"-a")) {\r
2039 Status = mIpSecConfig->SetData (\r
2040 mIpSecConfig,\r
2041 DataType,\r
2042 Selector,\r
2043 Data,\r
2044 NULL\r
2045 );\r
2046 } else {\r
2047 ValueStr = ShellCommandLineGetValue (ParamPackage, L"-i");\r
2048 if (ValueStr == NULL) {\r
2049 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IPSEC_CONFIG_INDEX_NOT_SPECIFIED), mHiiHandle, mAppName, ValueStr);\r
2050 return EFI_NOT_FOUND;\r
2051 }\r
2052\r
2053 Status = mConstructPolicyEntryIndexer[DataType] (&Context.Indexer, ParamPackage);\r
2054 if (!EFI_ERROR (Status)) {\r
2055 Context.DataType = DataType;\r
2056 Context.Status = EFI_NOT_FOUND;\r
2057 Context.Selector = Selector;\r
2058 Context.Data = Data;\r
2059\r
2060 ForeachPolicyEntry (DataType, (VISIT_POLICY_ENTRY) InsertPolicyEntry, &Context);\r
2061 Status = Context.Status;\r
2062 if (Status == EFI_NOT_FOUND) {\r
2063 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IPSEC_CONFIG_INDEX_NOT_FOUND), mHiiHandle, mAppName, ValueStr);\r
2064 }\r
2065 }\r
2066 }\r
2067\r
2068 gBS->FreePool (Selector);\r
2069 gBS->FreePool (Data);\r
2070 }\r
2071\r
2072 if (Status == EFI_UNSUPPORTED) {\r
2073 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IPSEC_CONFIG_INSERT_UNSUPPORT), mHiiHandle, mAppName);\r
2074 } else if (EFI_ERROR (Status)) {\r
2075 ShellPrintHiiEx (-1, -1, NULL, STRING_TOKEN (STR_IPSEC_CONFIG_INSERT_FAILED), mHiiHandle, mAppName);\r
2076 }\r
2077\r
2078 return Status;\r
2079}\r
EFI Development Kit II mirror for PVE