Modify function pci_find_next_cap_ttl to support returning position of next capability when size "n" is zero.
That can help caller to get next capability offset if caller just has a information of current capability offset. That will be used in a follow-on change. Signed-off-by: Jiqian Chen <jiqian.c...@amd.com> --- cc: Andrew Cooper <andrew.coop...@citrix.com> cc: Anthony PERARD <anthony.per...@vates.tech> cc: Michal Orzel <michal.or...@amd.com> cc: Jan Beulich <jbeul...@suse.com> cc: Julien Grall <jul...@xen.org> cc: "Roger Pau Monné" <roger....@citrix.com> cc: Stefano Stabellini <sstabell...@kernel.org> --- v2->v3 changes: * Only check if n == 0 and add assertion for array "caps". * Not to change pci_find_next_cap_ttl definition. v1->v2 changes: new patch. Best regards, Jiqian Chen. --- xen/drivers/pci/pci.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/xen/drivers/pci/pci.c b/xen/drivers/pci/pci.c index edf5b9f7ae9f..804f4e1e6066 100644 --- a/xen/drivers/pci/pci.c +++ b/xen/drivers/pci/pci.c @@ -55,6 +55,9 @@ unsigned int pci_find_next_cap_ttl(pci_sbdf_t sbdf, unsigned int pos, if ( id == 0xff ) break; + if ( n == 0 ) + return pos; + ASSERT(caps); for ( i = 0; i < n; i++ ) { if ( id == caps[i] ) -- 2.34.1