Конфликтующие типы при компиляции оболочки LD_PRELOAD

Основная проблема, с которой вы столкнулись, заключается в том, что ваш прототип для sprintf не соответствует официальному. Ваша функция имеет эту подпись:

int sprintf (char * src , const char *  format , char* argp);

В то время как официальный имеет:

int sprintf(char *str, const char *format, ...);

Вам нужно будет изменить свою функцию, чтобы иметь эту подпись. Как только вы это сделаете, вам нужно будет использовать va_list, чтобы получить вариативный аргумент. Затем вы должны использовать это для вызова vsprintf, который принимает аргумент этого типа, вместо использования dlsym для загрузки sprintf.

#include <stdio.h>
#include <stdarg.h>

int sprintf (char * src , const char *  format , ...)
{
    va_list args;
    va_start(args, format);
    int ret = vsprintf(src, format, args);
    va_end(args);

    FILE* output = fopen("log.txt","a");
    fprintf(output,"%s \n",src);
    fclose(output);
    return ret;
}

Первое решение Алекса хорошо решает одну проблему: конфликтующие объявления sprintf (хотя нет причин не использовать ту же подпись, что и в stdio.h, см. ответ dbush) . Однако даже в этом случае остается один большой слон: sprintf — это вариационная функция.

Это означает, что всякий раз, когда обернутая программа вызывает sprintf с чем-либо, кроме одного char * третьего аргумента, ваш вывод может быть неверным (и даже может зависеть от уровня -O вашего компилятора).

Вызов вариативных функций из вариативных функций (что, по сути, вы здесь и делаете) представляет собой известная проблема. Любое решение будет непереносимым. С gcc вы можете использовать __buitlin_apply и использовать собственный частный способ gcc обработки списков аргументов:

/* sprintf.c, compile with gcc -Wall -fPIC -shared -o sprintf.so sprintf.c -ldl 
   and use with LD_PRELOAD=./sprintf.so <program> */

#define _GNU_SOURCE
#define sprintf xsprintf
#include <stdio.h>
#include<dlfcn.h>
#undef sprintf

# define ENOUGH 100 /* how many bytes of our call stack 
                       to pass to the original function */

int sprintf (char *src) /* only needs the first argument */                                                                                     
{                                                                                                                                                   
  void *original_func = dlsym(RTLD_NEXT,"sprintf");                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                      
  void *arg = __builtin_apply_args();                                                                                                             
  void *ret = __builtin_apply((void *)original_func, arg, ENOUGH);                                                                                   
                                                                                                                                                
  FILE* output = fopen("log.txt","a");                                                                                                            
  fprintf(output,"%s \n",src);                                                                                                                    
  fclose(output);                                                                                                                                 
 __builtin_return(ret);                                                                                                                         
                                                                                                                                                
}               

Несколько замечаний:

• В хорошо спроектированных библиотеках функции с переменным числом аргументов будут иметь аналог без переменных, который использует один аргумент va_list вместо переменного числа аргументов. В этом случае (и sprintf - vsprintf является таким случаем) вы можете использовать второе решение Алекса (портативное) с макросами va_*. Если нет, то решение с __builtin_apply() является единственно возможным, хотя и специфичным для gcc.
• См. также: вызов printf с использованием va_list
• Возможно, в зависимости от версии компилятора, при компиляции main.c с флагом -O2 main() фактически вызовет __sprintf_chk() вместо sprintf() (независимо от -fno-builtin), и оболочка не будет работать. Чтобы продемонстрировать оболочку, скомпилируйте main.c с -O0. Менять основную программу, чтобы заставить обертку работать, — это, конечно, хвост, виляющий собакой. Это показывает хрупкость построения оболочек: программы часто вызывают не те библиотечные функции, которые вы ожидаете. ltrace <program> заранее может сэкономить много работы....

Вы можете переименовать символ в stdio, но тогда возникает другая проблема: компиляторы, такие как gcc, используют встроенные реализации, если вы не передадите флаг, например -fno-builtin , компилятор будет генерировать код, встроенный в исполняемый файл, он не будет связывать какую-либо библиотеку для таких функций, как sprintf.

sprintf.c:

#define _GNU_SOURCE
#define sprintf xsprintf
#include <stdio.h>
#include<dlfcn.h>
#undef sprintf

int sprintf (char * src , const char *  format , char* argp)
{
    int (*original_func)(char*,const char * , char*);
    original_func = dlsym(RTLD_NEXT,"sprintf");
    int ret = (*original_func)(src ,format,argp);
    FILE* output = fopen("log.txt","a");
    fprintf(output,"%s \n",src);
    fclose(output);
    return ret;
}

основной.с:

#include <stdio.h>

int main(int argc, char *argv[]) {
    char buffer[80];
    sprintf(buffer, "hello world");
    puts(buffer);
    return 0;
}

Makefile:

all: libsprintf.so main

main: main.c
    gcc -Wall -O2 -fno-builtin -o main main.c

libsprintf.so: sprintf.c
    gcc -Wall -O2 -shared -fPIC -fno-builtin -o libsprintf.so sprintf.c -ldl

.PHONY: clean
clean:
    -rm -f main libsprintf.so

Применение:

make
LD_PRELOAD=./libsprintf.so ./main

изменить

sprintf.c с вариативной реализацией (он не вызывает sprintf):

#define _GNU_SOURCE
#define sprintf xsprintf
#include <stdio.h>
#include<dlfcn.h>
#undef sprintf
#include <stdarg.h>

int sprintf (char * src , const char *  fmt , ...)
{
    va_list args;
    va_start(args, fmt);
    int ret = vsprintf(src, fmt, args);
    va_end(args);

    FILE* output = fopen("log.txt","a");
    fprintf(output,"%s \n", src);
    fclose(output);
    return ret;
}