driver/storage.h

   1 ////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //      FILE:           storage.h
   4 //
   5 //      DESCRIPTION:
   6 //      This file is C source for SWAP.
   7 //
   8 //      SEE ALSO:       storage.c
   9 //      AUTHOR:         L.Komkov, S.Dianov, A.Gerenkov
  10 //      COMPANY NAME:   Samsung Research Center in Moscow
  11 //      DEPT NAME:      Advanced Software Group
  12 //      CREATED:        2008.02.15
  13 //      VERSION:        1.0
  14 //      REVISION DATE:  2008.12.03
  15 //
  16 ////////////////////////////////////////////////////////////////////////////////////
  17
  18 #if !defined(__STORAGE_H__)
  19 #define __STORAGE_H__
  20
  21 #include <linux/mount.h>
  22 #include <linux/module.h>
  23 #include <linux/list.h>
  24 #include <linux/spinlock.h>
  25 #include "picl.h"
  26 #include "ec_ioctl.h"
  27 #include "ec_probe.h"
  28 #include "probes_manager.h"
  29 #include "probes.h"
  30 #include "event_tmpl.h"
  31 #include "error_storage.h"
  32
  33 //extern int update_errno_buffer(const char *buffer);
  34 //extern int get_last_error(void* u_addr);
  35 //extern int has_last_error(void);
  36 //extern void last_error_buffer_initialize(void);
  37 ///////////////////////////////////////////////////////////////////////////////////////////////////
  38
  39 extern int EnableContinuousRetrieval(void);
  40 extern int DisableContinuousRetrieval(void);
  41
  42 ///////////////////////////////////////////////////////////////////////////////////////////////////
  43
  44 extern unsigned int GetBufferSize(void);
  45 extern int SetBufferSize(unsigned int nSize);
  46 extern int ResetBuffer(void);
  47
  48 extern int SetPid(unsigned int pid);
  49
  50 //extern spinlock_t buffer_spinlock;
  51
  52 ///////////////////////////////////////////////////////////////////////////////////////////////////
  53
  54 /*
  55     Functions "storage_init()" and "storage_down()" are for initialization and
  56     shutdown respectively.
  57 */
  58 extern int storage_init (void);
  59 extern void storage_down (void);
  60
  61 /*
  62     Function "pack_event_info()" saves information about event into buffer. It
  63     is used in 'probes' to pack and save event data.
  64 */
  65 extern void pack_task_event_info (struct task_struct *task, probe_id_t probe_id,
  66                 record_type_t record_type, const char *fmt, ...);
  67 #define pack_event_info(probe_id, record_type, fmt, ...) \
  68         pack_task_event_info(current, probe_id, record_type, fmt, __VA_ARGS__)
  69
  70 /* Set most links from us_proc_info to data in the bundle */
  71 int link_bundle(void);
  72 /* Undo the actions of link_bundle() */
  73 void unlink_bundle(void);
  74
  75 /*
  76     Adds non-predefined kernel probe to the list.
  77 */
  78 extern int add_probe_to_list (unsigned long addr, kernel_probe_t ** pprobe);
  79
  80 /*
  81     Removes non-predefined kernel probe from the list.
  82 */
  83 extern int remove_probe_from_list (unsigned long addr);
  84
  85 /*
  86     Searches non-predefined kernel probe in the list.
  87 */
  88 extern kernel_probe_t *find_probe (unsigned long addr);
  89
  90 /*
  91     Copies event from user space to buffer and updates its pid/tid/cpu/time.
  92 */
  93 extern int put_us_event (char *data, unsigned long len);
  94
  95 /*
  96     Sets event mask.
  97 */
  98 extern int set_event_mask (int new_mask);
  99
 100 /*
 101     Gets event mask.
 102 */
 103 extern int get_event_mask (int *mask);
 104
 105 /*
 106     Sets predefined user space probes info.
 107 */
 108 extern int set_predef_uprobes (ioctl_predef_uprobes_info_t *data);
 109 /*
 110     Gets predefined user space probes info length.
 111 */
 112 extern int get_predef_uprobes_size(int *size);
 113 /*
 114     Gets predefined user space probes info.
 115 */
 116 extern int get_predef_uprobes(ioctl_predef_uprobes_info_t *data);
 117
 118
 119 // internal bookkeeping of storage
 120 extern char *p_buffer;
 121
 122 // list of selected non-predefined kernel probes
 123 extern struct hlist_head kernel_probes;
 124
 125 // multiple handlers stuff
 126 /*struct dbi_modules_handlers {
 127         struct list_head        modules_handlers;
 128         spinlock_t              lock;
 129 };
 130
 131 struct dbi_modules_handlers_info {
 132         struct module           *dbi_module;
 133         struct handler_map      *dbi_handlers;
 134         int                     dbi_nr_handlers;
 135         struct list_head        dbi_list_head;
 136 };
 137
 138 extern int dbi_register_handlers_module(struct dbi_modules_handlers_info *dbi_mhi);
 139 extern int dbi_unregister_handlers_module(struct dbi_modules_handlers_info *dbi_mhi);
 140 //unsigned long get_dbi_modules_handlers(void);*/
 141
 142 /* list of on-the-go installed kernel probes */
 143 extern struct hlist_head otg_kernel_probes;
 144
 145 // event mask
 146 extern int event_mask;
 147
 148 // process pid to instrument
 149 extern unsigned int inst_pid;
 150
 151 typedef struct
 152 {
 153         struct list_head list;
 154         char *name;
 155         int installed;
 156         struct jprobe jprobe;
 157         struct kretprobe retprobe;
 158         unsigned long offset;
 159         unsigned long got_addr;
 160
 161         unsigned flag_retprobe:1;
 162         unsigned flag_got:1;
 163 } us_proc_ip_t;
 164
 165 typedef struct
 166 {
 167         int installed;
 168         struct jprobe jprobe;
 169         unsigned long addr;
 170         struct list_head list;
 171 } us_proc_vtp_t;
 172
 173 typedef struct
 174 {
 175         char *name;
 176         char type;
 177         unsigned long size;
 178         signed char reg;        // -1 - memory, 0..127 - register number
 179         long off;
 180         struct list_head list;
 181 } us_proc_vtp_data_t;
 182
 183 typedef struct
 184 {
 185         unsigned func_addr;
 186         unsigned got_addr;
 187         unsigned real_func_addr;
 188 } us_proc_plt_t;
 189
 190 typedef struct
 191 {
 192         char *path;
 193         char *path_dyn;
 194         struct dentry *m_f_dentry;
 195         unsigned ips_count;
 196         us_proc_ip_t *p_ips;
 197         unsigned vtps_count;
 198         us_proc_vtp_t *p_vtps;
 199         int loaded;
 200         unsigned plt_count;
 201         us_proc_plt_t *p_plt;
 202         unsigned long vma_start;
 203         unsigned long vma_end;
 204         unsigned vma_flag;
 205 } us_proc_lib_t;
 206
 207
 208 typedef struct
 209 {
 210         char *path;
 211         struct dentry *m_f_dentry;
 212         pid_t tgid;
 213         unsigned unres_ips_count;
 214         unsigned unres_vtps_count;
 215         //kprobe_opcode_t *mapped_codelets;
 216         int is_plt;
 217         unsigned libs_count;
 218         us_proc_lib_t *p_libs;
 219
 220         // new_dpf
 221         struct sspt_procs *pp;
 222 } inst_us_proc_t;
 223
 224 typedef struct
 225 {
 226         unsigned int addr;
 227         unsigned int inst_type;
 228         char *name;
 229         char *class_name;
 230         char *method_name;
 231         char *prototype;
 232
 233 }dex_proc_ip_t;
 234
 235 typedef struct
 236 {
 237         char *path;
 238         unsigned ips_count;
 239         dex_proc_ip_t *p_ips;
 240
 241 }inst_dex_proc_t;
 242
 243
 244 struct cond {
 245         /* cond data itself */
 246         struct event_tmpl tmpl;
 247         /* linked list */
 248         struct list_head list;
 249         /* has been applied (for start and stop conditions) */
 250         int applied;
 251 };
 252
 253 extern struct cond cond_list;
 254
 255 /* macros for testing flags */
 256 #define ET_FIELD_CLR(flags, field) (flags &= ~field)
 257 #define ET_FIELD_SET(flags, field) (flags |= field)
 258 #define ET_FIELD_ISSET(flags, field) ((flags & field) != 0)
 259
 260 extern inst_us_proc_t us_proc_info;
 261 extern inst_dex_proc_t dex_proc_info;
 262
 263 #endif /* !defined(__STORAGE_H__) */