MASM无法识别我的TLS回调

时间:2017-09-21 10:59:32

标签: assembly callback linker masm ida

我已经制作了我的.C源文件的汇编列表。在C源代码中我实现了这样的:

char *msg = "callback";
void NTAPI tls_callback(PVOID DllHandle, DWORD dwReason, PVOID lpVd)
{
    MessageBoxA(0,msg,msg,0);

}

#ifdef _WIN64
#pragma comment (linker, "/INCLUDE:_tls_used")
#pragma comment (linker, "/INCLUDE:tls_callback_func") 
#else
#pragma comment (linker, "/INCLUDE:__tls_used")
#pragma comment (linker, "/INCLUDE:_tls_callback_func")
#endif

#ifdef _WIN64
#pragma const_seg(".CRT$XLF")
EXTERN_C const
#else
#pragma data_seg(".CRT$XLF")
EXTERN_C
#endif
PIMAGE_TLS_CALLBACK tls_callback_func = tls_callback;
#ifdef _WIN64
#pragma const_seg()
#else
#pragma data_seg()
#endif

__declspec(thread) char *tlsData = "tls static data";

我已经制作了这个C文件的汇编列表,现在看起来像这样:

PUBLIC  _tls_callback@12
PUBLIC  _tls_callback_func
PUBLIC  _tlsData

_TLS    SEGMENT
_tlsData 
    DB  00H
    DB  00H
    DB  00H
    DB  00H
    DB  00H
    DB  40H
    DB  30H
    DB  80H
_TLS    ENDS
CRT$XLF SEGMENT
_tls_callback_func DD FLAT:_tls_callback@12
CRT$XLF ENDS


_TEXT   SEGMENT
_DllHandle$ = 8                     ; size = 4
_dwReason$ = 12                     ; size = 4
_lpVd$ = 16                     ; size = 4
_tls_callback@12 PROC                   ; COMDAT

    push    ebp
    mov ebp, esp

    mov edx, DWORD PTR _msg

    push 0
    push edx
    push edx
    push 0
    call DWORD PTR __imp__MessageBoxA@16
; Line 34
    pop ebp
    ret 12                  ; 0000000cH
_tls_callback@12 ENDP
_TEXT   ENDS

我没有看到tls模式产生但是,我在IDA PRO中查找模式应该是:

.rdata:004921A8 __tls_used      dd offset __tls_start  
.rdata:004921AC TlsEnd_ptr      dd offset __tls_end
.rdata:004921B0 TlsIndex_ptr    dd offset __tls_index
.rdata:004921B4 TlsCallbacks_ptr dd offset _tls_callback_func
.rdata:004921B8 TlsSizeOfZeroFill dd 0
.rdata:004921BC TlsCharacteristics dd 100000h

那么我需要定义一个新的tls段并将那些模式放在那里吗?或者它应该在数据部分?

我这样编译:

ml.exe listing.asm /coff

我在ida pro中查找了生成的文件,我看到根本没有生成tls目录,我如何告诉masm或其链接器创建目录?

1 个答案:

答案 0 :(得分:2)

Microsoft链接器使用符号__tls_used(在x86系统上)或_tls_used(在非x86系统上)作为TLS目录的地址。 TLS目录包含指向零终止TLS回调数组的指针。因此,通过创建合适的TLS目录并为其命名__tls_used / _tls_used,您可以在汇编代码中使用TLS回调函数。

这是一个示例程序,用于演示:

    PUBLIC  __tls_used
    PUBLIC  start

    EXTERN  __imp__GetStdHandle@4:DWORD, __imp__WriteFile@20:DWORD, __imp__ExitProcess@4:DWORD

_BSS    SEGMENT PUBLIC DWORD FLAT 
tls_index DD    ?
_BSS    ENDS

_RDATA  SEGMENT PUBLIC PARA FLAT ALIAS('.rdata') READ

__tls_used:
    DD  OFFSET tls_start
    DD  OFFSET tls_end
    DD  OFFSET tls_index
    DD  OFFSET tls_callbacks
    DD  tls_bss_end - tls_end
    DD  0   ; chracterstics

tls_callbacks:
    DD  OFFSET tls_callback
    DD  0

main_msg:
    DB  "Main entry called.", 13, 10
main_msg_len = $ - main_msg

_RDATA  ENDS

_DATA   SEGMENT PUBLIC FLAT
tls_cb_msg:
    DB  "TLS callback called. Reason: 0", 13, 10
tls_cb_msg_len = $ - tls_cb_msg
_DATA   ENDS


_TLS    SEGMENT PUBLIC DWORD FLAT ALIAS('.tls')
tls_start:  
    ; put initialized TLS variable definitions here
tls_end:
    ; put uninitialized TLS variable definitions here
tls_bss_end:
_TLS    ENDS

_TEXT   SEGMENT PUBLIC PARA 'CODE' FLAT
    ASSUME  DS:FLAT

tls_callback:
    mov eax, [esp + 8]
    add BYTE PTR [tls_cb_msg + tls_cb_msg_len - 3], al

    push    -11 ; nStdHandle
    call    [__imp__GetStdHandle@4]

    push    0   ; lpOverlapped
    push    0   ; lpNumberOfBytesWritten
    pushd   tls_cb_msg_len ; nNumberOfBytesToWrite
    push    tls_cb_msg ; lpBuffer
    push    eax ; hFile
    call    [__imp__WriteFile@20]
    ret 12

start:
    push    -11 ; nStdHandle
    call    [__imp__GetStdHandle@4]

    push    0   ; lpOverlapped
    push    0   ; lpNumberOfBytesWritten
    pushd   main_msg_len ; nNumberOfBytesToWrite
    push    main_msg ; lpBuffer
    push    eax ; hFile
    call    [__imp__WriteFile@20]

    push    0   ; uExitCode
    call    [__imp__ExitProcess@4]
    int 3

_TEXT   ENDS

    END start

请注意,上面的代码与TLS的Visual C ++运行时(CRT)实现不兼容。如果您正在计划使用C ++代码,那么您希望让CRT提供TLS目录和其他信息。你可以告诉它使用你的一个回调,在.CRT$XL?部分放置一个指向该函数的指针,其中问号?B到{{1}的字母替换}。使用字母Y将导致在调用CRT TLS回调之前调用它。使用BD的字母会将其调用。所以你想要的代码就像:

Y