Udp6Driver.c

Go to the documentation of this file.

 
#include "Udp6Impl.h"
 
EFI_DRIVER_BINDING_PROTOCOL  gUdp6DriverBinding = {
  Udp6DriverBindingSupported,
  Udp6DriverBindingStart,
  Udp6DriverBindingStop,
  0xa,
  NULL,
  NULL
};
 
EFI_SERVICE_BINDING_PROTOCOL  mUdp6ServiceBinding = {
  Udp6ServiceBindingCreateChild,
  Udp6ServiceBindingDestroyChild
};
 
EFI_STATUS
EFIAPI
Udp6DriverBindingSupported (
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,
  IN EFI_HANDLE                   ControllerHandle,
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath  OPTIONAL
  )
{
  EFI_STATUS  Status;
 
  //
  // Test for the Udp6ServiceBinding Protocol
  //
  Status = gBS->OpenProtocol (
                  ControllerHandle,
                  &gEfiUdp6ServiceBindingProtocolGuid,
                  NULL,
                  This->DriverBindingHandle,
                  ControllerHandle,
                  EFI_OPEN_PROTOCOL_TEST_PROTOCOL
                  );
  if (!EFI_ERROR (Status)) {
    return EFI_ALREADY_STARTED;
  }
 
  //
  // Test for the Ip6ServiceBinding Protocol
  //
  Status = gBS->OpenProtocol (
                  ControllerHandle,
                  &gEfiIp6ServiceBindingProtocolGuid,
                  NULL,
                  This->DriverBindingHandle,
                  ControllerHandle,
                  EFI_OPEN_PROTOCOL_TEST_PROTOCOL
                  );
 
  return Status;
}
 
EFI_STATUS
EFIAPI
Udp6DriverBindingStart (
  IN EFI_DRIVER_BINDING_PROTOCOL  *This,
  IN EFI_HANDLE                   ControllerHandle,
  IN EFI_DEVICE_PATH_PROTOCOL     *RemainingDevicePath  OPTIONAL
  )
{
  EFI_STATUS         Status;
  UDP6_SERVICE_DATA  *Udp6Service;
 
  //
  // Allocate Private Context Data Structure.
  //
  Udp6Service = AllocateZeroPool (sizeof (UDP6_SERVICE_DATA));
  if (Udp6Service == NULL) {
    Status = EFI_OUT_OF_RESOURCES;
    goto EXIT;
  }
 
  Status = Udp6CreateService (Udp6Service, This->DriverBindingHandle, ControllerHandle);
  if (EFI_ERROR (Status)) {
    goto EXIT;
  }
 
  //
  // Install the Udp6ServiceBindingProtocol on the ControllerHandle.
  //
  Status = gBS->InstallMultipleProtocolInterfaces (
                  &ControllerHandle,
                  &gEfiUdp6ServiceBindingProtocolGuid,
                  &Udp6Service->ServiceBinding,
                  NULL
                  );
  if (EFI_ERROR (Status)) {
    Udp6CleanService (Udp6Service);
  }
 
EXIT:
  if (EFI_ERROR (Status)) {
    if (Udp6Service != NULL) {
      FreePool (Udp6Service);
    }
  }
 
  return Status;
}
 
EFI_STATUS
EFIAPI
Udp6DestroyChildEntryInHandleBuffer (
  IN LIST_ENTRY  *Entry,
  IN VOID        *Context
  )
{
  UDP6_INSTANCE_DATA            *Instance;
  EFI_SERVICE_BINDING_PROTOCOL  *ServiceBinding;
  UINTN                         NumberOfChildren;
  EFI_HANDLE                    *ChildHandleBuffer;
 
  if ((Entry == NULL) || (Context == NULL)) {
    return EFI_INVALID_PARAMETER;
  }
 
  Instance          = NET_LIST_USER_STRUCT_S (Entry, UDP6_INSTANCE_DATA, Link, UDP6_INSTANCE_DATA_SIGNATURE);
  ServiceBinding    = ((UDP6_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT *)Context)->ServiceBinding;
  NumberOfChildren  = ((UDP6_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT *)Context)->NumberOfChildren;
  ChildHandleBuffer = ((UDP6_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT *)Context)->ChildHandleBuffer;
 
  if (!NetIsInHandleBuffer (Instance->ChildHandle, NumberOfChildren, ChildHandleBuffer)) {
    return EFI_SUCCESS;
  }
 
  return ServiceBinding->DestroyChild (ServiceBinding, Instance->ChildHandle);
}
 
EFI_STATUS
EFIAPI
Udp6DriverBindingStop (
  IN  EFI_DRIVER_BINDING_PROTOCOL  *This,
  IN  EFI_HANDLE                   ControllerHandle,
  IN  UINTN                        NumberOfChildren,
  IN  EFI_HANDLE                   *ChildHandleBuffer OPTIONAL
  )
{
  EFI_STATUS                                Status;
  EFI_HANDLE                                NicHandle;
  EFI_SERVICE_BINDING_PROTOCOL              *ServiceBinding;
  UDP6_SERVICE_DATA                         *Udp6Service;
  LIST_ENTRY                                *List;
  UDP6_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT  Context;
 
  //
  // Find the NicHandle where UDP6 ServiceBinding Protocol is installed.
  //
  NicHandle = NetLibGetNicHandle (ControllerHandle, &gEfiIp6ProtocolGuid);
  if (NicHandle == NULL) {
    return EFI_SUCCESS;
  }
 
  //
  // Retrieve the UDP6 ServiceBinding Protocol.
  //
  Status = gBS->OpenProtocol (
                  NicHandle,
                  &gEfiUdp6ServiceBindingProtocolGuid,
                  (VOID **)&ServiceBinding,
                  This->DriverBindingHandle,
                  NicHandle,
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL
                  );
  if (EFI_ERROR (Status)) {
    return EFI_DEVICE_ERROR;
  }
 
  Udp6Service = UDP6_SERVICE_DATA_FROM_THIS (ServiceBinding);
 
  if (NumberOfChildren != 0) {
    //
    // NumberOfChildren is not zero, destroy the children instances in ChildHandleBuffer.
    //
    List                      = &Udp6Service->ChildrenList;
    Context.ServiceBinding    = ServiceBinding;
    Context.NumberOfChildren  = NumberOfChildren;
    Context.ChildHandleBuffer = ChildHandleBuffer;
    Status                    = NetDestroyLinkList (
                                  List,
                                  Udp6DestroyChildEntryInHandleBuffer,
                                  &Context,
                                  NULL
                                  );
  } else if (IsListEmpty (&Udp6Service->ChildrenList)) {
    Status = gBS->UninstallMultipleProtocolInterfaces (
                    NicHandle,
                    &gEfiUdp6ServiceBindingProtocolGuid,
                    &Udp6Service->ServiceBinding,
                    NULL
                    );
 
    Udp6CleanService (Udp6Service);
    FreePool (Udp6Service);
  }
 
  return Status;
}
 
EFI_STATUS
EFIAPI
Udp6ServiceBindingCreateChild (
  IN EFI_SERVICE_BINDING_PROTOCOL  *This,
  IN OUT EFI_HANDLE                *ChildHandle
  )
{
  EFI_STATUS          Status;
  UDP6_SERVICE_DATA   *Udp6Service;
  UDP6_INSTANCE_DATA  *Instance;
  EFI_TPL             OldTpl;
  VOID                *Ip6;
 
  if ((This == NULL) || (ChildHandle == NULL)) {
    return EFI_INVALID_PARAMETER;
  }
 
  Udp6Service = UDP6_SERVICE_DATA_FROM_THIS (This);
 
  //
  // Allocate the instance private data structure.
  //
  Instance = AllocateZeroPool (sizeof (UDP6_INSTANCE_DATA));
  if (Instance == NULL) {
    return EFI_OUT_OF_RESOURCES;
  }
 
  Udp6InitInstance (Udp6Service, Instance);
 
  //
  // Add an IpInfo for this instance.
  //
  Instance->IpInfo = IpIoAddIp (Udp6Service->IpIo);
  if (Instance->IpInfo == NULL) {
    Status = EFI_OUT_OF_RESOURCES;
    goto ON_ERROR;
  }
 
  //
  // Install the Udp6Protocol for this instance.
  //
  Status = gBS->InstallMultipleProtocolInterfaces (
                  ChildHandle,
                  &gEfiUdp6ProtocolGuid,
                  &Instance->Udp6Proto,
                  NULL
                  );
  if (EFI_ERROR (Status)) {
    goto ON_ERROR;
  }
 
  Instance->ChildHandle = *ChildHandle;
 
  //
  // Open the default Ip6 protocol in the IP_IO BY_CHILD.
  //
  Status = gBS->OpenProtocol (
                  Udp6Service->IpIo->ChildHandle,
                  &gEfiIp6ProtocolGuid,
                  (VOID **)&Ip6,
                  gUdp6DriverBinding.DriverBindingHandle,
                  Instance->ChildHandle,
                  EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
                  );
  if (EFI_ERROR (Status)) {
    goto ON_ERROR;
  }
 
  //
  // Open this instance's Ip6 protocol in the IpInfo BY_CHILD.
  //
  Status = gBS->OpenProtocol (
                  Instance->IpInfo->ChildHandle,
                  &gEfiIp6ProtocolGuid,
                  (VOID **)&Ip6,
                  gUdp6DriverBinding.DriverBindingHandle,
                  Instance->ChildHandle,
                  EFI_OPEN_PROTOCOL_BY_CHILD_CONTROLLER
                  );
  if (EFI_ERROR (Status)) {
    goto ON_ERROR;
  }
 
  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);
 
  //
  // Link this instance into the service context data and increase the ChildrenNumber.
  //
  InsertTailList (&Udp6Service->ChildrenList, &Instance->Link);
  Udp6Service->ChildrenNumber++;
 
  gBS->RestoreTPL (OldTpl);
 
  return EFI_SUCCESS;
 
ON_ERROR:
 
  if (Instance->ChildHandle != NULL) {
    gBS->UninstallMultipleProtocolInterfaces (
           Instance->ChildHandle,
           &gEfiUdp6ProtocolGuid,
           &Instance->Udp6Proto,
           NULL
           );
  }
 
  if (Instance->IpInfo != NULL) {
    IpIoRemoveIp (Udp6Service->IpIo, Instance->IpInfo);
  }
 
  Udp6CleanInstance (Instance);
 
  FreePool (Instance);
 
  return Status;
}
 
EFI_STATUS
EFIAPI
Udp6ServiceBindingDestroyChild (
  IN EFI_SERVICE_BINDING_PROTOCOL  *This,
  IN EFI_HANDLE                    ChildHandle
  )
{
  EFI_STATUS          Status;
  UDP6_SERVICE_DATA   *Udp6Service;
  EFI_UDP6_PROTOCOL   *Udp6Proto;
  UDP6_INSTANCE_DATA  *Instance;
  EFI_TPL             OldTpl;
 
  if ((This == NULL) || (ChildHandle == NULL)) {
    return EFI_INVALID_PARAMETER;
  }
 
  Udp6Service = UDP6_SERVICE_DATA_FROM_THIS (This);
 
  //
  // Try to get the Udp6 protocol from the ChildHandle.
  //
  Status = gBS->OpenProtocol (
                  ChildHandle,
                  &gEfiUdp6ProtocolGuid,
                  (VOID **)&Udp6Proto,
                  gUdp6DriverBinding.DriverBindingHandle,
                  ChildHandle,
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL
                  );
  if (EFI_ERROR (Status)) {
    return EFI_UNSUPPORTED;
  }
 
  Instance = UDP6_INSTANCE_DATA_FROM_THIS (Udp6Proto);
 
  if (Instance->InDestroy) {
    return EFI_SUCCESS;
  }
 
  //
  // Use the Destroyed flag to avoid the re-entering of the following code.
  //
  Instance->InDestroy = TRUE;
 
  //
  // Close the Ip6 protocol on the default IpIo.
  //
  Status = gBS->CloseProtocol (
                  Udp6Service->IpIo->ChildHandle,
                  &gEfiIp6ProtocolGuid,
                  gUdp6DriverBinding.DriverBindingHandle,
                  Instance->ChildHandle
                  );
  if (EFI_ERROR (Status)) {
    Instance->InDestroy = FALSE;
    return Status;
  }
 
  //
  // Close the Ip6 protocol on this instance's IpInfo.
  //
  Status = gBS->CloseProtocol (
                  Instance->IpInfo->ChildHandle,
                  &gEfiIp6ProtocolGuid,
                  gUdp6DriverBinding.DriverBindingHandle,
                  Instance->ChildHandle
                  );
  if (EFI_ERROR (Status)) {
    Instance->InDestroy = FALSE;
    return Status;
  }
 
  //
  // Uninstall the Udp6Protocol previously installed on the ChildHandle.
  //
  Status = gBS->UninstallMultipleProtocolInterfaces (
                  ChildHandle,
                  &gEfiUdp6ProtocolGuid,
                  (VOID *)&Instance->Udp6Proto,
                  NULL
                  );
  if (EFI_ERROR (Status)) {
    Instance->InDestroy = FALSE;
    return Status;
  }
 
  //
  // Reset the configuration in case the instance's consumer forgets to do this.
  //
  Udp6Proto->Configure (Udp6Proto, NULL);
 
  //
  // Remove the IpInfo this instance consumes.
  //
  IpIoRemoveIp (Udp6Service->IpIo, Instance->IpInfo);
 
  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);
 
  //
  // Remove this instance from the service context data's ChildrenList.
  //
  RemoveEntryList (&Instance->Link);
  Udp6Service->ChildrenNumber--;
 
  //
  // Clean the instance.
  //
  Udp6CleanInstance (Instance);
 
  gBS->RestoreTPL (OldTpl);
 
  FreePool (Instance);
 
  return EFI_SUCCESS;
}
 
EFI_STATUS
EFIAPI
Udp6DriverEntryPoint (
  IN EFI_HANDLE        ImageHandle,
  IN EFI_SYSTEM_TABLE  *SystemTable
  )
{
  EFI_STATUS  Status;
  UINT32      Random;
 
  Status = PseudoRandomU32 (&Random);
  if (EFI_ERROR (Status)) {
    DEBUG ((DEBUG_ERROR, "%a failed to generate random number: %r\n", __func__, Status));
    return Status;
  }
 
  //
  // Install the Udp6DriverBinding and Udp6ComponentName protocols.
  //
 
  Status = EfiLibInstallDriverBindingComponentName2 (
             ImageHandle,
             SystemTable,
             &gUdp6DriverBinding,
             ImageHandle,
             &gUdp6ComponentName,
             &gUdp6ComponentName2
             );
  if (!EFI_ERROR (Status)) {
    //
    // Initialize the UDP random port.
    //
    mUdp6RandomPort = (UINT16)(
                               ((UINT16)Random) %
                               UDP6_PORT_KNOWN +
                               UDP6_PORT_KNOWN
                               );
  }
 
  return Status;
}