darshan-posix.c 51.2 KB
Newer Older
1
2
3
4
5
/*
 *  (C) 2009 by Argonne National Laboratory.
 *      See COPYRIGHT in top-level directory.
 */

6
#include "darshan-runtime-config.h"
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdarg.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/uio.h>
#include <sys/mman.h>
#include <search.h>
#include <assert.h>
21
#include <libgen.h>
22
#include <limits.h>
23
24
#define __USE_GNU
#include <pthread.h>
25
26
27

#include "darshan.h"

28
#ifndef HAVE_OFF64_T
29
30
31
typedef int64_t off64_t;
#endif

32
33
extern char* __progname_full;

34
#ifdef DARSHAN_PRELOAD
35
36
37
#define __USE_GNU
#include <dlfcn.h>
#include <stdlib.h>
38

39
40
41
42
43
#define DARSHAN_FORWARD_DECL(name,ret,args) \
  ret (*__real_ ## name)args = NULL;

#define DARSHAN_DECL(__name) __name

44
45
46
47
48
49
50
51
52
53
54
55
56
#define DARSHAN_MPI_CALL(func) __real_ ## func

#define MAP_OR_FAIL(func) \
    if (!(__real_ ## func)) \
    { \
        __real_ ## func = dlsym(RTLD_NEXT, #func); \
        if(!(__real_ ## func)) { \
           fprintf(stderr, "Darshan failed to map symbol: %s\n", #func); \
           exit(1); \
       } \
    }


57
extern double (*__real_PMPI_Wtime)(void);
58

59
60
61
62
63
64
65
#else

#define DARSHAN_FORWARD_DECL(name,ret,args) \
  extern ret __real_ ## name args;

#define DARSHAN_DECL(__name) __wrap_ ## __name

66
67
68
69
#define MAP_OR_FAIL(func)

#define DARSHAN_MPI_CALL(func) func

70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#endif

DARSHAN_FORWARD_DECL(creat, int, (const char* path, mode_t mode));
DARSHAN_FORWARD_DECL(creat64, int, (const char* path, mode_t mode));
DARSHAN_FORWARD_DECL(open, int, (const char *path, int flags, ...));
DARSHAN_FORWARD_DECL(open64, int, (const char *path, int flags, ...));
DARSHAN_FORWARD_DECL(close, int, (int fd));
DARSHAN_FORWARD_DECL(write, ssize_t, (int fd, const void *buf, size_t count));
DARSHAN_FORWARD_DECL(read, ssize_t, (int fd, void *buf, size_t count));
DARSHAN_FORWARD_DECL(lseek, off_t, (int fd, off_t offset, int whence));
DARSHAN_FORWARD_DECL(lseek64, off64_t, (int fd, off64_t offset, int whence));
DARSHAN_FORWARD_DECL(pread, ssize_t, (int fd, void *buf, size_t count, off_t offset));
DARSHAN_FORWARD_DECL(pread64, ssize_t, (int fd, void *buf, size_t count, off64_t offset));
DARSHAN_FORWARD_DECL(pwrite, ssize_t, (int fd, const void *buf, size_t count, off_t offset));
DARSHAN_FORWARD_DECL(pwrite64, ssize_t, (int fd, const void *buf, size_t count, off64_t offset
));
DARSHAN_FORWARD_DECL(readv, ssize_t, (int fd, const struct iovec *iov, int iovcnt));
DARSHAN_FORWARD_DECL(writev, ssize_t, (int fd, const struct iovec *iov, int iovcnt));
DARSHAN_FORWARD_DECL(__fxstat, int, (int vers, int fd, struct stat *buf));
DARSHAN_FORWARD_DECL(__fxstat64, int, (int vers, int fd, struct stat64 *buf));
DARSHAN_FORWARD_DECL(__lxstat, int, (int vers, const char* path, struct stat *buf));
DARSHAN_FORWARD_DECL(__lxstat64, int, (int vers, const char* path, struct stat64 *buf));
DARSHAN_FORWARD_DECL(__xstat, int, (int vers, const char* path, struct stat *buf));
DARSHAN_FORWARD_DECL(__xstat64, int, (int vers, const char* path, struct stat64 *buf));
DARSHAN_FORWARD_DECL(mmap, void*, (void *addr, size_t length, int prot, int flags, int fd, off_t offset));
DARSHAN_FORWARD_DECL(mmap64, void*, (void *addr, size_t length, int prot, int flags, int fd, off64_t offset));
DARSHAN_FORWARD_DECL(fopen, FILE*, (const char *path, const char *mode));
DARSHAN_FORWARD_DECL(fopen64, FILE*, (const char *path, const char *mode));
DARSHAN_FORWARD_DECL(fclose, int, (FILE *fp));
DARSHAN_FORWARD_DECL(fread, size_t, (void *ptr, size_t size, size_t nmemb, FILE *stream));
DARSHAN_FORWARD_DECL(fwrite, size_t, (const void *ptr, size_t size, size_t nmemb, FILE *stream));
DARSHAN_FORWARD_DECL(fseek, int, (FILE *stream, long offset, int whence));
DARSHAN_FORWARD_DECL(fsync, int, (int fd));
DARSHAN_FORWARD_DECL(fdatasync, int, (int fd));

105
pthread_mutex_t cp_mutex = PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP;
106
107
struct darshan_job_runtime* darshan_global_job = NULL;
static int my_rank = -1;
108
static struct stat64 cp_stat_buf;
109
static int darshan_mem_alignment = 1;
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

/* these are paths that we will not trace */
static char* exclusions[] = {
"/etc/",
"/dev/",
"/usr/",
"/bin/",
"/boot/",
"/lib/",
"/opt/",
"/sbin/",
"/sys/",
"/proc/",
NULL
};

126
127
static double posix_wtime(void);

128
129
static void cp_access_counter(struct darshan_file_runtime* file, ssize_t size,     enum cp_counter_type type);

130
131
132
133
134
135
136
137
static struct darshan_file_ref* ref_by_handle(
    const void* handle,
    int handle_sz,
    enum darshan_handle_type handle_type);

static struct darshan_file_runtime* darshan_file_by_fd(int fd);
static void darshan_file_close_fd(int fd);
static struct darshan_file_runtime* darshan_file_by_name_setfd(const char* name, int fd);
138
static char* clean_path(const char* path);
139

Philip Carns's avatar
Philip Carns committed
140
#define CP_RECORD_WRITE(__ret, __fd, __count, __pwrite_flag, __pwrite_offset, __aligned, __stream_flag, __tm1, __tm2) do{ \
141
    size_t stride; \
Philip Carns's avatar
Philip Carns committed
142
    int64_t this_offset; \
143
144
    int64_t file_alignment; \
    struct darshan_file_runtime* file; \
145
    double __elapsed = __tm2-__tm1; \
146
147
148
    if(__ret < 0) break; \
    file = darshan_file_by_fd(__fd); \
    if(!file) break; \
Philip Carns's avatar
Philip Carns committed
149
150
151
152
    if(__pwrite_flag) \
        this_offset = __pwrite_offset; \
    else \
        this_offset = file->offset; \
153
    file_alignment = CP_VALUE(file, CP_FILE_ALIGNMENT); \
Philip Carns's avatar
Philip Carns committed
154
    if(this_offset > file->last_byte_written) \
155
        CP_INC(file, CP_SEQ_WRITES, 1); \
Philip Carns's avatar
Philip Carns committed
156
    if(this_offset == (file->last_byte_written + 1)) \
157
        CP_INC(file, CP_CONSEC_WRITES, 1); \
Philip Carns's avatar
Philip Carns committed
158
    if(this_offset > 0 && this_offset > file->last_byte_written \
159
        && file->last_byte_written != 0) \
Philip Carns's avatar
Philip Carns committed
160
        stride = this_offset - file->last_byte_written - 1; \
161
162
    else \
        stride = 0; \
Philip Carns's avatar
Philip Carns committed
163
164
165
    file->last_byte_written = this_offset + __ret - 1; \
    file->offset = this_offset + __ret; \
    CP_MAX(file, CP_MAX_BYTE_WRITTEN, (this_offset + __ret -1)); \
166
167
168
169
170
171
172
173
174
    CP_INC(file, CP_BYTES_WRITTEN, __ret); \
    if(__stream_flag) \
        CP_INC(file, CP_POSIX_FWRITES, 1); \
    else \
        CP_INC(file, CP_POSIX_WRITES, 1); \
    CP_BUCKET_INC(file, CP_SIZE_WRITE_0_100, __ret); \
    cp_access_counter(file, stride, CP_COUNTER_STRIDE); \
    if(!__aligned) \
        CP_INC(file, CP_MEM_NOT_ALIGNED, 1); \
Philip Carns's avatar
Philip Carns committed
175
    if(file_alignment > 0 && (this_offset % file_alignment) != 0) \
176
177
178
179
180
        CP_INC(file, CP_FILE_NOT_ALIGNED, 1); \
    cp_access_counter(file, __ret, CP_COUNTER_ACCESS); \
    if(file->last_io_type == CP_READ) \
        CP_INC(file, CP_RW_SWITCHES, 1); \
    file->last_io_type = CP_WRITE; \
181
    CP_F_INC_NO_OVERLAP(file, __tm1, __tm2, file->last_posix_write_end, CP_F_POSIX_WRITE_TIME); \
182
183
184
    if(CP_F_VALUE(file, CP_F_WRITE_START_TIMESTAMP) == 0) \
        CP_F_SET(file, CP_F_WRITE_START_TIMESTAMP, __tm1); \
    CP_F_SET(file, CP_F_WRITE_END_TIMESTAMP, __tm2); \
185
186
187
    if(CP_F_VALUE(file, CP_F_MAX_WRITE_TIME) < __elapsed){ \
        CP_F_SET(file, CP_F_MAX_WRITE_TIME, __elapsed); \
        CP_SET(file, CP_MAX_WRITE_TIME_SIZE, __ret); } \
188
189
} while(0)

190
#define CP_RECORD_READ(__ret, __fd, __count, __pread_flag, __pread_offset, __aligned, __stream_flag, __tm1, __tm2) do{ \
191
    size_t stride; \
192
    int64_t this_offset; \
193
194
    struct darshan_file_runtime* file; \
    int64_t file_alignment; \
195
    double __elapsed = __tm2-__tm1; \
196
197
198
    if(__ret < 0) break; \
    file = darshan_file_by_fd(__fd); \
    if(!file) break; \
199
200
201
202
    if(__pread_flag)\
        this_offset = __pread_offset; \
    else \
        this_offset = file->offset; \
203
    file_alignment = CP_VALUE(file, CP_FILE_ALIGNMENT); \
204
    if(this_offset > file->last_byte_read) \
205
        CP_INC(file, CP_SEQ_READS, 1); \
206
    if(this_offset == (file->last_byte_read + 1)) \
207
        CP_INC(file, CP_CONSEC_READS, 1); \
208
    if(this_offset > 0 && this_offset > file->last_byte_read \
209
        && file->last_byte_read != 0) \
210
        stride = this_offset - file->last_byte_read - 1; \
211
212
    else \
        stride = 0; \
213
214
215
    file->last_byte_read = this_offset + __ret - 1; \
    CP_MAX(file, CP_MAX_BYTE_READ, (this_offset + __ret -1)); \
    file->offset = this_offset + __ret; \
216
217
218
219
220
221
222
223
224
    CP_INC(file, CP_BYTES_READ, __ret); \
    if(__stream_flag)\
        CP_INC(file, CP_POSIX_FREADS, 1); \
    else\
        CP_INC(file, CP_POSIX_READS, 1); \
    CP_BUCKET_INC(file, CP_SIZE_READ_0_100, __ret); \
    cp_access_counter(file, stride, CP_COUNTER_STRIDE); \
    if(!__aligned) \
        CP_INC(file, CP_MEM_NOT_ALIGNED, 1); \
Philip Carns's avatar
Philip Carns committed
225
    if(file_alignment > 0 && (this_offset % file_alignment) != 0) \
226
227
228
229
230
        CP_INC(file, CP_FILE_NOT_ALIGNED, 1); \
    cp_access_counter(file, __ret, CP_COUNTER_ACCESS); \
    if(file->last_io_type == CP_WRITE) \
        CP_INC(file, CP_RW_SWITCHES, 1); \
    file->last_io_type = CP_READ; \
231
    CP_F_INC_NO_OVERLAP(file, __tm1, __tm2, file->last_posix_read_end, CP_F_POSIX_READ_TIME); \
232
233
234
    if(CP_F_VALUE(file, CP_F_READ_START_TIMESTAMP) == 0) \
        CP_F_SET(file, CP_F_READ_START_TIMESTAMP, __tm1); \
    CP_F_SET(file, CP_F_READ_END_TIMESTAMP, __tm2); \
235
236
237
    if(CP_F_VALUE(file, CP_F_MAX_READ_TIME) < __elapsed){ \
        CP_F_SET(file, CP_F_MAX_READ_TIME, __elapsed); \
        CP_SET(file, CP_MAX_READ_TIME_SIZE, __ret); } \
238
239
} while(0)

240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
#define CP_LOOKUP_RECORD_STAT(__path, __statbuf, __tm1, __tm2) do { \
    char* exclude; \
    int tmp_index = 0; \
    struct darshan_file_runtime* file; \
    while((exclude = exclusions[tmp_index])) { \
        if(!(strncmp(exclude, __path, strlen(exclude)))) \
            break; \
        tmp_index++; \
    } \
    if(exclude) break; \
    file = darshan_file_by_name(__path); \
    if (file) \
    { \
        CP_RECORD_STAT(file, __statbuf, __tm1, __tm2); \
    } \
} while(0)

    
258
#define CP_RECORD_STAT(__file, __statbuf, __tm1, __tm2) do { \
259
    if(!CP_VALUE((__file), CP_POSIX_STATS) && !CP_VALUE((__file), CP_POSIX_OPENS)){ \
260
261
262
263
264
        CP_SET((__file), CP_DEVICE, (__statbuf)->st_dev); \
        CP_SET((__file), CP_FILE_ALIGNMENT, (__statbuf)->st_blksize); \
        CP_SET((__file), CP_SIZE_AT_OPEN, (__statbuf)->st_size); \
    }\
    (__file)->log_file->rank = my_rank; \
265
    CP_F_INC_NO_OVERLAP(file, __tm1, __tm2, file->last_posix_meta_end, CP_F_POSIX_META_TIME); \
266
267
268
    CP_INC(__file, CP_POSIX_STATS, 1); \
} while(0)

269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
static inline dev_t get_device(const char* path, struct stat64* statbuf)
{
    dev_t device = 0;

#ifdef __CP_ST_DEV_WORKAROUND
    struct stat64 dirstat;
    char* tmp_path = strdup(path);
    char* parent = dirname(tmp_path);
    if(parent && stat64(parent, &dirstat) == 0)
        device = dirstat.st_dev;
    if(tmp_path) free(tmp_path);
#else
    device = statbuf->st_dev;
#endif

    return(device);
}

Philip Carns's avatar
Philip Carns committed
287
#ifdef __CP_STAT_AT_OPEN
288
#define CP_STAT_FILE(_f, _p, _r) do { \
289
    if(!CP_VALUE((_f), CP_POSIX_STATS) && !CP_VALUE((_f), CP_POSIX_OPENS)){ \
290
        if(fstat64(_r, &cp_stat_buf) == 0) { \
Philip Carns's avatar
Philip Carns committed
291
292
293
294
295
296
297
            CP_SET(_f, CP_DEVICE, get_device(_p, &cp_stat_buf)); \
            CP_SET(_f, CP_FILE_ALIGNMENT, cp_stat_buf.st_blksize); \
            CP_SET(_f, CP_SIZE_AT_OPEN, cp_stat_buf.st_size); \
        }\
    }\
}while(0)
#else
298
#define CP_STAT_FILE(_f, _p, _r) do { }while(0)
Philip Carns's avatar
Philip Carns committed
299
300
#endif

301
302
303
304
305
306
307
308
309
310
311
#define CP_RECORD_OPEN(__ret, __path, __mode, __stream_flag, __tm1, __tm2) do { \
    struct darshan_file_runtime* file; \
    char* exclude; \
    int tmp_index = 0; \
    if(__ret < 0) break; \
    while((exclude = exclusions[tmp_index])) { \
        if(!(strncmp(exclude, __path, strlen(exclude)))) \
            break; \
        tmp_index++; \
    } \
    if(exclude) break; \
312
    file = darshan_file_by_name_setfd(__path, __ret); \
313
    if(!file) break; \
314
    CP_STAT_FILE(file, __path, __ret); \
315
316
317
318
319
320
321
322
323
324
325
    file->log_file->rank = my_rank; \
    if(__mode) \
        CP_SET(file, CP_MODE, __mode); \
    file->offset = 0; \
    file->last_byte_written = 0; \
    file->last_byte_read = 0; \
    if(__stream_flag)\
        CP_INC(file, CP_POSIX_FOPENS, 1); \
    else \
        CP_INC(file, CP_POSIX_OPENS, 1); \
    if(CP_F_VALUE(file, CP_F_OPEN_TIMESTAMP) == 0) \
326
        CP_F_SET(file, CP_F_OPEN_TIMESTAMP, posix_wtime()); \
327
    CP_F_INC_NO_OVERLAP(file, __tm1, __tm2, file->last_posix_meta_end, CP_F_POSIX_META_TIME); \
328
329
} while (0)

330
int DARSHAN_DECL(close)(int fd)
331
332
333
334
335
336
{
    struct darshan_file_runtime* file;
    int tmp_fd = fd;
    double tm1, tm2;
    int ret;

337
338
    MAP_OR_FAIL(close);

339
340
341
342
343
344
345
346
347
348
    tm1 = darshan_wtime();
    ret = __real_close(fd);
    tm2 = darshan_wtime();

    CP_LOCK();
    file = darshan_file_by_fd(tmp_fd);
    if(file)
    {
        file->last_byte_written = 0;
        file->last_byte_read = 0;
349
        CP_F_SET(file, CP_F_CLOSE_TIMESTAMP, posix_wtime());
350
        CP_F_INC_NO_OVERLAP(file, tm1, tm2, file->last_posix_meta_end, CP_F_POSIX_META_TIME);
351
        darshan_file_close_fd(tmp_fd);
352
353
354
355
356
357
    }
    CP_UNLOCK();

    return(ret);
}

358
int DARSHAN_DECL(fclose)(FILE *fp)
359
360
361
362
363
364
{
    struct darshan_file_runtime* file;
    int tmp_fd = fileno(fp);
    double tm1, tm2;
    int ret;

365
366
    MAP_OR_FAIL(fclose);

367
368
369
370
371
372
373
374
375
376
    tm1 = darshan_wtime();
    ret = __real_fclose(fp);
    tm2 = darshan_wtime();
    
    CP_LOCK();
    file = darshan_file_by_fd(tmp_fd);
    if(file)
    {
        file->last_byte_written = 0;
        file->last_byte_read = 0;
377
        CP_F_SET(file, CP_F_CLOSE_TIMESTAMP, posix_wtime());
378
        CP_F_INC_NO_OVERLAP(file, tm1, tm2, file->last_posix_meta_end, CP_F_POSIX_META_TIME);
379
        darshan_file_close_fd(tmp_fd);
380
381
382
383
384
385
386
    }
    CP_UNLOCK();

    return(ret);
}


387
int DARSHAN_DECL(fsync)(int fd)
388
389
390
391
392
{
    int ret;
    struct darshan_file_runtime* file;
    double tm1, tm2;

393
394
    MAP_OR_FAIL(fsync);

395
396
397
398
399
400
401
402
403
404
405
    tm1 = darshan_wtime();
    ret = __real_fsync(fd);
    tm2 = darshan_wtime();

    if(ret < 0)
        return(ret);

    CP_LOCK();
    file = darshan_file_by_fd(fd);
    if(file)
    {
406
        CP_F_INC_NO_OVERLAP(file, tm1, tm2, file->last_posix_write_end, CP_F_POSIX_WRITE_TIME); \
407
408
409
410
411
412
413
        CP_INC(file, CP_POSIX_FSYNCS, 1);
    }
    CP_UNLOCK();

    return(ret);
}

414
int DARSHAN_DECL(fdatasync)(int fd)
415
416
417
418
419
{
    int ret;
    struct darshan_file_runtime* file;
    double tm1, tm2;

420
421
    MAP_OR_FAIL(fdatasync);

422
423
424
425
426
427
428
429
430
431
    tm1 = darshan_wtime();
    ret = __real_fdatasync(fd);
    tm2 = darshan_wtime();
    if(ret < 0)
        return(ret);

    CP_LOCK();
    file = darshan_file_by_fd(fd);
    if(file)
    {
432
        CP_F_INC_NO_OVERLAP(file, tm1, tm2, file->last_posix_write_end, CP_F_POSIX_WRITE_TIME); \
433
434
435
436
437
438
439
440
        CP_INC(file, CP_POSIX_FDSYNCS, 1);
    }
    CP_UNLOCK();

    return(ret);
}


441
void* DARSHAN_DECL(mmap64)(void *addr, size_t length, int prot, int flags,
442
443
444
445
446
    int fd, off64_t offset)
{
    void* ret;
    struct darshan_file_runtime* file;

447
448
    MAP_OR_FAIL(mmap64);

449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
    ret = __real_mmap64(addr, length, prot, flags, fd, offset);
    if(ret == MAP_FAILED)
        return(ret);

    CP_LOCK();
    file = darshan_file_by_fd(fd);
    if(file)
    {
        CP_INC(file, CP_POSIX_MMAPS, 1);
    }
    CP_UNLOCK();

    return(ret);
}


465
void* DARSHAN_DECL(mmap)(void *addr, size_t length, int prot, int flags,
466
467
468
469
470
    int fd, off_t offset)
{
    void* ret;
    struct darshan_file_runtime* file;

471
472
    MAP_OR_FAIL(mmap);

473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
    ret = __real_mmap(addr, length, prot, flags, fd, offset);
    if(ret == MAP_FAILED)
        return(ret);

    CP_LOCK();
    file = darshan_file_by_fd(fd);
    if(file)
    {
        CP_INC(file, CP_POSIX_MMAPS, 1);
    }
    CP_UNLOCK();

    return(ret);
}

488
int DARSHAN_DECL(creat)(const char* path, mode_t mode)
489
490
491
492
{
    int ret;
    double tm1, tm2;

493
494
    MAP_OR_FAIL(creat);

495
496
497
498
499
500
501
502
503
504
505
    tm1 = darshan_wtime();
    ret = __real_creat(path, mode);
    tm2 = darshan_wtime();

    CP_LOCK();
    CP_RECORD_OPEN(ret, path, mode, 0, tm1, tm2);
    CP_UNLOCK();

    return(ret);
}

506
int DARSHAN_DECL(creat64)(const char* path, mode_t mode)
507
508
509
510
{
    int ret;
    double tm1, tm2;

511
512
    MAP_OR_FAIL(creat64);

513
514
515
516
517
518
519
520
521
522
523
    tm1 = darshan_wtime();
    ret = __real_creat64(path, mode);
    tm2 = darshan_wtime();

    CP_LOCK();
    CP_RECORD_OPEN(ret, path, mode, 0, tm1, tm2);
    CP_UNLOCK();

    return(ret);
}

524
int DARSHAN_DECL(open64)(const char* path, int flags, ...)
525
526
527
528
529
{
    int mode = 0;
    int ret;
    double tm1, tm2;

530
531
    MAP_OR_FAIL(open64);

532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
    if (flags & O_CREAT) 
    {
        va_list arg;
        va_start(arg, flags);
        mode = va_arg(arg, int);
        va_end(arg);

        tm1 = darshan_wtime();
        ret = __real_open64(path, flags, mode);
        tm2 = darshan_wtime();
    }
    else
    {
        tm1 = darshan_wtime();
        ret = __real_open64(path, flags);
        tm2 = darshan_wtime();
    }

    CP_LOCK();
    CP_RECORD_OPEN(ret, path, mode, 0, tm1, tm2);
    CP_UNLOCK();

    return(ret);
}

557
int DARSHAN_DECL(open)(const char *path, int flags, ...)
558
559
560
561
562
{
    int mode = 0;
    int ret;
    double tm1, tm2;

563
564
    MAP_OR_FAIL(open);

565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
    if (flags & O_CREAT) 
    {
        va_list arg;
        va_start(arg, flags);
        mode = va_arg(arg, int);
        va_end(arg);

        tm1 = darshan_wtime();
        ret = __real_open(path, flags, mode);
        tm2 = darshan_wtime();
    }
    else
    {
        tm1 = darshan_wtime();
        ret = __real_open(path, flags);
        tm2 = darshan_wtime();
    }

    CP_LOCK();
    CP_RECORD_OPEN(ret, path, mode, 0, tm1, tm2);
    CP_UNLOCK();

    return(ret);
}

590
FILE* DARSHAN_DECL(fopen64)(const char *path, const char *mode)
591
592
593
594
595
{
    FILE* ret;
    int fd;
    double tm1, tm2;

596
597
    MAP_OR_FAIL(fopen64);

598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
    tm1 = darshan_wtime();
    ret = __real_fopen64(path, mode);
    tm2 = darshan_wtime();
    if(ret == 0)
        fd = -1;
    else
        fd = fileno(ret);

    CP_LOCK();
    CP_RECORD_OPEN(fd, path, 0, 1, tm1, tm2);
    CP_UNLOCK();

    return(ret);
}

613
FILE* DARSHAN_DECL(fopen)(const char *path, const char *mode)
614
615
616
617
618
{
    FILE* ret;
    int fd;
    double tm1, tm2;

619
620
    MAP_OR_FAIL(fopen);

621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
    tm1 = darshan_wtime();
    ret = __real_fopen(path, mode);
    tm2 = darshan_wtime();
    if(ret == 0)
        fd = -1;
    else
        fd = fileno(ret);

    CP_LOCK();
    CP_RECORD_OPEN(fd, path, 0, 1, tm1, tm2);
    CP_UNLOCK();

    return(ret);
}

636
int DARSHAN_DECL(__xstat64)(int vers, const char *path, struct stat64 *buf)
637
638
639
640
{
    int ret;
    double tm1, tm2;

641
642
    MAP_OR_FAIL(__xstat64);

643
644
645
646
647
648
649
    tm1 = darshan_wtime();
    ret = __real___xstat64(vers, path, buf);
    tm2 = darshan_wtime();
    if(ret < 0 || !S_ISREG(buf->st_mode))
        return(ret);

    CP_LOCK();
650
    CP_LOOKUP_RECORD_STAT(path, buf, tm1, tm2);
651
652
653
654
655
    CP_UNLOCK();

    return(ret);
}

656
int DARSHAN_DECL(__lxstat64)(int vers, const char *path, struct stat64 *buf)
657
658
659
660
{
    int ret;
    double tm1, tm2;

661
662
    MAP_OR_FAIL(__lxstat64);

663
664
665
666
667
668
669
    tm1 = darshan_wtime();
    ret = __real___lxstat64(vers, path, buf);
    tm2 = darshan_wtime();
    if(ret < 0 || !S_ISREG(buf->st_mode))
        return(ret);

    CP_LOCK();
670
    CP_LOOKUP_RECORD_STAT(path, buf, tm1, tm2);
671
672
673
674
675
    CP_UNLOCK();

    return(ret);
}

676
int DARSHAN_DECL(__fxstat64)(int vers, int fd, struct stat64 *buf)
677
678
679
680
681
{
    int ret;
    struct darshan_file_runtime* file;
    double tm1, tm2;

682
683
    MAP_OR_FAIL(__fxstat64);

684
685
686
687
688
689
690
691
692
693
694
695
696
697
    tm1 = darshan_wtime();
    ret = __real___fxstat64(vers, fd, buf);
    tm2 = darshan_wtime();
    if(ret < 0 || !S_ISREG(buf->st_mode))
        return(ret);

    /* skip logging if this was triggered internally */
    if(buf == &cp_stat_buf)
        return(ret);

    CP_LOCK();
    file = darshan_file_by_fd(fd);
    if(file)
    {
698
        CP_RECORD_STAT(file, buf, tm1, tm2);
699
700
701
702
703
704
705
    }
    CP_UNLOCK();

    return(ret);
}


706
int DARSHAN_DECL(__xstat)(int vers, const char *path, struct stat *buf)
707
708
709
710
{
    int ret;
    double tm1, tm2;

711
712
    MAP_OR_FAIL(__xstat);

713
714
715
716
717
718
719
    tm1 = darshan_wtime();
    ret = __real___xstat(vers, path, buf);
    tm2 = darshan_wtime();
    if(ret < 0 || !S_ISREG(buf->st_mode))
        return(ret);

    CP_LOCK();
720
    CP_LOOKUP_RECORD_STAT(path, buf, tm1, tm2);
721
722
723
724
725
    CP_UNLOCK();

    return(ret);
}

726
int DARSHAN_DECL(__lxstat)(int vers, const char *path, struct stat *buf)
727
728
729
730
{
    int ret;
    double tm1, tm2;

731
732
    MAP_OR_FAIL(__lxstat);

733
734
735
736
737
738
739
    tm1 = darshan_wtime();
    ret = __real___lxstat(vers, path, buf);
    tm2 = darshan_wtime();
    if(ret < 0 || !S_ISREG(buf->st_mode))
        return(ret);

    CP_LOCK();
740
    CP_LOOKUP_RECORD_STAT(path, buf, tm1, tm2);
741
742
743
744
745
    CP_UNLOCK();

    return(ret);
}

746
int DARSHAN_DECL(__fxstat)(int vers, int fd, struct stat *buf)
747
748
749
750
751
{
    int ret;
    struct darshan_file_runtime* file;
    double tm1, tm2;

752
753
    MAP_OR_FAIL(__fxstat);

754
755
756
757
758
759
760
    tm1 = darshan_wtime();
    ret = __real___fxstat(vers, fd, buf);
    tm2 = darshan_wtime();
    if(ret < 0 || !S_ISREG(buf->st_mode))
        return(ret);

    /* skip logging if this was triggered internally */
761
    if((void*)buf == (void*)&cp_stat_buf)
762
763
764
765
766
767
        return(ret);

    CP_LOCK();
    file = darshan_file_by_fd(fd);
    if(file)
    {
768
        CP_RECORD_STAT(file, buf, tm1, tm2);
769
770
771
772
773
774
    }
    CP_UNLOCK();

    return(ret);
}

775
ssize_t DARSHAN_DECL(pread64)(int fd, void *buf, size_t count, off64_t offset)
776
777
778
779
780
{
    ssize_t ret;
    int aligned_flag = 0;
    double tm1, tm2;

781
782
    MAP_OR_FAIL(pread64);

783
    if((unsigned long)buf % darshan_mem_alignment == 0)
784
785
786
787
788
789
        aligned_flag = 1;

    tm1 = darshan_wtime();
    ret = __real_pread64(fd, buf, count, offset);
    tm2 = darshan_wtime();
    CP_LOCK();
790
    CP_RECORD_READ(ret, fd, count, 1, offset, aligned_flag, 0, tm1, tm2);
791
792
793
794
    CP_UNLOCK();
    return(ret);
}

795
ssize_t DARSHAN_DECL(pread)(int fd, void *buf, size_t count, off_t offset)
796
797
798
799
800
{
    ssize_t ret;
    int aligned_flag = 0;
    double tm1, tm2;

801
802
    MAP_OR_FAIL(pread);

803
    if((unsigned long)buf % darshan_mem_alignment == 0)
804
805
806
807
808
809
        aligned_flag = 1;

    tm1 = darshan_wtime();
    ret = __real_pread(fd, buf, count, offset);
    tm2 = darshan_wtime();
    CP_LOCK();
810
    CP_RECORD_READ(ret, fd, count, 1, offset, aligned_flag, 0, tm1, tm2);
811
812
813
814
815
    CP_UNLOCK();
    return(ret);
}


816
ssize_t DARSHAN_DECL(pwrite)(int fd, const void *buf, size_t count, off_t offset)
817
818
819
820
821
{
    ssize_t ret;
    int aligned_flag = 0;
    double tm1, tm2;

822
823
    MAP_OR_FAIL(pwrite);

824
    if((unsigned long)buf % darshan_mem_alignment == 0)
825
826
827
828
829
830
        aligned_flag = 1;

    tm1 = darshan_wtime();
    ret = __real_pwrite(fd, buf, count, offset);
    tm2 = darshan_wtime();
    CP_LOCK();
Philip Carns's avatar
Philip Carns committed
831
    CP_RECORD_WRITE(ret, fd, count, 1, offset, aligned_flag, 0, tm1, tm2);
832
833
834
835
    CP_UNLOCK();
    return(ret);
}

836
ssize_t DARSHAN_DECL(pwrite64)(int fd, const void *buf, size_t count, off64_t offset)
837
838
839
840
841
{
    ssize_t ret;
    int aligned_flag = 0;
    double tm1, tm2;

842
843
    MAP_OR_FAIL(pwrite64);

844
    if((unsigned long)buf % darshan_mem_alignment == 0)
845
846
847
848
849
850
        aligned_flag = 1;

    tm1 = darshan_wtime();
    ret = __real_pwrite64(fd, buf, count, offset);
    tm2 = darshan_wtime();
    CP_LOCK();
Philip Carns's avatar
Philip Carns committed
851
    CP_RECORD_WRITE(ret, fd, count, 1, offset, aligned_flag, 0, tm1, tm2);
852
853
854
855
    CP_UNLOCK();
    return(ret);
}

856
ssize_t DARSHAN_DECL(readv)(int fd, const struct iovec *iov, int iovcnt)
857
858
859
860
861
862
{
    ssize_t ret;
    int aligned_flag = 1;
    int i;
    double tm1, tm2;

863
864
    MAP_OR_FAIL(readv);

865
866
    for(i=0; i<iovcnt; i++)
    {
867
        if(((unsigned long)iov[i].iov_base % darshan_mem_alignment) != 0)
868
869
870
871
872
873
874
            aligned_flag = 0;
    }

    tm1 = darshan_wtime();
    ret = __real_readv(fd, iov, iovcnt);
    tm2 = darshan_wtime();
    CP_LOCK();
875
    CP_RECORD_READ(ret, fd, count, 0, 0, aligned_flag, 0, tm1, tm2);
876
877
878
879
    CP_UNLOCK();
    return(ret);
}

880
ssize_t DARSHAN_DECL(writev)(int fd, const struct iovec *iov, int iovcnt)
881
882
883
884
885
886
{
    ssize_t ret;
    int aligned_flag = 1;
    int i;
    double tm1, tm2;

887
888
    MAP_OR_FAIL(writev);

889
890
    for(i=0; i<iovcnt; i++)
    {
891
        if(!((unsigned long)iov[i].iov_base % darshan_mem_alignment == 0))
892
893
894
895
896
897
898
            aligned_flag = 0;
    }

    tm1 = darshan_wtime();
    ret = __real_writev(fd, iov, iovcnt);
    tm2 = darshan_wtime();
    CP_LOCK();
Philip Carns's avatar
Philip Carns committed
899
    CP_RECORD_WRITE(ret, fd, count, 0, 0, aligned_flag, 0, tm1, tm2);
900
901
902
903
    CP_UNLOCK();
    return(ret);
}

904
size_t DARSHAN_DECL(fread)(void *ptr, size_t size, size_t nmemb, FILE *stream)
905
906
907
908
909
{
    size_t ret;
    int aligned_flag = 0;
    double tm1, tm2;

910
911
    MAP_OR_FAIL(fread);

912
    if((unsigned long)ptr % darshan_mem_alignment == 0)
913
914
915
916
917
918
919
        aligned_flag = 1;

    tm1 = darshan_wtime();
    ret = __real_fread(ptr, size, nmemb, stream);
    tm2 = darshan_wtime();
    CP_LOCK();
    if(ret > 0)
920
        CP_RECORD_READ(size*ret, fileno(stream), (size*nmemb), 0, 0, aligned_flag, 1, tm1, tm2);
921
    else
922
        CP_RECORD_READ(ret, fileno(stream), (size*nmemb), 0, 0, aligned_flag, 1, tm1, tm2);
923
924
925
926
    CP_UNLOCK();
    return(ret);
}

927
ssize_t DARSHAN_DECL(read)(int fd, void *buf, size_t count)
928
929
930
931
932
{
    ssize_t ret;
    int aligned_flag = 0;
    double tm1, tm2;

933
934
    MAP_OR_FAIL(read);

935
    if((unsigned long)buf % darshan_mem_alignment == 0)
936
937
938
939
940
941
        aligned_flag = 1;

    tm1 = darshan_wtime();
    ret = __real_read(fd, buf, count);
    tm2 = darshan_wtime();
    CP_LOCK();
942
    CP_RECORD_READ(ret, fd, count, 0, 0, aligned_flag, 0, tm1, tm2);
943
944
945
946
    CP_UNLOCK();
    return(ret);
}

947
ssize_t DARSHAN_DECL(write)(int fd, const void *buf, size_t count)
948
949
950
951
952
{
    ssize_t ret;
    int aligned_flag = 0;
    double tm1, tm2;

953
954
    MAP_OR_FAIL(write);

955
    if((unsigned long)buf % darshan_mem_alignment == 0)
956
957
958
959
960
961
        aligned_flag = 1;

    tm1 = darshan_wtime();
    ret = __real_write(fd, buf, count);
    tm2 = darshan_wtime();
    CP_LOCK();
Philip Carns's avatar
Philip Carns committed
962
    CP_RECORD_WRITE(ret, fd, count, 0, 0, aligned_flag, 0, tm1, tm2);
963
964
965
966
    CP_UNLOCK();
    return(ret);
}

967
size_t DARSHAN_DECL(fwrite)(const void *ptr, size_t size, size_t nmemb, FILE *stream)
968
969
970
971
972
{
    size_t ret;
    int aligned_flag = 0;
    double tm1, tm2;

973
974
    MAP_OR_FAIL(fwrite);

975
    if((unsigned long)ptr % darshan_mem_alignment == 0)
976
977
978
979
980
981
982
        aligned_flag = 1;

    tm1 = darshan_wtime();
    ret = __real_fwrite(ptr, size, nmemb, stream);
    tm2 = darshan_wtime();
    CP_LOCK();
    if(ret > 0)
Philip Carns's avatar
Philip Carns committed
983
        CP_RECORD_WRITE(size*ret, fileno(stream), (size*nmemb), 0, 0, aligned_flag, 1, tm1, tm2);
984
    else
Philip Carns's avatar
Philip Carns committed
985
        CP_RECORD_WRITE(ret, fileno(stream), 0, 0, 0, aligned_flag, 1, tm1, tm2);
986
987
988
989
    CP_UNLOCK();
    return(ret);
}

990
off64_t DARSHAN_DECL(lseek64)(int fd, off64_t offset, int whence)
991
{
Philip Carns's avatar
Philip Carns committed
992
    off64_t ret;
993
994
995
    struct darshan_file_runtime* file;
    double tm1, tm2;

996
997
    MAP_OR_FAIL(lseek64);

998
999
1000
1001
1002
1003
1004
1005
1006
1007
    tm1 = darshan_wtime();
    ret = __real_lseek64(fd, offset, whence);
    tm2 = darshan_wtime();
    if(ret >= 0)
    {
        CP_LOCK();
        file = darshan_file_by_fd(fd);
        if(file)
        {
            file->offset = ret;
1008
            CP_F_INC_NO_OVERLAP(file, tm1, tm2, file->last_posix_meta_end, CP_F_POSIX_META_TIME);
1009
1010
1011
1012
1013
1014
1015
            CP_INC(file, CP_POSIX_SEEKS, 1);
        }
        CP_UNLOCK();
    }
    return(ret);
}

1016
off_t DARSHAN_DECL(lseek)(int fd, off_t offset, int whence)
1017
1018
1019
1020
1021
{
    off_t ret;
    struct darshan_file_runtime* file;
    double tm1, tm2;

1022
1023
    MAP_OR_FAIL(lseek);

1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
    tm1 = darshan_wtime();
    ret = __real_lseek(fd, offset, whence);
    tm2 = darshan_wtime();
    if(ret >= 0)
    {
        CP_LOCK();
        file = darshan_file_by_fd(fd);
        if(file)
        {
            file->offset = ret;
1034
            CP_F_INC_NO_OVERLAP(file, tm1, tm2, file->last_posix_meta_end, CP_F_POSIX_META_TIME);
1035
1036
1037
1038
1039
1040
1041
            CP_INC(file, CP_POSIX_SEEKS, 1);
        }
        CP_UNLOCK();
    }
    return(ret);
}

1042
int DARSHAN_DECL(fseek)(FILE *stream, long offset, int whence)
1043
1044
1045
1046
1047
{
    int ret;
    struct darshan_file_runtime* file;
    double tm1, tm2;

1048
1049
    MAP_OR_FAIL(fseek);

1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
    tm1 = darshan_wtime();
    ret = __real_fseek(stream, offset, whence);
    tm2 = darshan_wtime();
    if(ret >= 0)
    {
        CP_LOCK();
        file = darshan_file_by_fd(fileno(stream));
        if(file)
        {
            file->offset = ret;
1060
            CP_F_INC_NO_OVERLAP(file, tm1, tm2, file->last_posix_meta_end, CP_F_POSIX_META_TIME);
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
            CP_INC(file, CP_POSIX_FSEEKS, 1);
        }
        CP_UNLOCK();
    }
    return(ret);
}

void darshan_finalize(struct darshan_job_runtime* job)
{
    if(!job)
    {
        return;
    }

    free(job);
}

void darshan_initialize(int argc, char** argv,  int nprocs, int rank)
{
    int i;
    char* disable;
    char* disable_timing;
1083
    char* envstr;
1084
1085
1086
    char* truncate_string = "<TRUNCATED>";
    int truncate_offset;
    int chars_left = 0;
1087
1088
    int ret;
    int tmpval;
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103

    disable = getenv("DARSHAN_DISABLE");
    if(disable)
    {
        /* turn off tracing */
        return;
    }

    disable_timing = getenv("DARSHAN_DISABLE_TIMING");

    if(darshan_global_job != NULL)
    {
        return;
    }

1104
1105
1106
    #if (__CP_MEM_ALIGNMENT < 1)
        #error Darshan must be configured with a positive value for --with-mem-align
    #endif
1107
1108
1109
    envstr = getenv("DARSHAN_MEMALIGN");
    if (envstr)
    {
1110
1111
1112
1113
1114
1115
        ret = sscanf(envstr, "%d", &tmpval);
        /* silently ignore if the env variable is set poorly */
        if(ret == 1 && tmpval > 0)
        {
            darshan_mem_alignment = tmpval;
        }
1116
1117
1118
1119
1120
1121
    }
    else
    {
        darshan_mem_alignment = __CP_MEM_ALIGNMENT;
    }

1122
1123
1124
1125
1126
1127
    /* avoid floating point errors on faulty input */
    if (darshan_mem_alignment < 1)
    {
        darshan_mem_alignment = 1;
    }

1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
    /* allocate structure to track darshan_global_job information */
    darshan_global_job = malloc(sizeof(*darshan_global_job));
    if(!darshan_global_job)
    {
        return;
    }
    memset(darshan_global_job, 0, sizeof(*darshan_global_job));

    if(disable_timing)
    {
        darshan_global_job->flags |= CP_FLAG_NOTIMING;
    }

    /* set up file records */
    for(i=0; i<CP_MAX_FILES; i++)
    {
        darshan_global_job->file_runtime_array[i].log_file = 
            &darshan_global_job->file_array[i];
    }

    strcpy(darshan_global_job->log_job.version_string, CP_VERSION);
1149
    darshan_global_job->log_job.magic_nr = CP_MAGIC_NR;
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
    darshan_global_job->log_job.uid = getuid();
    darshan_global_job->log_job.start_time = time(NULL);
    darshan_global_job->log_job.nprocs = nprocs;
    my_rank = rank;

    /* record exe and arguments */
    for(i=0; i<argc; i++)
    {
        chars_left = CP_EXE_LEN-strlen(darshan_global_job->exe);
        strncat(darshan_global_job->exe, argv[i], chars_left);
        if(i < (argc-1))
        {
            chars_left = CP_EXE_LEN-strlen(darshan_global_job->exe);
            strncat(darshan_global_job->exe, " ", chars_left);
        }
    }

    /* if we don't see any arguments, then use glibc symbol to get
     * program name at least (this happens in fortran)
     */
    if(argc == 0)
    {
        chars_left = CP_EXE_LEN-strlen(darshan_global_job->exe);
        strncat(darshan_global_job->exe, __progname_full, chars_left);
        chars_left = CP_EXE_LEN-strlen(darshan_global_job->exe);
        strncat(darshan_global_job->exe, " <unknown args>", chars_left);
    }

    if(chars_left == 0)
    {
        /* we ran out of room; mark that string was truncated */
        truncate_offset = CP_EXE_LEN - strlen(truncate_string);
        sprintf(&darshan_global_job->exe[truncate_offset], "%s", 
            truncate_string);
    }
1185
1186
1187
1188
1189
1190
1191
1192

    /* collect information about command line and 
     * mounted file systems 
     */
    darshan_global_job->trailing_data = 
        darshan_get_exe_and_mounts(darshan_global_job);

    return;
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
1218
1219
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
}

/* darshan_condense()
 *
 * collapses all file statistics into a single unified set of counters; used
 * when we have opened too many files to track independently
 */
void darshan_condense(void)
{
    struct darshan_file_runtime* base_file;
    struct darshan_file_runtime* iter_file;
    int i;
    int j;

    if(!darshan_global_job)
        return;

    base_file = &darshan_global_job->file_runtime_array[0];

    /* iterate through files */
    for(j=1; j<darshan_global_job->file_count; j++)
    {
        iter_file = &darshan_global_job->file_runtime_array[j];

        /* iterate through records */
        for(i=0; i<CP_NUM_INDICES; i++)
        {
            switch(i)
            {
                /* NOTE: several fields cease to make sense if the records
                 * have been condensed.  Just let them get summed anyway.
                 */
                /* TODO: double check this */

                /* keep up with global maxes in case they are helpful */
                case CP_MAX_BYTE_READ:
                case CP_MAX_BYTE_WRITTEN:
                    CP_MAX(base_file, i, CP_VALUE(iter_file, i));
                    break;

1233
1234
1235
1236
1237
1238
1239
                /* do nothing with these; they are handled in the floating
                 * point loop 
                 */
                case CP_MAX_WRITE_TIME_SIZE:
                case CP_MAX_READ_TIME_SIZE:
                    break;

1240
1241
                /* pick one */
                case CP_DEVICE:
1242
                case CP_SIZE_AT_OPEN:
1243
1244
1245
                    CP_SET(base_file, i, CP_VALUE(iter_file, i));
                    break;

1246
1247
1248
1249
1250
1251
1252
1253
                /* most records can simply be added */
                default:
                    CP_INC(base_file, i, CP_VALUE(iter_file, i));
                    break;
            }
        }
        for(i=0; i<CP_F_NUM_INDICES; i++)
        {
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
            switch(i)
            {
                case CP_F_MAX_WRITE_TIME:
                    if(CP_F_VALUE(iter_file, i) > CP_F_VALUE(base_file, i))
                    {
                        CP_F_SET(base_file, i, CP_F_VALUE(iter_file, i));
                        CP_SET(base_file, CP_MAX_WRITE_TIME_SIZE, 
                            CP_VALUE(iter_file, CP_MAX_WRITE_TIME_SIZE));
                    }
                    break;
                case CP_F_MAX_READ_TIME:
                    if(CP_F_VALUE(iter_file, i) > CP_F_VALUE(base_file, i))
                    {
                        CP_F_SET(base_file, i, CP_F_VALUE(iter_file, i));
                        CP_SET(base_file, CP_MAX_READ_TIME_SIZE, 
                            CP_VALUE(iter_file, CP_MAX_READ_TIME_SIZE));
                    }
                    break;
                default:
                    CP_F_SET(base_file, i, CP_F_VALUE(iter_file, i) + CP_F_VALUE(base_file, i));
                    break;
            }
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
        }
    }
    
    base_file->log_file->hash = 0;
    
    darshan_global_job->flags |= CP_FLAG_CONDENSED;
    darshan_global_job->file_count = 1;

    /* clear hash tables for safety */
    memset(darshan_global_job->name_table, 0, CP_HASH_SIZE*sizeof(struct darshan_file_runtime*));
1286
    memset(darshan_global_job->handle_table, 0, CP_HASH_SIZE*sizeof(*darshan_global_job->handle_table));
1287
1288
1289
1290
    
    return;
}

1291
static struct darshan_file_runtime* darshan_file_by_name_setfd(const char* name, int fd)
1292
1293
1294
{
    struct darshan_file_runtime* tmp_file;

1295
    tmp_file = darshan_file_by_name_sethandle(name, &fd, sizeof(fd), DARSHAN_FD);
1296
1297
1298
    return(tmp_file);
}

1299
static void darshan_file_close_fd(int fd)
1300
{
1301
1302
1303
    darshan_file_closehandle(&fd, sizeof(fd), DARSHAN_FD);
    return;
}
1304

1305
1306
1307
static struct darshan_file_runtime* darshan_file_by_fd(int fd)
{
    struct darshan_file_runtime* tmp_file;
1308

1309
1310
1311
    tmp_file = darshan_file_by_handle(&fd, sizeof(fd), DARSHAN_FD);
    
    return(tmp_file);
1312
1313
1314
1315
1316
1317
1318
1319
1320
1321
1322
1323
1324
1325
1326
1327
1328
1329
1330
1331
1332
1333
1334
1335
1336
1337
1338
1339
1340
1341
1342
1343
1344
1345
1346
1347
1348
1349
1350
1351
1352
1353
1354
1355
1356
1357
1358
1359
1360
1361
1362
1363
1364
1365
1366
1367
1368
1369
1370
1371
1372
1373
1374
1375
1376
1377
1378
1379
1380
1381
1382
1383
1384
1385
1386
1387
1388
1389
1390
1391
1392
1393
1394
1395
1396
1397
1398
1399
1400
1401
1402
1403
1404
1405
1406
1407
1408
1409
1410
1411
1412
1413
1414
1415
1416
1417
1418
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
}

static int access_comparison(const void* a_p, const void* b_p)
{
    const struct cp_access_counter* a = a_p;
    const struct cp_access_counter* b = b_p;

    if(a->size < b->size)
        return(-1);
    if(a->size > b->size)
        return(1);
    return(0);
}

/* cp_access_counter()
 *
 * records the occurance of a particular access size for a file,
 * current implementation uses glibc red black tree
 */
static void cp_access_counter(struct darshan_file_runtime* file, ssize_t size, enum cp_counter_type type)
{
    struct cp_access_counter* counter;
    struct cp_access_counter* found;
    void* tmp;
    void** root;
    int* count;
    struct cp_access_counter tmp_counter;

    /* don't count sizes or strides of 0 */
    if(size == 0)
        return;
    
    switch(type)
    {
        case CP_COUNTER_ACCESS:
            root = &file->access_root;
            count = &file->access_count;
            break;
        case CP_COUNTER_STRIDE:
            root = &file->stride_root;
            count = &file->stride_count;
            break;
        default:
            return;
    }

    /* check to see if this size is already recorded */
    tmp_counter.size = size;
    tmp_counter.freq = 1;
    tmp = tfind(&tmp_counter, root, access_comparison);
    if(tmp)
    {
        found = *(struct cp_access_counter**)tmp;
        found->freq++;
        return;
    }

    /* we can add a new one as long as we haven't hit the limit */
    if(*count < CP_MAX_ACCESS_COUNT_RUNTIME)
    {
        counter = malloc(sizeof(*counter));
        if(!counter)
        {
            return;
        }

        counter->size = size;
        counter->freq = 1;

        tmp = tsearch(counter, root, access_comparison);
        found = *(struct cp_access_counter**)tmp;
        /* if we get a new answer out here we are in trouble; this was
         * already checked with the tfind()
         */
        assert(found == counter);

        (*count)++;
    }

    return;
}

void darshan_shutdown_bench(int argc, char** argv, int rank, int nprocs)
{
    int* fd_array;
    int64_t* size_array;
    int i;
    int nfiles;
    char path[256];
    int iters;
    
    /* combinations to build:
     * - 1 unique file per proc
     * - 1 shared file per proc
     * - 1024 unique file per proc
     * - 1024 shared per proc
     */

    srand(rank);
    fd_array = malloc(sizeof(int)*CP_MAX_FILES);
    size_array = malloc(sizeof(int64_t)*CP_MAX_ACCESS_COUNT_RUNTIME);

    assert(fd_array&&size_array);

    for(i=0; i<CP_MAX_FILES; i++)
        fd_array[i] = i;
    for(i=0; i<CP_MAX_ACCESS_COUNT_RUNTIME; i++)
        size_array[i] = rand();

    /* clear out existing stuff */
    darshan_walk_file_accesses(darshan_global_job);
    darshan_finalize(darshan_global_job);
    darshan_global_job = NULL;

    /***********************************************************/
    /* reset darshan to start clean */
    darshan_initialize(argc, argv, nprocs, rank);

    /* populate one unique file per proc */
    nfiles = 1;
    iters = 1;
    for(i=0; i<nfiles; i++)
    {
        sprintf(path, "%d-%d", i, rank);
        CP_RECORD_OPEN(fd_array[i], path, 777, 0, 0, 0);
    }

    for(i=0; i<iters; i++)
    {
Philip Carns's avatar
Philip Carns committed
1441
        CP_RECORD_WRITE(size_array[(i/nfiles)%CP_MAX_ACCESS_COUNT_RUNTIME], fd_array[i%nfiles], size_array[(i/nfiles)%CP_MAX_ACCESS_COUNT_RUNTIME], 0, 0, 1, 0, 1, 2);
1442
1443
1444
1445
    }

    if(rank == 0)
        printf("# 1 unique file per proc\n");
1446
    DARSHAN_MPI_CALL(PMPI_Barrier)(MPI_COMM_WORLD);
1447
1448
1449
1450
1451
    darshan_shutdown(1);
    darshan_global_job = NULL;

    /***********************************************************/
    /* reset darshan to start clean */
1452
    sleep(1);
1453
1454
1455
1456
1457
1458
1459
1460
1461
1462
1463
1464
1465
    darshan_initialize(argc, argv, nprocs, rank);

    /* populate one shared file per proc */
    nfiles = 1;
    iters = 1;
    for(i=0; i<nfiles; i++)
    {
        sprintf(path, "%d", i);
        CP_RECORD_OPEN(fd_array[i], path, 777, 0, 0, 0);
    }

    for(i=0; i<iters; i++)
    {
Philip Carns's avatar
Philip Carns committed
1466
        CP_RECORD_WRITE(size_array[(i/nfiles)%CP_MAX_ACCESS_COUNT_RUNTIME], fd_array[i%nfiles], size_array[(i/nfiles)%CP_MAX_ACCESS_COUNT_RUNTIME], 0, 0, 1, 0, 1, 2);
1467
1468
1469
1470
    }

    if(rank == 0)
        printf("# 1 shared file across procs\n");
1471
    DARSHAN_MPI_CALL(PMPI_Barrier)(MPI_COMM_WORLD);
1472
1473
1474
1475
1476
    darshan_shutdown(1);
    darshan_global_job = NULL;

    /***********************************************************/
    /* reset darshan to start clean */
1477
    sleep(1);
1478
1479
1480
1481
1482
1483
1484
1485
1486
1487
1488
1489
1490
    darshan_initialize(argc, argv, nprocs, rank);

    /* populate 1024 unique file per proc */
    nfiles = 1024;
    iters = 1024;
    for(i=0; i<nfiles; i++)
    {
        sprintf(path, "%d-%d", i, rank);
        CP_RECORD_OPEN(fd_array[i], path, 777, 0, 0, 0);
    }

    for(i=0; i<iters; i++)
    {
Philip Carns's avatar
Philip Carns committed
1491
        CP_RECORD_WRITE(size_array[(i/nfiles)%CP_MAX_ACCESS_COUNT_RUNTIME], fd_array[i%nfiles], size_array[(i/nfiles)%CP_MAX_ACCESS_COUNT_RUNTIME], 0, 0, 1, 0, 1, 2);
1492
1493
1494
1495
    }

    if(rank == 0)
        printf("# 1024 unique files per proc\n");
1496
    DARSHAN_MPI_CALL(PMPI_Barrier)(MPI_COMM_WORLD);
1497
1498
1499
1500
1501
    darshan_shutdown(1);
    darshan_global_job = NULL;

    /***********************************************************/
    /* reset darshan to start clean */
1502
    sleep(1);
1503
1504
1505
1506
1507
1508
1509
1510
1511
1512
1513
1514
1515
    darshan_initialize(argc, argv, nprocs, rank);

    /* populate 1024 shared file per proc */
    nfiles = 1024;
    iters = 1024;
    for(i=0; i<nfiles; i++)
    {
        sprintf(path, "%d", i);
        CP_RECORD_OPEN(fd_array[i], path, 777, 0, 0, 0);
    }

    for(i=0; i<iters; i++)
    {
Philip Carns's avatar
Philip Carns committed
1516
        CP_RECORD_WRITE(size_array[(i/nfiles)%CP_MAX_ACCESS_COUNT_RUNTIME], fd_array[i%nfiles], size_array[(i/nfiles)%CP_MAX_ACCESS_COUNT_RUNTIME], 0, 0, 1, 0, 1, 2);
1517
1518
1519
1520
    }

    if(rank == 0)
        printf("# 1024 shared files across procs\n");
1521
    DARSHAN_MPI_CALL(PMPI_Barrier)(MPI_COMM_WORLD);
1522
1523
1524
1525
1526
1527
1528
1529
1530
1531
1532
1533
1534
1535
1536
1537
1538
1539
1540
1541
1542
1543
1544
1545
1546
1547
1548
1549
1550
1551
1552
1553
1554
1555
1556
1557
1558
1559
1560
1561
1562
1563
1564
1565
1566
1567
1568
1569
1570
1571
1572
1573
1574
1575
    darshan_shutdown(1);