Add test for a pty kernel bug.
Bug: 28979140
Change-Id: Iad79fcb7a3c5b068f625c7902785cc6e696fe9a1
diff --git a/tests/pty_test.cpp b/tests/pty_test.cpp
index 91d1f5e..a371434 100644
--- a/tests/pty_test.cpp
+++ b/tests/pty_test.cpp
@@ -14,11 +14,17 @@
* limitations under the License.
*/
+#include <pty.h>
+
#include <gtest/gtest.h>
-#include <pty.h>
+#include <pthread.h>
#include <sys/ioctl.h>
+#include <atomic>
+
+#include <android-base/file.h>
+
#include "utils.h"
TEST(pty, openpty) {
@@ -64,3 +70,82 @@
close(master);
}
+
+struct PtyReader_28979140_Arg {
+ int slave_fd;
+ uint32_t data_count;
+ bool finished;
+ std::atomic<bool> matched;
+};
+
+static void PtyReader_28979140(PtyReader_28979140_Arg* arg) {
+ arg->finished = false;
+ cpu_set_t cpus;
+ ASSERT_EQ(0, sched_getaffinity(0, sizeof(cpu_set_t), &cpus));
+ CPU_CLR(0, &cpus);
+ ASSERT_EQ(0, sched_setaffinity(0, sizeof(cpu_set_t), &cpus));
+
+ uint32_t counter = 0;
+ while (counter <= arg->data_count) {
+ char buf[4096]; // Use big buffer to read to hit the bug more easily.
+ size_t to_read = std::min(sizeof(buf), (arg->data_count + 1 - counter) * sizeof(uint32_t));
+ ASSERT_TRUE(android::base::ReadFully(arg->slave_fd, buf, to_read));
+ size_t num_of_value = to_read / sizeof(uint32_t);
+ uint32_t* p = reinterpret_cast<uint32_t*>(buf);
+ while (num_of_value-- > 0) {
+ if (*p++ != counter++) {
+ arg->matched = false;
+ }
+ }
+ }
+ close(arg->slave_fd);
+ arg->finished = true;
+}
+
+TEST(pty, bug_28979140) {
+ // This test is to test a kernel bug, which uses a lock free ring-buffer to
+ // pass data through a raw pty, but missing necessary memory barriers.
+ if (sysconf(_SC_NPROCESSORS_ONLN) == 1) {
+ GTEST_LOG_(INFO) << "This test tests bug happens only on multiprocessors.";
+ return;
+ }
+ constexpr uint32_t TEST_DATA_COUNT = 200000;
+
+ // 1. Open raw pty.
+ int master;
+ int slave;
+ ASSERT_EQ(0, openpty(&master, &slave, nullptr, nullptr, nullptr));
+ termios tattr;
+ ASSERT_EQ(0, tcgetattr(slave, &tattr));
+ cfmakeraw(&tattr);
+ ASSERT_EQ(0, tcsetattr(slave, TCSADRAIN, &tattr));
+
+ // 2. Create thread for slave reader.
+ pthread_t thread;
+ PtyReader_28979140_Arg arg;
+ arg.slave_fd = slave;
+ arg.data_count = TEST_DATA_COUNT;
+ arg.matched = true;
+ ASSERT_EQ(0, pthread_create(&thread, nullptr,
+ reinterpret_cast<void*(*)(void*)>(PtyReader_28979140),
+ &arg));
+
+ // 3. Make master thread and slave thread running on different cpus:
+ // master thread uses cpu 0, and slave thread uses other cpus.
+ cpu_set_t cpus;
+ CPU_ZERO(&cpus);
+ CPU_SET(0, &cpus);
+ ASSERT_EQ(0, sched_setaffinity(0, sizeof(cpu_set_t), &cpus));
+
+ // 4. Send data to slave.
+ uint32_t counter = 0;
+ while (counter <= TEST_DATA_COUNT) {
+ ASSERT_TRUE(android::base::WriteFully(master, &counter, sizeof(counter)));
+ ASSERT_TRUE(arg.matched) << "failed at count = " << counter;
+ counter++;
+ }
+ ASSERT_EQ(0, pthread_join(thread, nullptr));
+ ASSERT_TRUE(arg.finished);
+ ASSERT_TRUE(arg.matched);
+ close(master);
+}