framework crb: support socket paramater when require all logic cores

When case get lcore by get_core_list function, socket parameter should also
workable even requiring for all lcores.

Signed-off-by: Marvin Liu <yong.liu@intel.com>

1 files changed, 8 insertions, 1 deletions

diff --git a/framework/crb.py b/framework/crb.py
index a8e308a..9b41a0f 100644
--- a/framework/crb.py
+++ b/framework/crb.py
@@ -605,7 +605,14 @@ class Crb(object):
         We can specify the physical CPU socket by paramter "socket".
         """
         if config == 'all':
-            return [n['thread'] for n in self.cores]
+            cores = []
+            if socket != -1:
+                for core in self.cores:
+                    if int(core['socket']) == socket:
+                        cores.append(core['thread'])
+            else:
+                cores = [core['thread'] for core in self.cores]
+            return cores
 
         m = re.match("([1234])S/([0-9]+)C/([12])T", config)