You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
200 lines
3.5 KiB
200 lines
3.5 KiB
4 months ago
|
// 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;
|
||
|
}
|