Как определить, выполняется ли код в контексте обработчика сигналов?

Судя по всему, более новый Linux/x86 (вероятно, начиная с ядра 2.6.x) вызывает обработчики сигналов из файла vdso. Вы можете использовать этот факт, чтобы нанести следующий ужасный удар по ничего не подозревающему миру:

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <signal.h>

#include <unistd.h>

uintmax_t vdso_start = 0;
uintmax_t vdso_end = 0;             /* actually, next byte */

int check_stack_for_vdso(uint32_t *esp, size_t len)
{
    size_t i;

    for (i = 0; i < len; i++, esp++)
            if (*esp >= vdso_start && *esp < vdso_end)
                    return 1;

    return 0;
}

void handler(int signo)
{
    uint32_t *esp;

    __asm__ __volatile__ ("mov %%esp, %0" : "=r"(esp));
    /* XXX only for demonstration, don't call printf from a signal handler */
    printf("handler: check_stack_for_vdso() = %d\n", check_stack_for_vdso(esp, 20));
}

void parse_maps()
{
    FILE *maps;
    char buf[256];
    char path[7];
    uintmax_t start, end, offset, inode;
    char r, w, x, p;
    unsigned major, minor;

    maps = fopen("/proc/self/maps", "rt");
    if (maps == NULL)
            return;

    while (!feof(maps) && !ferror(maps)) {
            if (fgets(buf, 256, maps) != NULL) {
                    if (sscanf(buf, "%jx-%jx %c%c%c%c %jx %u:%u %ju %6s",
                                    &start, &end, &r, &w, &x, &p, &offset,
                                    &major, &minor, &inode, path) == 11) {
                            if (!strcmp(path, "[vdso]")) {
                                    vdso_start = start;
                                    vdso_end = end;
                                    break;
                            }
                    }
            }
    }

    fclose(maps);

    printf("[vdso] at %jx-%jx\n", vdso_start, vdso_end);
}

int main()
{
    struct sigaction sa;
    uint32_t *esp;

    parse_maps();
    memset(&sa, 0, sizeof(struct sigaction));
    sa.sa_handler = handler;
    sa.sa_flags = SA_RESTART;

    if (sigaction(SIGUSR1, &sa, NULL) < 0) {
            perror("sigaction");
            exit(1);
    }

    __asm__ __volatile__ ("mov %%esp, %0" : "=r"(esp));
    printf("before kill: check_stack_for_vdso() = %d\n", check_stack_for_vdso(esp, 20));

    kill(getpid(), SIGUSR1);

    __asm__ __volatile__ ("mov %%esp, %0" : "=r"(esp));
    printf("after kill: check_stack_for_vdso() = %d\n", check_stack_for_vdso(esp, 20));

    return 0;
}

ГКНР.

Если мы можем предположить, что ваше приложение не блокирует вручную сигналы с помощью sigprocmask() или pthread_sigmask(), то это довольно просто: получить идентификатор текущего потока (tid). Откройте /proc/tid/status и получите значения для SigBlk и SigCgt. AND эти два значения. Если результат этого AND не равен нулю, то этот поток в настоящее время выполняется внутри обработчика сигнала. Я проверил это на себе, и это работает.

Есть два правильных способа справиться с этим:

• Пусть ваши коллеги перестанут делать неправильные вещи. Однако удачи в том, чтобы справиться с боссом...

• Сделайте вашу функцию реентерабельной и асинхронно-безопасной. При необходимости предоставьте функцию с другой сигнатурой (например, с использованием широко используемого соглашения об именах *_r) с дополнительными аргументами, необходимыми для сохранения состояния.

Что касается неправильного способа сделать это, в Linux с GNU libc вы можете использовать backtrace() и друзья, чтобы просмотреть список абонентов вашей функции. Это нелегко сделать правильно, безопасно или портативно, но на какое-то время это может подойти:

/*
 * *** Warning ***
 *
 * Black, fragile and unportable magic ahead
 *
 * Do not use this, lest the daemons of hell be unleashed upon you
 */
int in_signal_handler_context() {
        int i, n;
        void *bt[1000];
        char **bts = NULL;

        n = backtrace(bt, 1000);
        bts = backtrace_symbols(bt, n);

        for (i = 0; i < n; ++i)
                printf("%i - %s\n", i, bts[i]);

        /* Have a look at the caller chain */
        for (i = 0; i < n; ++i) {
                /* Far more checks are needed here to avoid misfires */
                if (strstr(bts[i], "(__libc_start_main+") != NULL)
                        return 0;
                if (strstr(bts[i], "libc.so.6(+") != NULL)
                        return 1;
        }

        return 0;
}


void unsafe() {
        if (in_signal_handler_context())
                printf("John, you know you are an idiot, right?\n");
}

На мой взгляд, лучше просто выйти, чем писать этот код.

Вы можете что-то придумать, используя sigaltstack. Настройте альтернативный стек сигналов, получите указатель стека каким-либо асинхронно-безопасным способом, если в альтернативном стеке продолжите, в противном случае abort().

Я думаю, вам нужно сделать следующее. Это комплексное решение, сочетающее в себе лучшие практики не только программирования, но и разработки программного обеспечения!

1. Убедите своего начальника, что соглашение об именовании обработчиков сигналов — это хорошо. Предложите, например, венгерскую нотацию и сообщите, что она использовалась в Майкрософт с большим успехом. Таким образом, все обработчики сигналов будут начинаться с sighnd, например sighndInterrupt.
2. Your function that detects signal handling context would do the following:
   1. Get the backtrace().
   2. Посмотрите, не начинаются ли в нем какие-либо функции с sighnd.... Если да, то поздравляю, вы внутри обработчика сигнала!
   3. В противном случае, вы не.
3. Старайтесь не работать с Джимми в одной компании. "Может быть только один", вы знаете.

для кода, оптимизированного для -O2 или лучше (istr), необходимо добавить -fno-omit-frame-pointer

иначе gcc оптимизирует контекстную информацию стека