1 files changed, 145 insertions, 0 deletions
diff --git a/MdeModulePkg/Universal/Network/IScsiDxe/IScsiInitiatorName.c b/MdeModulePkg/Universal/Network/IScsiDxe/IScsiInitiatorName.c
new file mode 100644
index 0000000000..f93594fdcd
--- /dev/null
+++ b/MdeModulePkg/Universal/Network/IScsiDxe/IScsiInitiatorName.c
@@ -0,0 +1,145 @@
+/*++
+
+Copyright (c) 2004 - 2007, Intel Corporation
+All rights reserved. This program and the accompanying materials
+are licensed and made available under the terms and conditions of the BSD License
+which accompanies this distribution.  The full text of the license may be found at
+http://opensource.org/licenses/bsd-license.php
+
+THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
+WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
+
+Module Name:
+
+  IScsiInitiatorName.c
+
+Abstract:
+
+  Implementation for EFI iSCSI Initiator Name Protocol.
+
+--*/
+
+#include "IScsiImpl.h"
+
+EFI_ISCSI_INITIATOR_NAME_PROTOCOL gIScsiInitiatorName = {
+  IScsiGetInitiatorName,
+  IScsiSetInitiatorName
+};
+
+EFI_STATUS
+EFIAPI
+IScsiGetInitiatorName (
+  IN     EFI_ISCSI_INITIATOR_NAME_PROTOCOL  *This,
+  IN OUT UINTN                              *BufferSize,
+  OUT    VOID                               *Buffer
+  )
+/*++
+
+Routine Description:
+
+  Retrieves the current set value of iSCSI Initiator Name. 
+
+Arguments:
+
+  This       - Pointer to the EFI_ISCSI_INITIATOR_NAME_PROTOCOL instance.
+  BufferSize - Size of the buffer in bytes pointed to by Buffer / Actual size of
+               the variable data buffer.
+  Buffer     - Pointer to the buffer for data to be read.
+
+Returns:
+
+  EFI_SUCCESS           - Data was successfully retrieved into the provided 
+                          buffer and the BufferSize was sufficient to handle the
+                          iSCSI initiator name.
+  EFI_BUFFER_TOO_SMALL  - BufferSize is too small for the result. BufferSize will
+                          be updated with the size required to complete the request.
+                          Buffer will not be affected.
+  EFI_INVALID_PARAMETER - BufferSize is NULL. BufferSize and Buffer will not be
+                          affected.
+  EFI_INVALID_PARAMETER - Buffer is NULL. BufferSize and Buffer will not be
+                          affected.
+  EFI_DEVICE_ERROR      - The iSCSI initiator name could not be retrieved due to
+                          a hardware error.
+
+--*/
+{
+  EFI_STATUS  Status;
+
+  if ((BufferSize == NULL) || (Buffer == NULL)) {
+    return EFI_INVALID_PARAMETER;
+  }
+
+  Status = gRT->GetVariable (
+                  ISCSI_INITIATOR_NAME_VAR_NAME,
+                  &gEfiIScsiInitiatorNameProtocolGuid,
+                  NULL,
+                  BufferSize,
+                  Buffer
+                  );
+
+  return Status;
+}
+
+EFI_STATUS
+EFIAPI
+IScsiSetInitiatorName (
+  IN     EFI_ISCSI_INITIATOR_NAME_PROTOCOL  *This,
+  IN OUT UINTN                              *BufferSize,
+  OUT    VOID                               *Buffer
+  )
+/*++
+
+Routine Description:
+
+  Sets the iSSI Initiator Name. 
+
+Arguments:
+
+  This       - Pointer to the EFI_ISCSI_INITIATOR_NAME_PROTOCOL instance.
+  BufferSize - Size of the buffer in bytes pointed to by Buffer.
+  Buffer     - Pointer to the buffer for data to be written.
+
+Returns:
+
+  EFI_SUCCESS           - Data was successfully stored by the protocol.
+  EFI_UNSUPPORTED       - Platform policies do not allow for data to be written.
+  EFI_INVALID_PARAMETER - BufferSize exceeds the maximum allowed limit.
+                          BufferSize will be updated with the maximum size
+                          required to complete the request.
+  EFI_INVALID_PARAMETER - Buffersize is NULL. BufferSize and Buffer will not be
+                          affected.
+  EFI_INVALID_PARAMETER - Buffer is NULL. BufferSize and Buffer will not be affected.
+  EFI_DEVICE_ERROR      - The data could not be stored due to a hardware error.
+  EFI_OUT_OF_RESOURCES  - Not enough storage is available to hold the data
+  EFI_PROTOCOL_ERROR    - Input iSCSI initiator name does not adhere to RFC 3720
+
+--*/
+{
+  EFI_STATUS  Status;
+
+  if ((BufferSize == NULL) || (Buffer == NULL)) {
+    return EFI_INVALID_PARAMETER;
+  }
+
+  if (*BufferSize > ISCSI_NAME_MAX_SIZE) {
+    *BufferSize = ISCSI_NAME_MAX_SIZE;
+    return EFI_INVALID_PARAMETER;
+  }
+  //
+  // only support iqn iSCSI names.
+  //
+  Status = IScsiNormalizeName ((CHAR8 *) Buffer, *BufferSize - 1);
+  if (EFI_ERROR (Status)) {
+    return Status;
+  }
+
+  Status = gRT->SetVariable (
+                  ISCSI_INITIATOR_NAME_VAR_NAME,
+                  &gEfiIScsiInitiatorNameProtocolGuid,
+                  EFI_VARIABLE_NON_VOLATILE | EFI_VARIABLE_BOOTSERVICE_ACCESS,
+                  *BufferSize,
+                  Buffer
+                  );
+
+  return Status;
+}