Commit | Line | Data |
---|---|---|
81a23a0f LL |
1 | /*++ |
2 | ||
3 | Copyright (c) 2005 - 2007, Intel Corporation. All rights reserved.<BR> | |
4 | This program and the accompanying materials | |
5 | are licensed and made available under the terms and conditions of the BSD License | |
6 | which accompanies this distribution. The full text of the license may be found at | |
7 | http://opensource.org/licenses/bsd-license.php | |
8 | ||
9 | THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS, | |
10 | WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. | |
11 | ||
12 | Module Name: | |
13 | ||
14 | PciDriverOverride.c | |
15 | ||
16 | Abstract: | |
17 | ||
18 | PCI Bus Driver | |
19 | ||
20 | Revision History | |
21 | ||
22 | --*/ | |
23 | ||
24 | #include "PciBus.h" | |
25 | ||
26 | EFI_STATUS | |
27 | EFIAPI | |
28 | GetDriver( | |
29 | IN EFI_BUS_SPECIFIC_DRIVER_OVERRIDE_PROTOCOL *This, | |
30 | IN OUT EFI_HANDLE *DriverImageHandle | |
31 | ); | |
32 | ||
33 | ||
34 | ||
35 | EFI_STATUS | |
36 | InitializePciDriverOverrideInstance ( | |
37 | PCI_IO_DEVICE *PciIoDevice | |
38 | ) | |
39 | /*++ | |
40 | ||
41 | Routine Description: | |
42 | ||
43 | Initializes a PCI Driver Override Instance | |
44 | ||
45 | Arguments: | |
46 | ||
47 | Returns: | |
48 | ||
49 | None | |
50 | ||
51 | --*/ | |
52 | ||
53 | { | |
54 | PciIoDevice->PciDriverOverride.GetDriver = GetDriver; | |
55 | return EFI_SUCCESS; | |
56 | } | |
57 | ||
58 | EFI_STATUS | |
59 | EFIAPI | |
60 | GetDriver ( | |
61 | IN EFI_BUS_SPECIFIC_DRIVER_OVERRIDE_PROTOCOL *This, | |
62 | IN OUT EFI_HANDLE *DriverImageHandle | |
63 | ) | |
64 | /*++ | |
65 | ||
66 | Routine Description: | |
67 | ||
68 | Get a overriding driver image | |
69 | ||
70 | Arguments: | |
71 | ||
72 | Returns: | |
73 | ||
74 | None | |
75 | ||
76 | --*/ | |
77 | { | |
78 | PCI_IO_DEVICE *PciIoDevice; | |
79 | LIST_ENTRY *CurrentLink; | |
80 | PCI_DRIVER_OVERRIDE_LIST *Node; | |
81 | ||
82 | PciIoDevice = PCI_IO_DEVICE_FROM_PCI_DRIVER_OVERRIDE_THIS (This); | |
83 | ||
84 | CurrentLink = PciIoDevice->OptionRomDriverList.ForwardLink; | |
85 | ||
86 | while (CurrentLink && CurrentLink != &PciIoDevice->OptionRomDriverList) { | |
87 | ||
88 | Node = DRIVER_OVERRIDE_FROM_LINK (CurrentLink); | |
89 | ||
90 | if (*DriverImageHandle == NULL) { | |
91 | ||
92 | *DriverImageHandle = Node->DriverImageHandle; | |
93 | return EFI_SUCCESS; | |
94 | } | |
95 | ||
96 | if (*DriverImageHandle == Node->DriverImageHandle) { | |
97 | ||
98 | if (CurrentLink->ForwardLink == &PciIoDevice->OptionRomDriverList || | |
99 | CurrentLink->ForwardLink == NULL) { | |
100 | return EFI_NOT_FOUND; | |
101 | } | |
102 | ||
103 | // | |
104 | // Get next node | |
105 | // | |
106 | Node = DRIVER_OVERRIDE_FROM_LINK (CurrentLink->ForwardLink); | |
107 | *DriverImageHandle = Node->DriverImageHandle; | |
108 | return EFI_SUCCESS; | |
109 | } | |
110 | ||
111 | CurrentLink = CurrentLink->ForwardLink; | |
112 | } | |
113 | ||
114 | return EFI_INVALID_PARAMETER; | |
115 | } | |
116 | ||
117 | EFI_STATUS | |
118 | AddDriver ( | |
119 | IN PCI_IO_DEVICE *PciIoDevice, | |
120 | IN EFI_HANDLE DriverImageHandle | |
121 | ) | |
122 | /*++ | |
123 | ||
124 | Routine Description: | |
125 | ||
126 | Add a overriding driver image | |
127 | ||
128 | Arguments: | |
129 | ||
130 | Returns: | |
131 | ||
132 | None | |
133 | ||
134 | --*/ | |
135 | ||
136 | { | |
137 | EFI_STATUS Status; | |
138 | EFI_LOADED_IMAGE_PROTOCOL *LoadedImage; | |
139 | PE_COFF_LOADER_IMAGE_CONTEXT ImageContext; | |
140 | PCI_DRIVER_OVERRIDE_LIST *Node; | |
141 | ||
142 | Status = gBS->HandleProtocol (DriverImageHandle, &gEfiLoadedImageProtocolGuid, (VOID **) &LoadedImage); | |
143 | if (EFI_ERROR (Status)) { | |
144 | return Status; | |
145 | } | |
146 | ||
147 | Node = AllocatePool (sizeof (PCI_DRIVER_OVERRIDE_LIST)); | |
148 | if (Node == NULL) { | |
149 | return EFI_OUT_OF_RESOURCES; | |
150 | } | |
151 | ||
152 | Node->Signature = DRIVER_OVERRIDE_SIGNATURE; | |
153 | Node->DriverImageHandle = DriverImageHandle; | |
154 | ||
155 | InsertTailList (&PciIoDevice->OptionRomDriverList, &(Node->Link)); | |
156 | ||
157 | PciIoDevice->BusOverride = TRUE; | |
158 | ||
159 | ||
160 | ImageContext.Handle = LoadedImage->ImageBase; | |
161 | ImageContext.ImageRead = PeCoffLoaderImageReadFromMemory; | |
162 | ||
163 | // | |
164 | // Get information about the image | |
165 | // | |
166 | Status = PeCoffLoaderGetImageInfo (&ImageContext); | |
167 | if (EFI_ERROR (Status)) { | |
168 | return EFI_SUCCESS; | |
169 | } | |
170 | ||
171 | if (ImageContext.Machine != EFI_IMAGE_MACHINE_EBC) { | |
172 | return EFI_SUCCESS; | |
173 | } | |
174 | ||
175 | return EFI_SUCCESS; | |
176 | } |