From d6ac4187fcb0bae48e421cdb265dd1636543a996 Mon Sep 17 00:00:00 2001
From: Joan Lledó <jlledom@member.fsf.org>
Date: Sat, 15 Feb 2020 11:52:21 +0100
Subject: pci-arbiter: Add --device command line option

Shortcut for -d, -b, -s and -f
Usage: --device [<domain>:]<bus>:<slot>.<func>
E.G. --device 00:05.0 is shortcut for -d 0 -b 0 -s 5 -f 0

* pci-arbiter/options.c: Implement --slot option
* pci-arbiter/options.h: Add --slot to options list
---
 pci-arbiter/options.c | 57 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)

(limited to 'pci-arbiter/options.c')

diff --git a/pci-arbiter/options.c b/pci-arbiter/options.c
index 01686fcd..2023fd9e 100644
--- a/pci-arbiter/options.c
+++ b/pci-arbiter/options.c
@@ -27,9 +27,13 @@
 #include <argp.h>
 #include <argz.h>
 #include <error.h>
+#include <regex.h>
 
 #include "pcifs.h"
 
+#define PCI_SLOT_REGEX "^(([0-9a-fA-F]{4}):)?([0-9a-fA-F]{2}):([0-9a-fA-F]{2})\\.([0-7])$"
+#define PCI_SLOT_REGEX_GROUPS 6	// 2: Domain, 3: Bus, 4: Dev, 5: Func
+
 /* Fsysopts and command line option parsing */
 
 /* Adds an empty interface slot to H, and sets H's current interface to it, or
@@ -91,6 +95,9 @@ parse_opt (int opt, char *arg, struct argp_state *state)
 {
   error_t err = 0;
   struct parse_hook *h = state->hook;
+  regex_t slot_regex;
+  regmatch_t slot_regex_groups[PCI_SLOT_REGEX_GROUPS];
+  char regex_group_val[5];
 
   /* Return _ERR from this routine */
 #define RETURN(_err)                          \
@@ -111,6 +118,11 @@ parse_opt (int opt, char *arg, struct argp_state *state)
       state->next++;
     }
 
+  /* Compile regular expression to check --slot option */
+  err = regcomp (&slot_regex, PCI_SLOT_REGEX, REG_EXTENDED);
+  if (err)
+    FAIL (err, 1, err, "option parsing");
+
   switch (opt)
     {
     case 'C':
@@ -134,6 +146,48 @@ parse_opt (int opt, char *arg, struct argp_state *state)
     case 'f':
       h->curset->func = strtol (arg, 0, 16);
       break;
+    case 'D':
+      err =
+	regexec (&slot_regex, arg, PCI_SLOT_REGEX_GROUPS, slot_regex_groups,
+		 0);
+      if (!err)
+	{
+	  // Domain, 0000 by default
+	  if (slot_regex_groups[2].rm_so >= 0)
+	    {
+	      strncpy (regex_group_val, arg + slot_regex_groups[2].rm_so, 4);
+	      regex_group_val[4] = 0;
+	    }
+	  else
+	    {
+	      strncpy (regex_group_val, "0000", 5);
+	    }
+
+	  h->curset->domain = strtol (regex_group_val, 0, 16);
+
+	  // Bus
+	  strncpy (regex_group_val, arg + slot_regex_groups[3].rm_so, 2);
+	  regex_group_val[2] = 0;
+
+	  h->curset->bus = strtol (regex_group_val, 0, 16);
+
+	  // Dev
+	  strncpy (regex_group_val, arg + slot_regex_groups[4].rm_so, 2);
+	  regex_group_val[2] = 0;
+
+	  h->curset->dev = strtol (regex_group_val, 0, 16);
+
+	  // Func
+	  regex_group_val[0] = arg[slot_regex_groups[5].rm_so];
+	  regex_group_val[1] = 0;
+
+	  h->curset->func = strtol (regex_group_val, 0, 16);
+	}
+      else
+	{
+	  PERR (err, "Wrong PCI slot. Format: [<domain>:]<bus>:<dev>.<func>");
+	}
+      break;
     case 'U':
       if (h->curset->uid >= 0)
 	parse_hook_add_set (h);
@@ -227,6 +281,9 @@ parse_opt (int opt, char *arg, struct argp_state *state)
       return ARGP_ERR_UNKNOWN;
     }
 
+  /* Free allocated regular expression for the --slot option */
+  regfree (&slot_regex);
+
   return err;
 }
 
-- 
cgit v1.2.3