ioshark/ioshark_bench.h

   1 /*
   2  * Copyright (C) 2016 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #ifdef IOSHARK_MAIN
  18 const char *IO_op[] = {
  19         "LSEEK",
  20         "LLSEEK",
  21         "PREAD64",
  22         "PWRITE64",
  23         "READ",
  24         "WRITE",
  25         "MMAP",
  26         "MMAP2",
  27         "OPEN",
  28         "FSYNC",
  29         "FDATASYNC",
  30         "CLOSE",
  31         "MAPPED_PREAD",
  32         "MAPPED_PWRITE",
  33         "MAX_FILE_OP"
  34 };
  35 #endif
  36
  37 #define MAX(A, B)       ((A) > (B) ? (A) : (B))
  38 #define MIN(A, B)       ((A) < (B) ? (A) : (B))
  39
  40 #define MINBUFLEN       (16*1024)
  41
  42 #define FILE_DB_HASHSIZE        8192
  43
  44 struct files_db_s {
  45         char *filename;
  46         int fileno;
  47         size_t  size;
  48         int fd;
  49         int readonly;
  50         int debug_open_flags;
  51         struct files_db_s *next;
  52 };
  53
  54 struct files_db_handle {
  55         struct files_db_s *files_db_buckets[FILE_DB_HASHSIZE];
  56 };
  57
  58 struct IO_operation_s {
  59         char *IO_op;
  60 };
  61
  62 struct rw_bytes_s {
  63         u_int64_t bytes_read;
  64         u_int64_t bytes_written;
  65 };
  66
  67 static inline void
  68 files_db_update_size(void *node, u_int64_t new_size)
  69 {
  70         struct files_db_s *db_node = (struct files_db_s *)node;
  71
  72         if (db_node->size < new_size)
  73                 db_node->size = new_size;
  74 }
  75
  76 static inline void
  77 files_db_update_filename(void *node, char *filename)
  78 {
  79         ((struct files_db_s *)node)->filename = strdup(filename);
  80 }
  81
  82 static inline int
  83 files_db_get_fileno(void *node)
  84 {
  85         return (((struct files_db_s *)node)->fileno);
  86 }
  87
  88 static inline int
  89 files_db_get_fd(void *node)
  90 {
  91         return (((struct files_db_s *)node)->fd);
  92 }
  93
  94 static inline char *
  95 files_db_get_filename(void *node)
  96 {
  97         return (((struct files_db_s *)node)->filename);
  98 }
  99
 100 static inline int
 101 files_db_readonly(void *node)
 102 {
 103         return (((struct files_db_s *)node)->readonly);
 104 }
 105
 106 static inline u_int64_t
 107 get_msecs(struct timeval *tv)
 108 {
 109         return ((tv->tv_sec * 1000) + (tv->tv_usec / 1000));
 110 }
 111
 112 static inline u_int64_t
 113 get_usecs(struct timeval *tv)
 114 {
 115         return (tv->tv_usec % 1000);
 116 }
 117
 118 static inline void
 119 update_delta_time(struct timeval *start,
 120                   struct timeval *destination)
 121 {
 122         struct timeval res, finish;
 123
 124         (void)gettimeofday(&finish, (struct timezone *)NULL);
 125         timersub(&finish, start, &res);
 126         timeradd(destination, &res, &finish);
 127         *destination = finish;
 128 }
 129
 130 void *files_db_create_handle(void);
 131 void *files_db_lookup_byfileno(void *handle, int fileno);
 132 void *files_db_add_byfileno(void *handle, int fileno, int readonly);
 133 void files_db_update_fd(void *node, int fd);
 134 void files_db_unlink_files(void *db_handle);
 135 void files_db_close_files(void *handle);
 136 void files_db_close_fd(void *node);
 137 void files_db_free_memory(void *handle);
 138 void create_file(char *path, size_t size,
 139                  struct rw_bytes_s *rw_bytes);
 140 char *get_buf(char **buf, int *buflen, int len, int do_fill);
 141 void files_db_fsync_discard_files(void *handle);
 142 void print_op_stats(u_int64_t *op_counts);
 143 void print_bytes(char *desc, struct rw_bytes_s *rw_bytes);
 144 void ioshark_handle_mmap(void *db_node,
 145                          struct ioshark_file_operation *file_op,
 146                          char **bufp, int *buflen, u_int64_t *op_counts,
 147                          struct rw_bytes_s *rw_bytes);
 148 void capture_util_state_before(void);
 149 void report_cpu_disk_util(void);
 150
 151 char *get_ro_filename(int ix);
 152 void init_filename_cache(void);
 153 void free_filename_cache(void);
 154 int is_readonly_mount(char *filename, size_t size);