1// Copyright (c) 2010, Google Inc.
2// All rights reserved.
3//
4// Redistribution and use in source and binary forms, with or without
5// modification, are permitted provided that the following conditions are
6// met:
7//
8//     * Redistributions of source code must retain the above copyright
9// notice, this list of conditions and the following disclaimer.
10//     * Redistributions in binary form must reproduce the above
11// copyright notice, this list of conditions and the following disclaimer
12// in the documentation and/or other materials provided with the
13// distribution.
14//     * Neither the name of Google Inc. nor the names of its
15// contributors may be used to endorse or promote products derived from
16// this software without specific prior written permission.
17//
18// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29//
30// Helper program for the linux_dumper class, which creates a bunch of
31// threads. The first word of each thread's stack is set to the thread
32// id.
33
34#include <pthread.h>
35#include <stdint.h>
36#include <stdio.h>
37#include <stdlib.h>
38#include <sys/syscall.h>
39#include <unistd.h>
40
41#include "common/scoped_ptr.h"
42#include "third_party/lss/linux_syscall_support.h"
43
44#if defined(__ARM_EABI__)
45#define TID_PTR_REGISTER "r3"
46#elif defined(__aarch64__)
47#define TID_PTR_REGISTER "x3"
48#elif defined(__i386)
49#define TID_PTR_REGISTER "ecx"
50#elif defined(__x86_64)
51#define TID_PTR_REGISTER "rcx"
52#elif defined(__mips__)
53#define TID_PTR_REGISTER "$1"
54#else
55#error This test has not been ported to this platform.
56#endif
57
58void *thread_function(void *data) {
59  int pipefd = *static_cast<int *>(data);
60  volatile pid_t thread_id = syscall(__NR_gettid);
61  // Signal parent that a thread has started.
62  uint8_t byte = 1;
63  if (write(pipefd, &byte, sizeof(byte)) != sizeof(byte)) {
64    perror("ERROR: parent notification failed");
65    return NULL;
66  }
67  register volatile pid_t *thread_id_ptr asm(TID_PTR_REGISTER) = &thread_id;
68  while (true)
69    asm volatile ("" : : "r" (thread_id_ptr));
70  return NULL;
71}
72
73int main(int argc, char *argv[]) {
74  if (argc < 3) {
75    fprintf(stderr,
76            "usage: linux_dumper_unittest_helper <pipe fd> <# of threads>\n");
77    return 1;
78  }
79  int pipefd = atoi(argv[1]);
80  int num_threads = atoi(argv[2]);
81  if (num_threads < 1) {
82    fprintf(stderr, "ERROR: number of threads is 0");
83    return 1;
84  }
85  google_breakpad::scoped_array<pthread_t> threads(new pthread_t[num_threads]);
86  pthread_attr_t thread_attributes;
87  pthread_attr_init(&thread_attributes);
88  pthread_attr_setdetachstate(&thread_attributes, PTHREAD_CREATE_DETACHED);
89  for (int i = 1; i < num_threads; i++) {
90    pthread_create(&threads[i], &thread_attributes, &thread_function, &pipefd);
91  }
92  thread_function(&pipefd);
93  return 0;
94}
95