original files of m241

components/ip_net2-6.9/ipcom/src/ipcom_cmd_cpu.c

@@ -0,0 +1,188 @@
+/*
+ ******************************************************************************
+ *                     INTERPEAK SOURCE FILE
+ *
+ *   Document no: @(#) $Name: VXWORKS_ITER40_2019102207 $ $RCSfile: ipcom_cmd_cpu.c,v $ $Revision: 1.4 $
+ *   $Source: /home/interpeak/CVSRoot/ipcom/src/ipcom_cmd_cpu.c,v $
+ *   $Author: kenneth $
+ *   $State: Exp $ $Locker:  $
+ *
+ *   INTERPEAK_COPYRIGHT_STRING
+ *   Design and implementation by Markus Carlstedt <markus.carlstedt@windriver.com>
+ ******************************************************************************
+ */
+
+
+/*
+ ****************************************************************************
+ * 1                    DESCRIPTION
+ ****************************************************************************
+ */
+
+/*
+ ****************************************************************************
+ * 2                    CONFIGURATION
+ ****************************************************************************
+ */
+
+#include "ipcom_config.h"
+
+
+/*
+ ****************************************************************************
+ * 3                    INCLUDE FILES
+ ****************************************************************************
+ */
+
+
+#define IPCOM_USE_CLIB_PROTO
+#include "ipcom_type.h"
+#include "ipcom_cstyle.h"
+#include "ipcom_clib.h"
+#include "ipcom_os.h"
+#include "ipcom_err.h"
+
+
+/*
+ ****************************************************************************
+ * 4                    DEFINES
+ ****************************************************************************
+ */
+
+/*
+ ****************************************************************************
+ * 5                    TYPES
+ ****************************************************************************
+ */
+
+/*
+ ****************************************************************************
+ * 6                    EXTERN PROTOTYPES
+ ****************************************************************************
+ */
+
+/*
+ ****************************************************************************
+ * 7                    LOCAL PROTOTYPES
+ ****************************************************************************
+ */
+
+IP_PUBLIC int ipcom_cmd_cpu(int argc, char **argv);
+
+
+/*
+ ****************************************************************************
+ * 8                    DATA
+ ****************************************************************************
+ */
+
+/*
+ ****************************************************************************
+ * 9                    LOCAL FUNCTIONS
+ ****************************************************************************
+ */
+
+/*
+ *===========================================================================
+ *                    ipcom_cmd_cpu_getpid
+ *===========================================================================
+ * Description:
+ * Parameters:
+ * Returns:
+ */
+IP_STATIC Ip_pid_t
+ipcom_cmd_cpu_getpid(char *pidstr)
+{
+    if (pidstr[0] == '0' && pidstr[1] == 'x')
+        return (Ip_pid_t)ipcom_strtoul(pidstr, IP_NULL, 16);
+    else
+        return (Ip_pid_t)ipcom_strtoul(pidstr, IP_NULL, 10);
+}
+
+
+/*
+ ****************************************************************************
+ * 10                    PUBLIC FUNCTIONS
+ ****************************************************************************
+ */
+
+/*
+ *===========================================================================
+ *                    ipcom_cmd_cpu
+ *===========================================================================
+ * Description:     Shell command: set/get cpu affinity
+ * Parameters:      argc - number of arguments
+ *                  argv - pointer to pointer to the arguments
+ * Returns:         0
+ *
+ */
+IP_PUBLIC int
+ipcom_cmd_cpu(int argc, char **argv)
+{
+    Ip_pid_t pid;
+    Ip_u32   mask;
+    Ip_cpu_set_t cpuset;
+    Ip_err ret;
+    Ip_bool all = IP_FALSE;
+
+    if (argc < 3)
+    {
+        if (argc == 2 && ipcom_strcmp(argv[1], "num") == 0)
+            ipcom_printf("number of CPUs: configured %d online %d\n",
+                         ipcom_num_configured_cpus(),
+                         ipcom_num_online_cpus());
+        else
+            goto usage;
+    }
+    else
+    {
+        IP_CPU_ZERO(&cpuset);
+        if (ipcom_strcmp(argv[1], "get") == 0)
+        {
+            pid = ipcom_cmd_cpu_getpid(argv[2]);
+            ret = ipcom_proc_cpu_affinity_get(pid, &cpuset);
+            if (ret != IPCOM_SUCCESS)
+                ipcom_printf("failed to get cpu affinity for pid %x\n", pid);
+            else
+                ipcom_printf("cpu affinity is 0x%x for pid %x\n", cpuset.set[0], pid);
+        }
+        else if (ipcom_strcmp(argv[1], "set") == 0)
+        {
+            if (argc < 4)
+                goto usage;
+            if (ipcom_strcmp(argv[2], "all") == 0)
+                all = IP_TRUE;
+            else
+                pid = ipcom_cmd_cpu_getpid(argv[2]);
+            mask = (Ip_u32)ipcom_strtol(argv[3], IP_NULL, 16);
+            cpuset.set[0] = mask;
+            if (all == IP_TRUE)
+            {
+                ret = ipcom_proc_cpu_affinity_set_all(&cpuset);
+                if (ret != IPCOM_SUCCESS)
+                    ipcom_printf("failed to set cpu affinity to 0x%x for all pids\n",  mask);
+            }
+            else
+            {
+                ret = ipcom_proc_cpu_affinity_set(pid, &cpuset);
+                if (ret != IPCOM_SUCCESS)
+                    ipcom_printf("failed to set cpu affinity to 0x%x for pid %x\n",  mask, pid);
+            }
+        }
+        else
+            goto usage;
+    }
+    return 0;
+
+usage:
+    ipcom_printf("usage: 'cpu <num | get <pid> | set <pid> <mask (in hex)>>'\n");
+    return 0;
+}
+
+
+/*
+ ****************************************************************************
+ *                      END OF FILE
+ ****************************************************************************
+ */
+