summaryrefslogtreecommitdiff
path: root/ldso/ldso/dl-vdso.c
blob: c8d724cd0c92deee8ef494e44cff5e534802a032 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362

#include <elf.h>
//#include <stdio.h>
#include <string.h>

#include "sys/auxv.h"
//#include <linux/time.h>
//#include <time.h>

#include "ldso.h"
#include "generated/autoconf.h"

#ifndef ELF_BITS
# if ULONG_MAX > 0xffffffffUL
#  define ELF_BITS 64
# else
#  define ELF_BITS 32
# endif
#endif

#define ELF_BITS_XFORM2(bits, x) Elf##bits##_##x
#define ELF_BITS_XFORM(bits, x) ELF_BITS_XFORM2(bits, x)
#define ELF(x) ELF_BITS_XFORM(ELF_BITS, x)


#ifndef __VDSO_SUPPORT__
    void load_vdso( uint32_t sys_info_ehdr, char **envp ){
#ifdef __SUPPORT_LD_DEBUG__
        if ( _dl_debug_vdso != 0 ){
            _dl_dprintf(2,"_dl_vdso support not enabled\n" );
        }

#endif
    }
#else



   


//typedef long (*gtod_t)(struct timeval *tv, struct timezone *tz);
void* _dl__vdso_gettimeofday  = 0;


//typedef long (*clock_gettime_t)(int clk_id, struct timespec *tp);
void* _dl__vdso_clock_gettime = 0;



typedef struct{

    void* base_addr;

    ELF(Ehdr) *hdr;

    char* section_header_strtab;

    ELF(Sym) *dynsym_table;
    uint32_t   dynsym_table_num;

    char* dynstr_table;

    uint16_t  *versym_table;

    ELF(Verdef) *verdef_table;
    uint32_t      verdef_num;

    ELF(Dyn) *dynamic_section;
    uint32_t  dynamic_section_num;

    char* text_section;


    char* vers_strings[10];


}elf_infos;

/*
 * the raise() dummy function is needed because of divisons in this code
 * but keep it hidden in this object
 *
 * fixes link error with gcc 12 for arm
 */
#pragma GCC visibility push(hidden)
int raise(int sig){
    sig = sig;
    return 0;
}
#pragma GCC visibility pop


static int vdso_check_elf_header( elf_infos* elf ){
    
    if ( 0 != _dl_memcmp( ELFMAG, elf->base_addr, 4 ) ){
        return 1;
    }
    
    if (elf->hdr->e_ident[EI_CLASS] != (ELF_BITS == 32 ? ELFCLASS32 : ELFCLASS64)) {
        _dl_dprintf(2,"vdso ELF Bits check error\n");
        return 1;  /* Wrong ELF class -- check ELF_BITS */
    }
    
    return 0;
}

static ELF(Shdr) *vdso_get_sec_header( elf_infos* elf, int index ){
    
    return (ELF(Shdr) *) ( elf->base_addr + elf->hdr->e_shoff + ( index * sizeof( ELF(Shdr) )) );
    
}


void load_vdso( uint32_t sys_info_ehdr, char **envp ){

    elf_infos vdso_infos;
    
    
    
    if ( sys_info_ehdr == 0 ){
#ifdef __SUPPORT_LD_DEBUG__
        if ( _dl_debug_vdso != 0 ){
            _dl_dprintf(2,"_dl_vdso no vdso provied by kernel\n" );
        }
#endif
        return;
        
    }
    
    char* _dl_vdso_disable = _dl_getenv("VDSO_DISABLE", envp);
    if ( _dl_vdso_disable != 0 ){
#ifdef __SUPPORT_LD_DEBUG__
        if ( _dl_debug_vdso != 0 ){
            _dl_dprintf(2,"_dl_vdso vdso support disabled\n" );
        }
#endif
        return;
    }
    
    
    _dl_memset( &vdso_infos, 0 , sizeof( elf_infos ) );
    
    
    vdso_infos.base_addr = (void*)sys_info_ehdr;
    vdso_infos.hdr = (ELF(Ehdr)*)vdso_infos.base_addr;
    
    //printf("base : %p\n",vdso_infos.base_addr);
    
    if ( 0 != vdso_check_elf_header( &vdso_infos ) ){
        return;
    }
    
    
    
    ELF(Shdr) *sec_header = vdso_get_sec_header( &vdso_infos, vdso_infos.hdr->e_shstrndx);
    vdso_infos.section_header_strtab = ( vdso_infos.base_addr + sec_header->sh_offset );
    
    /*
     * 
     * load ELF section headers
     * 
     */
    
    for ( int i = 0 ; i < vdso_infos.hdr->e_shnum; i++ ){
    
        sec_header = vdso_get_sec_header( &vdso_infos, i );
        
        char* name = vdso_infos.section_header_strtab + sec_header->sh_name;
        
        if( ( SHT_DYNSYM == sec_header->sh_type ) && ( 0 == _dl_strcmp( ".dynsym",name ) ) ){
            vdso_infos.dynsym_table = ( vdso_infos.base_addr + sec_header->sh_offset );
            vdso_infos.dynsym_table_num = sec_header->sh_size / sec_header->sh_entsize ;
            continue;
        }
        
        if( ( SHT_STRTAB == sec_header->sh_type ) && ( 0 == _dl_strcmp( ".dynstr",name ) ) ){
            vdso_infos.dynstr_table = ( vdso_infos.base_addr + sec_header->sh_offset );
            continue;
        }
        
        if( ( SHT_GNU_versym == sec_header->sh_type ) && ( 0 == _dl_strcmp( ".gnu.version",name ) ) ){
            vdso_infos.versym_table = ( vdso_infos.base_addr + sec_header->sh_offset );
            continue;
        }
        
        if( ( SHT_GNU_verdef == sec_header->sh_type ) && ( 0 == _dl_strcmp( ".gnu.version_d",name ) ) ){
            vdso_infos.verdef_table = ( vdso_infos.base_addr + sec_header->sh_offset );
            continue;
        }
        
        if( ( SHT_DYNAMIC == sec_header->sh_type ) && ( 0 == _dl_strcmp( ".dynamic",name ) ) ){
            vdso_infos.dynamic_section = ( vdso_infos.base_addr + sec_header->sh_offset );
            vdso_infos.dynamic_section_num = sec_header->sh_size / sec_header->sh_entsize ;
            continue;
        }
        
        if( ( SHT_PROGBITS == sec_header->sh_type ) && ( 0 == _dl_strcmp( ".text",name ) ) ){
            vdso_infos.text_section = ( vdso_infos.base_addr + sec_header->sh_offset );
            continue;
        }
       
    }


    /*
     * 
     * check section header -> dynamic table consistence
     * 
     */
    
    
    for( int i = 0 ; i < vdso_infos.dynamic_section_num ; i++ ){
        ELF(Dyn) *dyn_sec = &vdso_infos.dynamic_section[i];
        if ( dyn_sec->d_tag == 0 ) continue;
        
        
        if ( dyn_sec->d_tag == DT_STRTAB ){
            char* strtab = ( vdso_infos.base_addr + dyn_sec->d_un.d_ptr );
            if ( strtab != (char*) vdso_infos.dynstr_table ){
                _dl_dprintf(2,"vdso elf DT_STRTAB check error\n");
                return;
            }
            continue;
        }
        
        if ( dyn_sec->d_tag == DT_SYMTAB ){
            char* symtab = ( vdso_infos.base_addr + dyn_sec->d_un.d_ptr );
            if ( symtab != (char*) vdso_infos.dynsym_table ){
                _dl_dprintf(2,"vdso elf DT_SYMTAB check error\n");
                return;
            }
            continue;
        }
        
        if ( dyn_sec->d_tag == DT_VERDEF ){
            Elf32_Verdef* verdef = ( vdso_infos.base_addr + dyn_sec->d_un.d_ptr );
            if ( verdef != (Elf32_Verdef*) vdso_infos.verdef_table ){
                _dl_dprintf(2,"vdso elf DT_VERDEF check error\n");
                return;
            }
            continue;
        }
        
        if ( dyn_sec->d_tag == DT_VERDEFNUM ){
            vdso_infos.verdef_num = dyn_sec->d_un.d_val;
            continue;
        }
        
        if ( dyn_sec->d_tag == DT_VERSYM ){
            uint16_t*  versym = ( vdso_infos.base_addr + dyn_sec->d_un.d_ptr );
            if ( versym !=  vdso_infos.versym_table ){
                _dl_dprintf(2,"vdso elf DT_VERSYM check error\n");
                return;
            }
            continue;
        }
        
    }
    
    /*
     * 
     * load vdso version definition strings
     * 
     */
    
    ELF(Verdef) *vd = vdso_infos.verdef_table;
    for( int i = 0 ; i < vdso_infos.verdef_num ; i++ ){
        
        ELF(Verdaux) *vd_aux = (ELF(Verdaux)     *)(( ( char*)vd ) + vd->vd_aux);
        
        vdso_infos.vers_strings[ vd->vd_ndx ]   = vdso_infos.dynstr_table + vd_aux->vda_name;
        
        vd = ( ELF(Verdef) *)(( ( char*)vd ) + vd->vd_next);
        
    }
    
    /*
     * 
     * load function from the vdso
     * 
     */
#ifdef __SUPPORT_LD_DEBUG__ 
    if ( _dl_debug_vdso != 0 ){
        int vdso_functions = 0;
        for( int i = 0 ; i < vdso_infos.dynsym_table_num ; i++ ){
            ELF(Sym)* sym = &vdso_infos.dynsym_table[i];
            
            if (ELF64_ST_TYPE(sym->st_info) != STT_FUNC)
                continue;
            
            char* name = vdso_infos.dynstr_table + sym->st_name;
            if ( name[0] == 0 ){
                continue;
            }

            vdso_functions++;
        }
        _dl_dprintf(2,"_dl_vdso_load functions found : %d\n", vdso_functions );
    }

#endif    
    
    for( int i = 0 ; i < vdso_infos.dynsym_table_num ; i++ ){
        ELF(Sym)* sym = &vdso_infos.dynsym_table[i];
        
        if (ELF64_ST_TYPE(sym->st_info) != STT_FUNC)
            continue;



        char* name = vdso_infos.dynstr_table + sym->st_name;
        char* vers = vdso_infos.vers_strings[ vdso_infos.versym_table[i] ];
        void* func_addr = (void*)( vdso_infos.base_addr + sym->st_value );
        
        // the function name is patched to zero if the kernel has no timer which is
        // usable for the function
        if ( name[0] == 0 ){
#ifdef __SUPPORT_LD_DEBUG__
            if ( _dl_debug_vdso != 0 ){
                _dl_dprintf(2,"   function at address %p disabled by the kernel\n", sym->st_value );
            }
#endif
            continue;
        }

        //printf("   %s@@%s\n", name , vers );
        
        //print_sym( sym );
        if ( 0 == _dl_strcmp( name, "__vdso_gettimeofday" ) ){
            _dl__vdso_gettimeofday = func_addr;
#ifdef __SUPPORT_LD_DEBUG__
            if ( _dl_debug_vdso != 0 ){
                _dl_dprintf(2,"   %s at address %p\n", name, func_addr );
            }
#endif
            continue;
        }
        
        if ( 0 == _dl_strcmp( name, "__vdso_clock_gettime" ) ){
            _dl__vdso_clock_gettime = func_addr;
#ifdef __SUPPORT_LD_DEBUG__
            if ( _dl_debug_vdso != 0 ){
                _dl_dprintf(2,"   %s at address %p\n", name, func_addr );
            }
#endif
            continue;
        }
        
#ifdef __SUPPORT_LD_DEBUG__
        if ( _dl_debug_vdso != 0 ){
            _dl_dprintf(2,"   <%s> not handled\n", name );
        }
#endif
        
        
    }
    
 
}

#endif // __VDSO_SUPPORT__