/* See LICENSE file for copyright and license details. */
#include <sys/mount.h>
#include <pthread.h>
#include <libsimple.h>
#define NANOSECONDS(X) X##L
#define MICROSECONDS(X) NANOSECONDS(X##000)
#define MILLISECONDS(X) MICROSECONDS(X##000)
#define CENTISECONDS(X) MILLISECONDS(X##0)
#define DECISECONDS(X) CENTISECONDS(X##0)
#define WHOLE_SECOND DECISECONDS(10)
#define WRAVG_STEPS 5
enum direction {
FORWARDS = 0,
BACKWARDS = 1
};
struct span {
off_t start;
off_t end;
off_t bad;
size_t blocksize;
};
struct status {
struct timespec now;
off_t shredded;
off_t bad_bytes;
uintmax_t bad_writes;
uintmax_t bad_sections;
uintmax_t pass_nr;
struct timespec last_success;
enum direction direction;
int write_average_i;
struct timespec write_average_begin_times[WRAVG_STEPS];
off_t write_average_amounts[WRAVG_STEPS];
size_t nspans;
size_t span_off;
size_t spans_size;
struct span *spans;
};
#define STATUS_INIT {.pass_nr = 1, .last_success = {-1, 0}, .direction = FORWARDS}
#define STATUS_TRANSFER_SIZE (offsetof(struct status, nspans) + sizeof(size_t))
/* io.c */
char *readtextfileat(int dfd, const char *path);
void writeall(int fd, const void *data, size_t n, const char *fname);
void filecpy(int fd, off_t src, off_t len, const char *fname);
off_t filesize(int fd, const char *fname);
#if defined(__linux__)
char *get_device_name(unsigned int devmajor, unsigned int devminor);
#endif
/* fmt.c */
const char *humansize1000(off_t s, char *buf);
const char *humansize1024(off_t s, char *buf);
#if defined(__GNUC__)
__attribute__((__pure__))
#endif
off_t unhumansize(const char *s, char flag);
const char *durationstr(const struct timespec *dur, char *buf, int second_decimals);
const char *humanbytespersecond(double bytes_per_second, char *buf);
const char *exact_and_human_size(off_t bytes, char *buf, int with_unit);
/* text.c */
int have_micro_symbol(void);
/* avg.c */
void wravg_init(const struct timespec *start_time, struct status *s);
void wravg_update_write(ssize_t amount, struct status *s);
void wravg_update_time(struct status *s);
const char *wravg_get(struct status *s);
/* rnd.c */
void init_random(int fd, const char *fname, off_t blksize);
void destroy_random(void);
#if defined(__GNUC__)
__attribute__((__pure__))
#endif
size_t max_blksize(void);
const char *get_random(size_t needed);
void used_random(ssize_t amount);
/* sig.c */
extern _Atomic volatile sig_atomic_t exiting;
void setup_sighandler(void);
void block_sigs(void);
void unblock_sigs(void);
/* ask.c */
int confirm(const char *device, off_t off, off_t len, off_t end);
/* map.c */
void add_span(struct status *status, off_t off, off_t amount, size_t blocksize, int try_join);
void dump_map(int fd, const struct status *status, const char *fname);
off_t measure_map_dump(const struct status *status);
off_t load_map(int fd, struct status *status, enum direction direction, const char *fname);
void update_map(int fd, const struct status *status, const char *fname);