1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Data Differential's libhostle
4 *
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
17 * distribution.
18 *
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
21 * written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34 *
35 */
36
37 #include "gear_config.h"
38
39 #if defined(TARGET_OS_LINUX) && TARGET_OS_LINUX
40
41 #include <libhostile/function.h>
42 #include <libhostile/initialize.h>
43
44 #include <dlfcn.h>
45 #include <pthread.h>
46 #include <stdio.h>
47 #include <stdlib.h>
48
49 static int64_t function_cache_index= 0;
50 struct function_st function_cache[20];
51
print_function_cache_usage(void)52 void print_function_cache_usage(void)
53 {
54 for (struct function_st *ptr= function_cache;
55 ptr->name == NULL; ptr++)
56 {
57 if (ptr->frequency)
58 {
59 fprintf(stderr, "--------------------------------------------------------\n\n");
60 fprintf(stderr, "%s hostilized %d\n", ptr->name, ptr->frequency);
61 fprintf(stderr, "\n--------------------------------------------------------\n");
62 }
63 }
64 }
65
set_function(const char * name,const char * environ_name)66 struct function_st set_function(const char *name, const char *environ_name)
67 {
68 struct function_st set;
69
70 set.name= name;
71 set._used= 0;
72
73 (void)dlerror();
74 set.function.ptr= dlsym(RTLD_NEXT, set.name);
75
76 if (set.function.ptr == NULL)
77 {
78 fprintf(stderr, "libhostile: %s(%s)", set.name, dlerror());
79 exit(EXIT_FAILURE);
80 }
81
82 set._corrupt= false;
83
84 if (set.function.ptr == NULL)
85 {
86 fprintf(stderr, "libhostile: %s(%s)", set.name, dlerror());
87 exit(EXIT_FAILURE);
88 }
89
90 char *ptr;
91 if ((ptr= getenv(environ_name)))
92 {
93 set.frequency= atoi(ptr);
94 }
95 else
96 {
97 set.frequency= 0;
98 }
99
100 if (set.frequency)
101 {
102 fprintf(stderr, "--------------------------------------------------------\n\n");
103 fprintf(stderr, "\t\tHostile Engaged -> %s\n\n", set.name);
104 fprintf(stderr, "Frequency used %d\n", set.frequency);
105 fprintf(stderr, "\n--------------------------------------------------------\n");
106 }
107
108 function_cache[function_cache_index]= set;
109 function_cache_index++;
110
111 print_function_cache_usage();
112
113 return set;
114 }
115
116 #endif
117