darshan-logutils.h 4.31 KB
Newer Older
1
/*
Shane Snyder's avatar
Shane Snyder committed
2 3 4
 * Copyright (C) 2015 University of Chicago.
 * See COPYRIGHT notice in top-level directory.
 *
5 6
 */

7 8
#ifndef __DARSHAN_LOG_UTILS_H
#define __DARSHAN_LOG_UTILS_H
9

10
#include <linux/limits.h>
11 12 13 14 15
#include <zlib.h>
#ifdef HAVE_LIBBZ2
#include <bzlib.h>
#endif

16
#include "uthash-1.9.2/src/uthash.h"
17

18 19
#include "darshan-log-format.h"

20 21 22 23
/* TODO: can we refactor this def out of header? modules currently poke at swap_flag
 * directly, but other than that there's no reason for another module to know this
 * definition.
 */
24 25
struct darshan_fd_s
{
26
    int fildes;
27
    int o_flags;
28
    int64_t pos;
29
    enum darshan_comp_type comp_type;
30
    char logfile_path[PATH_MAX];
31 32
    char version[8];
    int swap_flag;
33 34
    char *exe_mnt_data;
    struct darshan_log_map job_map;
35 36
    struct darshan_log_map rec_map;
    struct darshan_log_map mod_map[DARSHAN_MAX_MODS];
37 38 39 40 41 42 43 44
    int err;

    /* XXX */
    void *dz_strm;
    int dz_size;
    unsigned char *dz_buf;
    int dz_eor;
    int dz_prev_reg_id;
45
};
46
typedef struct darshan_fd_s* darshan_fd;
47

48 49 50 51 52 53
struct darshan_record_ref
{
    struct darshan_record rec;
    UT_hash_handle hlink;
};

54 55 56
struct darshan_mod_logutil_funcs
{
    int (*log_get_record)(
57 58 59 60 61 62 63
        darshan_fd fd,
        void* buf,
        darshan_record_id* rec_id
    );
    int (*log_put_record)(
        darshan_fd fd,
        void *buf
64 65 66 67 68 69 70 71 72 73 74 75 76
    );
    void (*log_print_record)(
        void *file_rec,
        char *file_name,
        char *mnt_pt,
        char *fs_type
    );
};

extern struct darshan_mod_logutil_funcs *mod_logutils[DARSHAN_MAX_MODS];

#include "darshan-posix-logutils.h"
#include "darshan-mpiio-logutils.h"
77
#include "darshan-hdf5-logutils.h"
78
#include "darshan-pnetcdf-logutils.h"
79
#include "darshan-bgq-logutils.h"
80

81 82
darshan_fd darshan_log_open(const char *name);
darshan_fd darshan_log_create(const char *name, enum darshan_comp_type comp_type);
83
int darshan_log_getjob(darshan_fd fd, struct darshan_job *job);
84
int darshan_log_putjob(darshan_fd fd, struct darshan_job *job);
85
int darshan_log_getexe(darshan_fd fd, char *buf);
86
int darshan_log_putexe(darshan_fd fd, char *buf);
87
int darshan_log_getmounts(darshan_fd fd, char*** mnt_pts,
88
    char*** fs_types, int* count);
89 90
int darshan_log_putmounts(darshan_fd fd, char** mnt_pts,
    char** fs_types, int count);
91
int darshan_log_gethash(darshan_fd fd, struct darshan_record_ref **hash);
92 93
int darshan_log_puthash(darshan_fd fd, struct darshan_record_ref *hash);
int darshan_log_getmod(darshan_fd fd, darshan_module_id mod_id,
94
    void *buf, int len);
95 96
int darshan_log_putmod(darshan_fd fd, darshan_module_id mod_id,
    void *mod_buf, int mod_buf_sz);
97
void darshan_log_close(darshan_fd file);
98

99 100 101 102 103 104 105 106 107 108 109
/* convenience macros for printing Darshan counters */
#define DARSHAN_PRINT_HEADER() \
    printf("\n#<module>\t<rank>\t<file>\t<counter>\t<value>" \
           "\t<file name>\t<mount pt>\t<fs type>\n")

#define DARSHAN_COUNTER_PRINT(__mod_name, __rank, __file_id, \
                              __counter, __counter_val, __file_name, \
                              __mnt_pt, __fs_type) do { \
    printf("%s\t%" PRId64 "\t%" PRIu64 "\t%s\t%" PRId64 "\t%s\t%s\t%s\n", \
        __mod_name, __rank, __file_id, __counter, __counter_val, \
        __file_name, __mnt_pt, __fs_type); \
110
} while(0)
111 112 113 114 115 116 117

#define DARSHAN_F_COUNTER_PRINT(__mod_name, __rank, __file_id, \
                                __counter, __counter_val, __file_name, \
                                __mnt_pt, __fs_type) do { \
    printf("%s\t%" PRId64 "\t%" PRIu64 "\t%s\t%f\t%s\t%s\t%s\n", \
        __mod_name, __rank, __file_id, __counter, __counter_val, \
        __file_name, __mnt_pt, __fs_type); \
118 119
} while(0)

120 121 122 123 124 125 126 127 128 129 130 131 132 133
/* naive byte swap implementation */
#define DARSHAN_BSWAP64(__ptr) do {\
    char __dst_char[8]; \
    char* __src_char = (char*)__ptr; \
    __dst_char[0] = __src_char[7]; \
    __dst_char[1] = __src_char[6]; \
    __dst_char[2] = __src_char[5]; \
    __dst_char[3] = __src_char[4]; \
    __dst_char[4] = __src_char[3]; \
    __dst_char[5] = __src_char[2]; \
    __dst_char[6] = __src_char[1]; \
    __dst_char[7] = __src_char[0]; \
    memcpy(__ptr, __dst_char, 8); \
} while(0)
134 135 136 137 138 139 140 141 142 143
#define DARSHAN_BSWAP32(__ptr) do {\
    char __dst_char[4]; \
    char* __src_char = (char*)__ptr; \
    __dst_char[0] = __src_char[3]; \
    __dst_char[1] = __src_char[2]; \
    __dst_char[2] = __src_char[1]; \
    __dst_char[3] = __src_char[0]; \
    memcpy(__ptr, __dst_char, 4); \
} while(0)

144
#endif