birl/dlfunc.c at master · linuxdeepin-packages/birl · GitHub

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
/* -*- Mode: C; tab-width: 4; -*- */
/*
* Copyright (C) 2009, HustMoon Studio
*
* 文件名称：dlfunc.c
* 摘	要：动态载入库函数
* 作	者：HustMoon@BYHH
* 邮	箱：www.ehust@gmail.com
* 日	期：2009.11.11
*/
#include "dlfunc.h"

#ifndef NO_DYLOAD
#include <dlfcn.h>

int (*pcap_findalldevs)(pcap_if_t **, char *);
void (*pcap_freealldevs)(pcap_if_t *);
pcap_t *(*pcap_open_live)(const char *, int, int, int, char *);
int (*pcap_compile)(pcap_t *, struct bpf_program *, const char *, int, bpf_u_int32);
int (*pcap_setfilter)(pcap_t *, struct bpf_program *);
char *(*pcap_geterr)(pcap_t *);
void (*pcap_freecode)(struct bpf_program *);
int (*pcap_loop)(pcap_t *, int, pcap_handler, unsigned char *);
void (*pcap_close)(pcap_t *);
void (*pcap_breakloop)(pcap_t *);
int (*pcap_sendpacket)(pcap_t *, const unsigned char *, int);

static void *libpcap = NULL;

int load_libpcap(void)
{
    char *error;
    char *file[] = {"libpcap.so", "libpcap.so.1", "libpcap.so.1.0", "libpcap.so.0.9", "libpcap.so.0.8"};
    int i, count = 5;
    for (i=0; i<count && !libpcap; i++)
    {
        libpcap = dlopen(file[i], RTLD_LAZY);
        error = dlerror();
    }
    if (libpcap == NULL)
    {
        printf("!! 打开libpcap失败，请检查是否已安装该库文件。\n");
        return -1;
    }
    if ((pcap_findalldevs = dlsym(libpcap, "pcap_findalldevs"), error = dlerror()) != NULL
            || (pcap_freealldevs = dlsym(libpcap, "pcap_freealldevs"), error = dlerror()) != NULL
            || (pcap_open_live = dlsym(libpcap, "pcap_open_live"), error = dlerror()) != NULL
            || (pcap_compile = dlsym(libpcap, "pcap_compile"), error = dlerror()) != NULL
            || (pcap_setfilter = dlsym(libpcap, "pcap_setfilter"), error = dlerror()) != NULL
            || (pcap_geterr = dlsym(libpcap, "pcap_geterr"), error = dlerror()) != NULL
            || (pcap_freecode = dlsym(libpcap, "pcap_freecode"), error = dlerror()) != NULL
            || (pcap_loop = dlsym(libpcap, "pcap_loop"), error = dlerror()) != NULL
            || (pcap_close = dlsym(libpcap, "pcap_close"), error = dlerror()) != NULL
            || (pcap_breakloop = dlsym(libpcap, "pcap_breakloop"), error = dlerror()) != NULL
            || (pcap_sendpacket = dlsym(libpcap, "pcap_sendpacket"), error = dlerror()) != NULL)
    {
        printf("!! 从libpcap获取函数失败: %s\n", error);
        free_libpcap();
        return -1;
    }
    return 0;
}

void free_libpcap(void)
{
    if (libpcap)
    {
        dlclose(libpcap);
        dlerror();
        libpcap = NULL;
    }
}

#endif	/* NO_DYLOAD */