commit e4d8b28cbe45df992d1ffb26e0d3623b0cfe0f0b
author Brian Silverman <brian@peloton-tech.com> Thu Dec 24 13:44:48 2015 -0800
committer Brian Silverman <brians> Fri Dec 25 23:53:52 2015 -0800
tree db2f7e02f0510699306025ea88e94231153e77b2
parent 279e2e331f31fc9f39e89a1a0a89e7ac69e95acc

Add a tool for testing various IPC methods

Change-Id: I2fc2e641cee4d26b9ec865f4d7555036def3abfd

aos/linux_code/init.cc

diff --git a/aos/linux_code/init.cc b/aos/linux_code/init.cc
index 80c71e3..09b11a2 100644
--- a/aos/linux_code/init.cc
+++ b/aos/linux_code/init.cc
@@ -146,6 +146,9 @@
 }
 
 void SetCurrentThreadRealtimePriority(int priority) {
+  // Make sure we will only be allowed to run for 3 seconds straight.
+  SetSoftRLimit(RLIMIT_RTTIME, 3000000, true);
+
   struct sched_param param;
   param.sched_priority = priority;
   if (sched_setscheduler(0, SCHED_FIFO, &param) == -1) {
@@ -153,6 +156,13 @@
   }
 }
 
+void PinCurrentThreadToCPU(int number) {
+  cpu_set_t cpuset;
+  CPU_ZERO(&cpuset);
+  CPU_SET(number, &cpuset);
+  PRCHECK(pthread_setaffinity_np(pthread_self(), sizeof(cpuset), &cpuset));
+}
+
 void SetCurrentThreadName(const ::std::string &name) {
   if (name.size() > 16) {
     LOG(FATAL, "thread name '%s' too long\n", name.c_str());