hc
2024-01-03 2f7c68cb55ecb7331f2381deb497c27155f32faf
kernel/drivers/misc/ocxl/config.c
....@@ -204,6 +204,18 @@
204204 return 0;
205205 }
206206
207
+/**
208
+ * get_dvsec_vendor0() - Find a related PCI device (function 0)
209
+ * @dev: PCI device to match
210
+ * @dev0: The PCI device (function 0) found
211
+ * @out_pos: The position of PCI device (function 0)
212
+ *
213
+ * Returns 0 on success, negative on failure.
214
+ *
215
+ * NOTE: If it's successful, the reference of dev0 is increased,
216
+ * so after using it, the callers must call pci_dev_put() to give
217
+ * up the reference.
218
+ */
207219 static int get_dvsec_vendor0(struct pci_dev *dev, struct pci_dev **dev0,
208220 int *out_pos)
209221 {
....@@ -213,10 +225,14 @@
213225 dev = get_function_0(dev);
214226 if (!dev)
215227 return -1;
228
+ } else {
229
+ dev = pci_dev_get(dev);
216230 }
217231 pos = find_dvsec(dev, OCXL_DVSEC_VENDOR_ID);
218
- if (!pos)
232
+ if (!pos) {
233
+ pci_dev_put(dev);
219234 return -1;
235
+ }
220236 *dev0 = dev;
221237 *out_pos = pos;
222238 return 0;
....@@ -233,6 +249,7 @@
233249
234250 pci_read_config_dword(dev0, pos + OCXL_DVSEC_VENDOR_RESET_RELOAD,
235251 &reset_reload);
252
+ pci_dev_put(dev0);
236253 *val = !!(reset_reload & BIT(0));
237254 return 0;
238255 }
....@@ -254,6 +271,7 @@
254271 reset_reload &= ~BIT(0);
255272 pci_write_config_dword(dev0, pos + OCXL_DVSEC_VENDOR_RESET_RELOAD,
256273 reset_reload);
274
+ pci_dev_put(dev0);
257275 return 0;
258276 }
259277