123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199 |
- // Copyright 2015 The Go Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style
- // license that can be found in the LICENSE file.
- // Test that a signal handler works in non-Go code when using
- // os/signal.Notify.
- // This is a lot like misc/cgo/testcarchive/main3.c.
- #include <signal.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <time.h>
- #include <sched.h>
- #include <dlfcn.h>
- static void die(const char* msg) {
- perror(msg);
- exit(EXIT_FAILURE);
- }
- static volatile sig_atomic_t sigioSeen;
- static void ioHandler(int signo, siginfo_t* info, void* ctxt) {
- sigioSeen = 1;
- }
- int main(int argc, char** argv) {
- int verbose;
- struct sigaction sa;
- void* handle;
- void (*fn1)(void);
- int (*sawSIGIO)(void);
- int i;
- struct timespec ts;
- verbose = argc > 2;
- setvbuf(stdout, NULL, _IONBF, 0);
- if (verbose) {
- printf("calling sigaction\n");
- }
- memset(&sa, 0, sizeof sa);
- sa.sa_sigaction = ioHandler;
- if (sigemptyset(&sa.sa_mask) < 0) {
- die("sigemptyset");
- }
- sa.sa_flags = SA_SIGINFO;
- if (sigaction(SIGIO, &sa, NULL) < 0) {
- die("sigaction");
- }
- if (verbose) {
- printf("calling dlopen\n");
- }
- handle = dlopen(argv[1], RTLD_NOW | RTLD_GLOBAL);
- if (handle == NULL) {
- fprintf(stderr, "%s\n", dlerror());
- exit(EXIT_FAILURE);
- }
- // At this point there should not be a Go signal handler
- // installed for SIGIO.
- if (verbose) {
- printf("raising SIGIO\n");
- }
- if (raise(SIGIO) < 0) {
- die("raise");
- }
- if (verbose) {
- printf("waiting for sigioSeen\n");
- }
- // Wait until the signal has been delivered.
- i = 0;
- while (!sigioSeen) {
- ts.tv_sec = 0;
- ts.tv_nsec = 1000000;
- nanosleep(&ts, NULL);
- i++;
- if (i > 5000) {
- fprintf(stderr, "looping too long waiting for signal\n");
- exit(EXIT_FAILURE);
- }
- }
- sigioSeen = 0;
- // Tell the Go code to catch SIGIO.
- if (verbose) {
- printf("calling dlsym\n");
- }
- fn1 = (void(*)(void))dlsym(handle, "CatchSIGIO");
- if (fn1 == NULL) {
- fprintf(stderr, "%s\n", dlerror());
- exit(EXIT_FAILURE);
- }
- if (verbose) {
- printf("calling CatchSIGIO\n");
- }
- fn1();
- if (verbose) {
- printf("raising SIGIO\n");
- }
- if (raise(SIGIO) < 0) {
- die("raise");
- }
- if (verbose) {
- printf("calling dlsym\n");
- }
- // Check that the Go code saw SIGIO.
- sawSIGIO = (int (*)(void))dlsym(handle, "SawSIGIO");
- if (sawSIGIO == NULL) {
- fprintf(stderr, "%s\n", dlerror());
- exit(EXIT_FAILURE);
- }
- if (verbose) {
- printf("calling SawSIGIO\n");
- }
- if (!sawSIGIO()) {
- fprintf(stderr, "Go handler did not see SIGIO\n");
- exit(EXIT_FAILURE);
- }
- if (sigioSeen != 0) {
- fprintf(stderr, "C handler saw SIGIO when only Go handler should have\n");
- exit(EXIT_FAILURE);
- }
- // Tell the Go code to stop catching SIGIO.
- if (verbose) {
- printf("calling dlsym\n");
- }
- fn1 = (void(*)(void))dlsym(handle, "ResetSIGIO");
- if (fn1 == NULL) {
- fprintf(stderr, "%s\n", dlerror());
- exit(EXIT_FAILURE);
- }
- if (verbose) {
- printf("calling ResetSIGIO\n");
- }
- fn1();
- if (verbose) {
- printf("raising SIGIO\n");
- }
- if (raise(SIGIO) < 0) {
- die("raise");
- }
- if (verbose) {
- printf("calling SawSIGIO\n");
- }
- if (sawSIGIO()) {
- fprintf(stderr, "Go handler saw SIGIO after Reset\n");
- exit(EXIT_FAILURE);
- }
- if (verbose) {
- printf("waiting for sigioSeen\n");
- }
- // Wait until the signal has been delivered.
- i = 0;
- while (!sigioSeen) {
- ts.tv_sec = 0;
- ts.tv_nsec = 1000000;
- nanosleep(&ts, NULL);
- i++;
- if (i > 5000) {
- fprintf(stderr, "looping too long waiting for signal\n");
- exit(EXIT_FAILURE);
- }
- }
- printf("PASS\n");
- return 0;
- }
|