歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
您现在的位置: Linux教程網 >> UnixLinux >  >> Linux綜合 >> 學習Linux

linux下實現在程序運行時的函數替換(熱補丁),linux程序運行

linux下實現在程序運行時的函數替換(熱補丁),linux程序運行


linux下實現在程序運行時的函數替換(熱補丁),linux程序運行


聲明:以下的代碼成果,是參考了網上的injso技術,在本文的最後會給出地址,同時非常感謝injso技術原作者的分享。

   但是injso文章中的代碼存在一些問題,所以後面出現的代碼是經過作者修改和檢測的。也正因為這些錯誤,加深了我的學習深度。

 

  最近因為在學習一些調試的技術,但是很少有提到如何在函數運行時實現函數替換的。

  為什麼會想到這一點?因為在學習調試時,難免會看到一些內核方面的調試技術,內核中的調試有一個kprobe,很強大,可以實現運行時的函數替換。其原理就是hook,鉤子,但是學習了這個kprobe之後會發現,kprobe內部有檢測所要鉤的函數是不是屬於內核空間,必須是內核函數才能實現替換。而實際上,我的工作大部分還是在應用層的,所以想要實現應用程序的熱補丁技術。

  一些基礎的知識這邊的就不展開了,需要的基礎有,elf文件格式,ptrace,waitpid,應用程序間通信時的信號,匯編。

  • 1、elf文件加載過程

  elf簡單地說是由以下四部分組成的,elf文件頭,program header和section header,內容。其中program header是運行時使用的,而section header並不會被加載進程序運行空間,但他們可以在編譯時被指定該段的加載地址等信息,當然一般這個鏈接腳本.lds是由gcc默認的。

  第一步,加載elf文件頭,檢驗文件類型版本等,重要的是找到program header的地址和header的個數,如果連接器腳本是默認的,那麼elf文件頭會被加載在0x804800地址處。

  第二步,加載program header,接著掃描program header,找到一個類型為PT_INTERP的program header,這個header裡面放著的是有關解釋器的地址,這時候將解釋器程序的elf文件頭加載進來。一般是這樣:

      INTERP 0x000134 0x08048134 0x08048134 0x00013 0x00013 R 0x1
        [Requesting program interpreter: /lib/ld-linux.so.2]

  第三步,掃描program header,如果類型為PT_LOAD,則將該段加載進來。

  第四步,判斷是否需要解釋器程序,如果需要,把解釋器程序加載進來,並把程序入口設置為解釋器程序的地址。否則是應用程序本身的入口。反匯編為_start標號。

  第五步,設置命令行傳入的參數等應用程序需要的信息。

  第六步,解釋器程序開始運行,加載程序需要的庫,填寫重定向符號表中的地址信息。

  • 2.elf文件動態鏈接過程

  上一步,解釋器程序根據program header已經將應用程序的段都加載進內存了,接下來再掃描program header,找到類型為PT_DYNAMIC,這裡面包含了很多由section header描述的內容,包括重定向表,符號表,字符串表等等。解釋器需要這個段描述的一些信息。

  DT_NEEDED描述了所需要的動態庫名稱,DT_REL描述了重定位表地址,DT_JMPREL描述了重定位表地址(這個表是懶惰鏈接使用的),DT_PLTGOT全局偏移表地址。

  此時解釋器程序就可以根據所需要的動態庫,將其加載進內存。每一個被加載進來的庫的相關信息會被記錄在link_map結構中,這個結構是一個鏈表,保存了所有的動態信息。

  其中,全局偏移表got,got[0]保存了PT_DYNAMIC的起始地址,got[1]保存link_map的地址,而link_map中就可以找到PT_DYNAMIC的起始地址,和下一個或者上一個共享文件或者可執行文件的link_map地址。

  DT_REL這個重定向表中的符號必須在此時就被解析完成。

  而DT_JMPREL這個重定向表中的符號可以在運行時再解析。

  所有的庫和符號全部解析完成之後,解釋器程序就會把控制權交給可執行文件的_start。程序開始執行。

  • 3.替換函數和被替換函數

  被替換程序源碼。 

#include <stdio.h>
#include <time.h>
int main()
{
        while(1){
                sleep(10);
                printf("%d : original\n",time(0));
        }
}

  替換新庫代碼。

#include <stdio.h>

int newmyprint()
{
	write(1,"hahahahahahaha",14);
	return 0;
}

  夠簡單明了吧,如果替換成功,目標程序將會一直輸出“哈哈哈哈哈哈”。

  • 4.功能函數  

  ptrace相關代碼:

/* 讀進程寄存器 */
void ptrace_readreg(int pid, struct user_regs_struct *regs)
{
    if(ptrace(PTRACE_GETREGS, pid, NULL, regs))
        printf("*** ptrace_readreg error ***\n");
	/*printf("ptrace_readreg\n");
	printf("%x\n",regs->ebx);
	printf("%x\n",regs->ecx);
	printf("%x\n",regs->edx);
	printf("%x\n",regs->esi);
	printf("%x\n",regs->edi);
	printf("%x\n",regs->ebp);
	printf("%x\n",regs->eax);
	printf("%x\n",regs->xds);
	printf("%x\n",regs->xes);
	printf("%x\n",regs->xfs);
	printf("%x\n",regs->xgs);
	printf("%x\n",regs->orig_eax);
	printf("%x\n",regs->eip);
	printf("%x\n",regs->xcs);
	printf("%x\n",regs->eflags);
	printf("%x\n",regs->esp);
	printf("%x\n",regs->xss);*/

}

/* 寫進程寄存器 */
void ptrace_writereg(int pid, struct user_regs_struct *regs)
{
	/*printf("ptrace_writereg\n");
	printf("%x\n",regs->ebx);
	printf("%x\n",regs->ecx);
	printf("%x\n",regs->edx);
	printf("%x\n",regs->esi);
	printf("%x\n",regs->edi);
	printf("%x\n",regs->ebp);
	printf("%x\n",regs->eax);
	printf("%x\n",regs->xds);
	printf("%x\n",regs->xes);
	printf("%x\n",regs->xfs);
	printf("%x\n",regs->xgs);
	printf("%x\n",regs->orig_eax);
	printf("%x\n",regs->eip);
	printf("%x\n",regs->xcs);
	printf("%x\n",regs->eflags);
	printf("%x\n",regs->esp);
	printf("%x\n",regs->xss);*/

    if(ptrace(PTRACE_SETREGS, pid, NULL, regs))
        printf("*** ptrace_writereg error ***\n");
}

/* 關聯到進程 */
void ptrace_attach(int pid)
{
    if(ptrace(PTRACE_ATTACH, pid, NULL, NULL) < 0) {
        perror("ptrace_attach");
        exit(-1);
    }

    waitpid(pid, NULL, /*WUNTRACED*/0);   
   
    ptrace_readreg(pid, &oldregs);
}

/* 進程繼續 */
void ptrace_cont(int pid)
{
    int stat;

    if(ptrace(PTRACE_CONT, pid, NULL, NULL) < 0) {
        perror("ptrace_cont");
        exit(-1);
    }
    /*while(!WIFSTOPPED(stat))
        waitpid(pid, &stat, WNOHANG);*/
}

/* 脫離進程 */
void ptrace_detach(int pid)
{
    ptrace_writereg(pid, &oldregs);

    if(ptrace(PTRACE_DETACH, pid, NULL, NULL) < 0) {
        perror("ptrace_detach");
        exit(-1);
    }
}

/* 寫指定進程地址 */
void ptrace_write(int pid, unsigned long addr, void *vptr, int len)
{
    int count;
    long word;

    count = 0;

    while(count < len) {
        memcpy(&word, vptr + count, sizeof(word));
        word = ptrace(PTRACE_POKETEXT, pid, addr + count, word);
        count += 4;

        if(errno != 0)
            printf("ptrace_write failed\t %ld\n", addr + count);
    }
}

/* 讀指定進程 */
int ptrace_read(int pid, unsigned long addr, void *vptr, int len)
{
    int i,count;
    long word;
    unsigned long *ptr = (unsigned long *)vptr;

    i = count = 0;
	//printf("ptrace_read addr = %x\n",addr);
    while (count < len) {
		//printf("ptrace_read addr+count = %x\n",addr + count);
        word = ptrace(PTRACE_PEEKTEXT, pid, addr + count, NULL);
		while(word < 0)
		{
			if(errno == 0)
				break;
			//printf("ptrace_read word = %x\n",word);
			perror("ptrace_read failed");
			return 2;
		}
        count += 4;
        ptr[i++] = word;
    }
	return 0;
}

/*
 在進程指定地址讀一個字符串
 */
char * ptrace_readstr(int pid, unsigned long addr)
{
    char *str = (char *) malloc(64);
    int i,count;
    long word;
    char *pa;

    i = count = 0;
    pa = (char *)&word;

    while(i <= 60) {
        word = ptrace(PTRACE_PEEKTEXT, pid, addr + count, NULL);
        count += 4;

        if (pa[0] == 0) {
            str[i] = 0;
        break;
        }
        else
            str[i++] = pa[0];

        if (pa[1] == 0) {
            str[i] = 0;
            break;
        }
        else
            str[i++] = pa[1];

        if (pa[2] ==0) {
            str[i] = 0;
            break;
        }
        else
            str[i++] = pa[2];

        if (pa[3] ==0) {
            str[i] = 0;
            break;
        }
        else
            str[i++] = pa[3];
    }
   
    return str;
}




/*
 將指定數據壓入進程堆棧並返回堆棧指針
 */
void * ptrace_push(int pid, void *paddr, int size)
{
    unsigned long esp;
    struct user_regs_struct regs;

    ptrace_readreg(pid, &regs);
    esp = regs.esp;
    esp -= size;
    esp = esp - esp % 4;
    regs.esp = esp;

    ptrace_writereg(pid, &regs);

    ptrace_write(pid, esp, paddr, size);

    return (void *)esp;
}

/*
 在進程內調用指定地址的函數
 */
void ptrace_call(int pid, unsigned long addr)
{
    void *pc;
    struct user_regs_struct regs;
    int stat;
    void *pra;

    pc = (void *) 0x41414140;
    pra = ptrace_push(pid, &pc, sizeof(pc));

    ptrace_readreg(pid, &regs);
    regs.eip = addr;
    ptrace_writereg(pid, &regs);

    ptrace_cont(pid);
    //while(WIFSIGNALED(stat))
       // waitpid(pid, &stat, WNOHANG);
}

  這裡面的東西我就不展開了,對ptrace的學習,請自行man。

  

/*
因為應用程序可能不存在hash表,所以通過讀取源文件的section header獲取符號表的入口數,
其實是被誤導了,但也學習了hash表的作用,用來快速查找符號表中的信息和字符串表中的信息
*/
/*int getnchains(int pid,unsigned long base_addr)
{
	printf("getnchains enter \n");
    Elf32_Ehdr *ehdr = (Elf32_Ehdr *) malloc(sizeof(Elf32_Ehdr));       
	Elf32_Shdr *shdr = (Elf32_Shdr *)malloc(sizeof(Elf32_Shdr));
	unsigned long shdr_addr;
	int i = 0;
	int fd;
	char filename[1024] = {0};
    ptrace_read(pid, base_addr, ehdr, sizeof(Elf32_Ehdr));
    shdr_addr = base_addr + ehdr->e_shoff;
    //printf("getnchains ehdr->e_shoff\t %p\n", ehdr->e_shoff);
	
	snprintf(filename, sizeof(filename), "/proc/%d/exe", pid);
	fd = open(filename, O_RDONLY);
	if (lseek(fd, ehdr->e_shoff, SEEK_SET) < 0) 
		exit(-1);
	
	/*while(i<ehdr->e_shnum)
	{
		read(fd, shdr, ehdr->e_shentsize);
		printf("getnchains i = %d\n",i);
		printf("getnchains shdr->sh_type = %x\n",shdr->sh_type);
		printf("getnchains shdr->sh_name = %x\n",shdr->sh_name);
		printf("getnchains shdr->sh_size = %x\n",shdr->sh_size);
		printf("getnchains shdr->sh_entsize = %x\n",shdr->sh_entsize);
		i++;
	}
	
    while(shdr->sh_type != SHT_SYMTAB)
		read(fd, shdr, ehdr->e_shentsize);
	nchains = shdr->sh_size/shdr->sh_entsize;
	//printf("getnchains shdr->sh_type = %d\n",shdr->sh_type);
	//printf("getnchains shdr->sh_name = %d\n",shdr->sh_name);
	//printf("getnchains shdr->sh_size = %d\n",shdr->sh_size);
	//printf("getnchains shdr->sh_entsize = %d\n",shdr->sh_entsize);
	//printf("getnchains nchains = %x\n",nchains);	
	close(fd);
	free(ehdr);
	free(shdr);
	printf("getnchains exit \n");
}
*/


/*
 取得指向link_map鏈表首項的指針
 */
struct link_map * get_linkmap(int pid)
{
    Elf32_Ehdr *ehdr = (Elf32_Ehdr *) malloc(sizeof(Elf32_Ehdr));       
    Elf32_Phdr *phdr = (Elf32_Phdr *) malloc(sizeof(Elf32_Phdr));
    Elf32_Dyn  *dyn =  (Elf32_Dyn *) malloc(sizeof(Elf32_Dyn));
    Elf32_Word got;
    struct link_map *map = (struct link_map *)malloc(sizeof(struct link_map));
    int i = 1;
	unsigned long tmpaddr;

    ptrace_read(pid, IMAGE_ADDR, ehdr, sizeof(Elf32_Ehdr));
    phdr_addr = IMAGE_ADDR + ehdr->e_phoff;
    printf("phdr_addr\t %p\n", phdr_addr);

    ptrace_read(pid, phdr_addr, phdr, sizeof(Elf32_Phdr));
    while(phdr->p_type != PT_DYNAMIC)
        ptrace_read(pid, phdr_addr += sizeof(Elf32_Phdr), phdr,sizeof(Elf32_Phdr));
    dyn_addr = phdr->p_vaddr;
    printf("dyn_addr\t %p\n", dyn_addr);

    ptrace_read(pid, dyn_addr, dyn, sizeof(Elf32_Dyn));
    while(dyn->d_tag != DT_PLTGOT) {
		tmpaddr = dyn_addr + i * sizeof(Elf32_Dyn);
		//printf("get_linkmap tmpaddr = %x\n",tmpaddr);
        ptrace_read(pid,tmpaddr, dyn, sizeof(Elf32_Dyn));
        i++;
    }

    got = (Elf32_Word)dyn->d_un.d_ptr;
    got += 4;
    //printf("GOT\t\t %p\n", got);

    ptrace_read(pid, got, &map_addr, 4);
    printf("map_addr\t %p\n", map_addr);
	map = map_addr;
    //ptrace_read(pid, map_addr, map, sizeof(struct link_map));
   
    free(ehdr);
    free(phdr);
    free(dyn);

    return map;
}

/*
 取得給定link_map指向的SYMTAB、STRTAB、HASH、JMPREL、PLTRELSZ、RELAENT、RELENT信息
 這些地址信息將被保存到全局變量中,以方便使用
 */
void get_sym_info(int pid, struct link_map *lm)
{
    Elf32_Dyn *dyn = (Elf32_Dyn *) malloc(sizeof(Elf32_Dyn));
    unsigned long dyn_addr;
	//printf("get_sym_info lm = %x\n",lm);
	//printf("get_sym_info lm->l_ld's offset = %x\n",&((struct link_map *)0)->l_ld);
	//printf("get_sym_info &lm->l_ld = %x\n",&(lm->l_ld));
    //dyn_addr = (unsigned long)&(lm->l_ld);
	//進入被跟蹤進程獲取動態節的地址	
    ptrace_read(pid,&(lm->l_ld) , &dyn_addr, sizeof(dyn_addr));
    ptrace_read(pid,&(lm->l_addr) , &link_addr, sizeof(dyn_addr));
    ptrace_read(pid, dyn_addr, dyn, sizeof(Elf32_Dyn));
	//if(link_addr == 0)
	//	getnchains(pid,IMAGE_ADDR);
	/*else
		getnchains(pid,link_addr);*/
    while(dyn->d_tag != DT_NULL){
		//printf("get_sym_info dyn->d_tag = %x\n",dyn->d_tag);
		//printf("get_sym_info dyn->d_un.d_ptr = %x\n",dyn->d_un.d_ptr);
        switch(dyn->d_tag)
        {
        case DT_SYMTAB:
            symtab = dyn->d_un.d_ptr;
			
			break;
        case DT_STRTAB:
            strtab = dyn->d_un.d_ptr;
            break;
        /*case DT_HASH://可能不存在哈希表,此時nchains是錯誤的,這個值可以通過符號表得到
			//printf("get_sym_info hash table's addr = %x\n",dyn->d_un.d_ptr);
			//printf("get_sym_info symtbl's entry = %x\n",(dyn->d_un.d_ptr) + 4);
            ptrace_read(pid, (dyn->d_un.d_ptr) + 4,&nchains, sizeof(nchains));
            break;*/
        case DT_JMPREL:
            jmprel = dyn->d_un.d_ptr;
            break;
        case DT_PLTRELSZ:
            totalrelsize = dyn->d_un.d_val;
            break;
        case DT_RELAENT:
            relsize = dyn->d_un.d_val;
            break;
        case DT_RELENT:
            relsize = dyn->d_un.d_val;
            break;
		case DT_REL:
			reldyn = dyn->d_un.d_ptr;		
			break;
		case DT_RELSZ:
			reldynsz = dyn->d_un.d_val;
			break;
        }
		ptrace_read(pid, dyn_addr += sizeof(Elf32_Dyn), dyn, sizeof(Elf32_Dyn));
    }
	
	//printf("get_sym_info link_addr = %x\n",link_addr);
	//printf("get_sym_info symtab = %x\n",symtab);
	//printf("get_sym_info relsize = %x\n",relsize);
	//printf("get_sym_info reldyn = %x\n",reldyn);
	//printf("get_sym_info totalrelsize = %x\n",totalrelsize);
	//printf("get_sym_info jmprel = %x\n",jmprel);
	//printf("get_sym_info nchains = %x\n",nchains);
	//printf("get_sym_info strtab = %x\n",strtab);

    nrels = totalrelsize / relsize;
	nreldyns = reldynsz/relsize;
	
	//printf("get_sym_info nreldyns = %d\n",nreldyns);
	//printf("get_sym_info nrels = %d\n",nrels);

    free(dyn);
	printf("get_sym_info exit\n");
}
/*
 在指定的link_map指向的符號表查找符號,它僅僅是被上面的find_symbol使用
 */
unsigned long  find_symbol_in_linkmap(int pid, struct link_map *lm, char *sym_name)
{
    Elf32_Sym *sym = (Elf32_Sym *) malloc(sizeof(Elf32_Sym));
	int i = 0;
    char *str;
    unsigned long ret;
	int flags = 0;

    get_sym_info(pid, lm);
   
    do{
		if(ptrace_read(pid, symtab + i * sizeof(Elf32_Sym), sym, sizeof(Elf32_Sym)))
			return 0;
		i++;
		//printf("find_symbol_in_linkmap sym->st_name = %x\tsym->st_size = %x\tsym->st_value = %x\n",sym->st_name,sym->st_size,sym->st_value);
		//printf("find_symbol_in_linkmap Elf32_Sym's size = %d\n",sizeof(Elf32_Sym));
		//printf("\nfind_symbol_in_linkmap sym->st_name = %x\n",sym->st_name);        
		if (!sym->st_name && !sym->st_size && !sym->st_value)//全為0是符號表的第一項
            continue;
		//printf("\nfind_symbol_in_linkmap strtab = %x\n",strtab);
        str = (char *) ptrace_readstr(pid, strtab + sym->st_name);
		//printf("\nfind_symbol_in_linkmap str = %s\n",str);
		//printf("\nfind_symbol_in_linkmap sym->st_value = %x\n",sym->st_value);
        if (strcmp(str, sym_name) == 0) {
			printf("\nfind_symbol_in_linkmap str = %s\n",str);
			printf("\nfind_symbol_in_linkmap sym->st_value = %x\n",sym->st_value);
            free(str);
			if(sym->st_value == 0)//值為0代表這個符號本身就是重定向的內容
				continue;
			flags = 1;
			
            //str = ptrace_readstr(pid, (unsigned long)lm->l_name);
            //printf("find_symbol_in_linkmap lib name [%s]\n", str);
            //free(str);
            break;
        }
		
        free(str);
    }while(1);


    if (flags != 1)
        ret = 0;
    else
    	ret =  link_addr + sym->st_value;

    free(sym);

    return ret;
}

/*
 解析指定符號
 */
unsigned long  find_symbol(int pid, struct link_map *map, char *sym_name)
{
    struct link_map *lm = map;
    unsigned long sym_addr;
    char *str;
	unsigned long tmp;
   
    //sym_addr = find_symbol_in_linkmap(pid, map, sym_name); 
	//return 0;
    //if (sym_addr)
     //   return sym_addr;
	//printf("\nfind_symbol map = %x\n",map);
	//ptrace_read(pid,(char *)map+12,&tmp,4);
	//lm = tmp;
	//printf("find_symbol lm = %x\n",lm);
    //ptrace_read(pid, (unsigned long)map->l_next, lm, sizeof(struct link_map));
    sym_addr = find_symbol_in_linkmap(pid, lm, sym_name);
    while(!sym_addr ) {
        ptrace_read(pid, (char *)lm+12, &tmp, 4);//獲取下一個庫的link_map地址
		if(tmp == 0)
			return 0;
		lm = tmp;
		//printf("find_symbol lm = %x\n",lm);
        /*str = ptrace_readstr(pid, (unsigned long)lm->l_name);
        if(str[0] == '/0')
            continue;
        printf("[%s]\n", str);
        free(str);*/

        if ((sym_addr = find_symbol_in_linkmap(pid, lm, sym_name)))
            break;
    }

    return sym_addr;
}


/* 查找符號的重定位地址 */
unsigned long  find_sym_in_rel(int pid, char *sym_name)
{
    Elf32_Rel *rel = (Elf32_Rel *) malloc(sizeof(Elf32_Rel));
    Elf32_Sym *sym = (Elf32_Sym *) malloc(sizeof(Elf32_Sym));
    int i;
    char *str;
    unsigned long ret;
    struct link_map *lm;
	lm = map_addr;
	
    //get_dyn_info(pid);
    do{
		get_sym_info(pid,lm);
        ptrace_read(pid, (char *)lm+12, &lm, 4);
    	//首先查找過程連接的重定位表
	    for(i = 0; i< nrels ;i++) {
	        ptrace_read(pid, (unsigned long)(jmprel + i * sizeof(Elf32_Rel)),
	                                                                 rel, sizeof(Elf32_Rel));
	        if(ELF32_R_SYM(rel->r_info)) {
	            ptrace_read(pid, symtab + ELF32_R_SYM(rel->r_info) *
	                                               sizeof(Elf32_Sym), sym, sizeof(Elf32_Sym));
	            str = ptrace_readstr(pid, strtab + sym->st_name);
	            if (strcmp(str, sym_name) == 0) {
					if(sym->st_value != 0){
						free(str);
						continue;
					}
					modifyflag = 1;
	                free(str);
	                break;
	            }
	            free(str);
	        }
	    }
		
		if(modifyflag == 1)
			break;
		//沒找到的話,再找在鏈接時就重定位的重定位表
		for(i = 0; i< nreldyns;i++) {
	        ptrace_read(pid, (unsigned long)(reldyn+ i * sizeof(Elf32_Rel)),
	                                                                 rel, sizeof(Elf32_Rel));
	        if(ELF32_R_SYM(rel->r_info)) {
	            ptrace_read(pid, symtab + ELF32_R_SYM(rel->r_info) *
	                                               sizeof(Elf32_Sym), sym, sizeof(Elf32_Sym));
	            str = ptrace_readstr(pid, strtab + sym->st_name);
	            if (strcmp(str, sym_name) == 0) {
					if(sym->st_value != 0){
						free(str);
						continue;
					}
					modifyflag = 2;
	                free(str);
	                break;
	            }
	            free(str);
	        }
	    }
		
		if(modifyflag == 2)
			break;
		
    }while(lm);
	//printf("find_sym_in_rel flags = %d\n",flags);
    if (modifyflag == 0)
        ret = 0;
    else
    	ret =  link_addr + rel->r_offset;
	//printf("find_sym_in_rel link_addr = %x\t sym->st_value = %x\n",link_addr , sym->st_value);
    free(rel);
	free(sym);

    return ret;
}

/*
 在進程自身的映象中(即不包括動態共享庫,無須遍歷link_map鏈表)獲得各種動態信息
 */
/*void get_dyn_info(int pid)
{
    Elf32_Dyn *dyn = (Elf32_Dyn *) malloc(sizeof(Elf32_Dyn));
    int i = 0;

    ptrace_read(pid, dyn_addr + i * sizeof(Elf32_Dyn), dyn, sizeof(Elf32_Dyn));
    i++;
    while(dyn->d_tag){
        switch(dyn->d_tag)
        {
        case DT_SYMTAB:
            //puts("DT_SYMTAB");
            symtab = dyn->d_un.d_ptr;
            break;
        case DT_STRTAB:
            strtab = dyn->d_un.d_ptr;
            //puts("DT_STRTAB");
            break;
        case DT_JMPREL:
            jmprel = dyn->d_un.d_ptr;
            //puts("DT_JMPREL");
            //printf("jmprel\t %p\n", jmprel);
            break;
        case DT_PLTRELSZ:
            totalrelsize = dyn->d_un.d_val;
            //puts("DT_PLTRELSZ");
            break;
        case DT_RELAENT:
            relsize = dyn->d_un.d_val;
            //puts("DT_RELAENT");
            break;
        case DT_RELENT:
            relsize = dyn->d_un.d_val;
            //puts("DT_RELENT");
            break;
        }

        ptrace_read(pid, dyn_addr + i * sizeof(Elf32_Dyn), dyn, sizeof(Elf32_Dyn));
        i++;
    }

    nrels = totalrelsize / relsize;

    free(dyn);
}*/

/*void call_dl_open(int pid, unsigned long addr, char *libname)
{
    void *pRLibName;
    struct user_regs_struct regs;

    /*
      先找個空間存放要裝載的共享庫名,我們可以簡單的把它放入堆棧
     
    pRLibName = ptrace_push(pid, libname, strlen(libname) + 1);

    /* 設置參數到寄存器 
    ptrace_readreg(pid, &regs);
    regs.eax = (unsigned long) pRLibName;
    regs.ecx = 0x0;
    regs.edx = RTLD_LAZY;
    ptrace_writereg(pid, &regs);

    /* 調用_dl_open 
    ptrace_call(pid, addr);
    puts("call _dl_open ok");
}*/




/*#define RTLD_LAZY	0x00001	
#define RTLD_NOW	0x00002	
#define	RTLD_BINDING_MASK   0x3	
#define RTLD_NOLOAD	0x00004	
#define RTLD_DEEPBIND	0x00008	

#define RTLD_GLOBAL	0x00100

#define RTLD_LOCAL	0

#define RTLD_NODELETE	0x01000 */

void call__libc_dlopen_mode(int pid, unsigned long addr, char *libname)
{
    void *plibnameaddr;

    //printf("call__libc_dlopen_mode libname = %s\n",libname);
	//printf("call__libc_dlopen_mode addr = %x\n",addr);
	//將需要加載的共享庫地址壓棧
    plibnameaddr = ptrace_push(pid, libname, strlen(libname) + 1);
	ptrace_push(pid,&mode,sizeof(int));
	ptrace_push(pid,&plibnameaddr,sizeof(plibnameaddr));

    /* 調用__libc_dlopen_mode */
    ptrace_call(pid, addr);
}
void call_printf(int pid, unsigned long addr, char *string)
{
    void *paddr;

    paddr = ptrace_push(pid, string, strlen(string) + 1);
	ptrace_push(pid,&paddr,sizeof(paddr));

    ptrace_call(pid, addr);
}

  作者所做的修改,讀者可以對比文章最後的連接中的代碼。

  這邊對於程序的具體解釋,就不具體展開了。

  需要注意的是,原來是采用_dl_open的方式加載庫函數,但是ld庫並沒有這個符號導出。而libc庫中導出了一個可以加載庫的__libc_dlopen_mode函數。

  • 5.主函數

  先說一下流程,

  a.獲取被跟蹤進程的link_map地址

  b.根據link_map給出的信息,搜索符號表,遍歷每一個link_map中的符號表,直到找到想要找的符號。這裡是printf或者__libc_dlopen_mode函數

  c.將庫路徑包括庫名稱傳遞給調用__libc_dlopen_mode的函數,該函數即call__libc_dlopen_mode會把__libc_dlopen_mode函數需要的參數,路徑和加載方式壓棧,在讓被跟蹤進

   程開始運行之前,壓入一個非法地址,當__libc_dlopen_mode返回時返回到一個非法地址時,就會發生中斷,此時跟蹤進程可以waitpid跟蹤到。好,設置寄存器,並讓被跟蹤進程開

   始運行。打開庫之後,被跟蹤進程因中斷而被跟蹤進程再次獲得控制權。

  d.再一次根據之前保存的link_map信息,當然完全可以直接用上一次搜索結果結束之後的link_map往後找,因為新庫一定在最後,但是本文還是從頭開始找,找到新庫中的

   newmyprint地址。

  e.還是根據link_map信息查找printf的重定向地址,在rel.dyn節中,有關這個rel.dyn和rel.plt等節之間的關系,可以看我的其他博文。

  f.將newmyprint的地址填入printf的重定向地址。

  g.將被跟蹤進程原先的寄存器設置回去,釋放控制。

  h.被跟蹤進程開始輸出“哈哈哈哈哈”。

  上源碼:

  

int main(int argc, char *argv[])
{
    int pid;
    struct link_map *map;
    char sym_name[256];
    unsigned long sym_addr;
    unsigned long new_addr,old_addr,rel_addr;
	int status = 0;
	char libpath[1024];
	char oldfunname[128];
	char newfunname[128];
	//mode = atoi(argv[2]);
	if(argc < 5){
		printf("usage : ./injso pid libpath oldfunname newfunname\n");
		exit(-1);
	}
    /* 從命令行取得目標進程PID*/
    pid = atoi(argv[1]); 
	
    /* 從命令行取得新庫名稱*/
	memset(libpath,0,sizeof(libpath));
	memcpy(libpath,argv[2],strlen(argv[2]));
	
    /* 從命令行取得舊函數的名稱*/
	memset(oldfunname,0,sizeof(oldfunname));
	memcpy(oldfunname,argv[3],strlen(argv[3]));
	
    /* 從命令行取得新函數的名稱*/
	memset(newfunname,0,sizeof(newfunname));
	memcpy(newfunname,argv[4],strlen(argv[4]));

	printf("main pid = %d\n",pid);
	printf("main libpath : %s\n",libpath);
	printf("main oldfunname : %s\n",oldfunname);
	printf("main newfunname : %s\n",newfunname);
    /* 關聯到目標進程*/
    ptrace_attach(pid);
   
    /* 得到指向link_map鏈表的指針 */
    map = get_linkmap(pid);                    /* get_linkmap */

	
    sym_addr = find_symbol(pid, map, "printf");       
    printf("found printf at addr %p\n", sym_addr);  
	if(sym_addr == 0)
		goto detach;
	call_printf(pid,sym_addr,"injso successed\n");
	waitpid(pid,&status,0);
	printf("status = %x\n",status);
	
    /*ptrace_writereg(pid, &oldregs);
	ptrace_cont(pid);

	

	waitpid(pid,&status,0);
	//printf("status = %x\n",status);
    //ptrace_readreg(pid, &oldregs);
	//oldregs.eip = 0x8048414;
    //ptrace_writereg(pid, &oldregs);
	ptrace_cont(int pid)(pid);
	
	ptrace_detach(pid);

	exit(0);*/
	
    /* 發現__libc_dlopen_mode,並調用它 */
    sym_addr = find_symbol(pid, map, "__libc_dlopen_mode");        /* call _dl_open */
    printf("found __libc_dlopen_mode at addr %p\n", sym_addr);  
	if(sym_addr == 0)
		goto detach;
    call__libc_dlopen_mode(pid, sym_addr,libpath);    /* 注意裝載的庫地址 */   
	//while(1);
	waitpid(pid,&status,0);
	/* 找到新函數的地址 */
    strcpy(sym_name, newfunname);                /* intercept */
    sym_addr = find_symbol(pid, map, sym_name);
    printf("%s addr\t %p\n", sym_name, sym_addr);
	if(sym_addr == 0)
		goto detach;

    /* 找到舊函數在重定向表的地址 */
    strcpy(sym_name, oldfunname);               
    rel_addr = find_sym_in_rel(pid, sym_name);
    printf("%s rel addr\t %p\n", sym_name, rel_addr);
	if(rel_addr == 0)
		goto detach;

    /* 找到用於保存read地址的指針 */
    //strcpy(sym_name, "oldread");               
    //old_addr = find_symbol(pid, map, sym_name);
    //printf("%s addr\t %p\n", sym_name, old_addr);

    /* 函數重定向 */
    puts("intercept...");                    /* intercept */
    //ptrace_read(pid, rel_addr, &new_addr, sizeof(new_addr));
    //ptrace_write(pid, old_addr, &new_addr, sizeof(new_addr));
    //rel_addr = 0x8048497;如果是靜態地址,也就是未導出該符號地址,那麼只能通過反匯編先找到該函數被調用的地方,將這個地方的跳轉地址修改
    
	if(modifyflag == 2)
		sym_addr = sym_addr - rel_addr - 4;
	printf("main modify sym_addr = %x\n",sym_addr);
	ptrace_write(pid, rel_addr, &sym_addr, sizeof(sym_addr));
	
    puts("injectso ok");
detach:
	printf("prepare to detach\n");
	ptrace_detach(pid);
	
	return 0;
  
}

  這裡面有一個很重要的地方,如果不先在目標進程中調用printf就不能夠調用__lib_dlopen_mode成功,這個原因很奇怪,根據當時的core文件來看崩潰在了下面的這個函數,原因是_dl_open_hook這個全局變量為0,但實際上運行過printf之後,這個_dl_open_hook還是0。這個有待後續檢驗。

void *
__libc_dlsym (void *map, const char *name)
{
  struct do_dlsym_args args;
  args.map = map;
  args.name = name;

#ifdef SHARED
  if (__builtin_expect (_dl_open_hook != NULL, 0))
    return _dl_open_hook->dlsym (map, name);
#endif
  return (dlerror_run (do_dlsym, &args) ? NULL
	  : (void *) (DL_SYMBOL_ADDRESS (args.loadbase, args.ref)));
}

  運行結果:

root@leo-desktop:injso# ./test
1467364356 : original
injso successed
hahahahahahahahahahahahahaha

  • 6.如何替換未導出符號的地址

  被替換函數源碼:

#include <stdio.h>


//int fun2();

int fun1()
{
        printf("fun1\n");
//      fun2();
}

int main()
{
        signed int i  = 0x40011673 ;
        i = i - 0x4001172d ;
        printf("i = %x\n",i);
        while(1){
                i = fun1();
                sleep(10);
        }
        return 1;
}

  這個怎麼來替換fun1函數的地址呢?

  首先反匯編得到main的機器碼,如下,

08048468 <main>:
 8048468:       55                      push   %ebp
 8048469:       89 e5                   mov    %esp,%ebp
 804846b:       83 e4 f0                and    $0xfffffff0,%esp
 804846e:       83 ec 20                sub    $0x20,%esp
 8048471:       c7 44 24 1c 73 16 01    movl   $0x40011673,0x1c(%esp)
 8048478:       40 
 8048479:       81 6c 24 1c 2d 17 01    subl   $0x4001172d,0x1c(%esp)
 8048480:       40 
 8048481:       b8 75 85 04 08          mov    $0x8048575,%eax
 8048486:       8b 54 24 1c             mov    0x1c(%esp),%edx
 804848a:       89 54 24 04             mov    %edx,0x4(%esp)
 804848e:       89 04 24                mov    %eax,(%esp)
 8048491:       e8 ce fe ff ff          call   8048364 <printf@plt>
 8048496:       e8 b9 ff ff ff          call   8048454 <fun1>
 804849b:       89 44 24 1c             mov    %eax,0x1c(%esp)
 804849f:       c7 04 24 0a 00 00 00    movl   $0xa,(%esp)
 80484a6:       e8 c9 fe ff ff          call   8048374 <sleep@plt>
 80484ab:       eb e9                   jmp    8048496 <main+0x2e>
 80484ad:       90                      nop
 80484ae:       90                      nop
 80484af:       90                      nop

  可以看到在地址0x8048496處的機器碼是跳轉到fun1函數的,那麼這個ffffffb9就是call的操作數,操作數地址0x8048497,也就是說把這個地址中的數值改掉就可以了,有關這個call或者jmp的地址計算可以查看我的另外一篇博文。

  有關這個如何跳轉的方法,已經在主函數的代碼中給出了,但是被我注釋掉了,大家感興趣的話,可以自己試試。

  效果:

root@leo-desktop:lib2lib# ./a.out
i = ffffff46
fun1
injso successed
hahahahahahaha^C

  這裡面的無關代碼,大家仔細看,是為了證明call的函數地址計算方式的。

  • 7.總結

  那麼講到現在的話,已經實現了不管函數符號是否導出都可以實現運行時替換的代碼。

  這裡面主要的技術是,elf文件格式,運行時加載的過程,跳轉地址的計算,運行時鏈接的過程,也就是plt表(當然這個也可以從我的另一篇博文中看到)。

  比較遺憾的是有關那個奔潰,有網友如果找到了原因,請回復下,3q。當然我也會自己再研究下。

 

   最後補上全局變量和頭文件:

#include <stdio.h>
#include <string.h>
#include <elf.h>
#include <sys/types.h>
#include <stdio.h>
#include <sys/ptrace.h>
#include <sys/wait.h>
#include <sys/errno.h>
#include <sys/user.h>
#include <link.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <bits/dlfcn.h>

#define IMAGE_ADDR 0x08048000

int mode = 2;

struct user_regs_struct oldregs;
Elf32_Addr phdr_addr;
Elf32_Addr dyn_addr;
Elf32_Addr map_addr;
Elf32_Addr symtab;
Elf32_Addr strtab;
Elf32_Addr jmprel;
Elf32_Addr reldyn;
Elf32_Word reldynsz;
Elf32_Word totalrelsize;
Elf32_Word relsize;
unsigned long link_addr;
int nrels;
int nreldyns;
//int nchains;
int modifyflag = 0;
/*char libpath[128] = "/mnt/hgfs/svnroot/test/injectsov2/prj_linux/so.so";*/

  

 

  • 8.修正

  針對在調用__libc_dlopen_mode函數之前需要調用printf的問題,終於讓我在晚上解決了。
  首先,我嘗試了調用其他函數而不是printf函數,發現效果一樣,包括第一次是調用__libc_dlopen_mode,第二次對該函數的調用都可以成功。
  其次,那麼現在問題就集中在了這兩個__libc_dlopen_mode調用之間的差別在哪裡,程序段肯定是一致的,棧也是一致的,而堆空間未使用,還有一個重要的因素,那就是寄存器。
  最後,發現在調用__libc_dlopen_mode前,有四個寄存器不同,分別是eax,orig_eax,eflags和esp。我一開始認為,通用寄存器eax和orig_eax不會對程序的執行造成影響。但是通過實驗,僅調一次__libc_dlopen_mode,部分寄存器賦正確執行時的值,發現對eax和orig_eax被賦於正確執行時的值時,程序可以正常運行,而且不僅僅必須是一種值,比如eax可以是0,1,0xffffffff,很多值都可以,但是被賦予0xfffffdfc和0xfffffdff等值時會失敗,試驗過並不是因為d這一位決定的,0xfffffdf0或者d00是可以運行成功的。

(gdb) disassemble __libc_dlopen_mode
Dump of assembler code for function __libc_dlopen_mode:
   0x00232640 <+0>:	push   %ebp
   0x00232641 <+1>:	mov    %esp,%ebp
   0x00232643 <+3>:	sub    $0x1c,%esp
   0x00232646 <+6>:	mov    %ebx,-0x8(%ebp)
   0x00232649 <+9>:	mov    0x8(%ebp),%eax
   0x0023264c <+12>:	call   0x144a0f
   0x00232651 <+17>:	add    $0x519a3,%ebx
   0x00232657 <+23>:	mov    0xc(%ebp),%edx
   0x0023265a <+26>:	mov    %esi,-0x4(%ebp)
   0x0023265d <+29>:	mov    %eax,-0x14(%ebp)
   0x00232660 <+32>:	mov    %edx,-0x10(%ebp)
   0x00232663 <+35>:	mov    0x354c(%ebx),%esi
   0x00232669 <+41>:	test   %esi,%esi

  在實驗中,還發現對eax賦於不正確的值時,當時忘了記了,還讓程序跑飛了。崩了,但是新庫已經加載上了。所以這個函數替換還是有一定的風險,或者說libc庫本身存在一定的bug。
  所以現在問題找到了,在於eax和orig_eax上,但是對__libc_dlopen_mode反匯編發現,eax在函數開頭就被賦予了通過棧傳遞的參數2的值,所以eax不應該影響程序的運行,但實際上影響了,這一點讓我覺得很奇怪,如果有任何網友對這個原因知曉的話,麻煩回復,萬分感謝。

 

  linux共享庫注射地址:http://www.docin.com/p-634172083.html

 

 

  

 

http://xxxxxx/Linuxjc/1139030.html TechArticle

Copyright © Linux教程網 All Rights Reserved