Skip to content

Commit

Permalink
Merge branch 'fix-stat-overflow' of https://github.com/stilor/fio
Browse files Browse the repository at this point in the history
* 'fix-stat-overflow' of https://github.com/stilor/fio:
  Handle 32-bit overflows in disk utilization stats
  Change memcpy() calls to assignments
  • Loading branch information
vincentkfu committed Oct 6, 2023
2 parents 6f9cdcf + 5f7bd35 commit 16b9f29
Showing 1 changed file with 24 additions and 6 deletions.
30 changes: 24 additions & 6 deletions diskutil.c
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,23 @@ static int get_io_ticks(struct disk_util *du, struct disk_util_stat *dus)
return ret != 10;
}

static uint64_t safe_32bit_diff(uint64_t nval, uint64_t oval)
{
/* Linux kernel prints some of the stat fields as 32-bit integers. It is
* possible that the value overflows, but since fio uses unsigned 64-bit
* arithmetic in update_io_tick_disk(), it instead results in a huge
* bogus value being added to the respective accumulating field. Just
* in case Linux starts reporting these metrics as 64-bit values in the
* future, check that overflow actually happens around the 32-bit
* unsigned boundary; assume overflow only happens once between
* successive polls.
*/
if (oval <= nval || oval >= (1ull << 32))
return nval - oval;
else
return (1ull << 32) + nval - oval;
}

static void update_io_tick_disk(struct disk_util *du)
{
struct disk_util_stat __dus, *dus, *ldus;
Expand All @@ -96,15 +113,16 @@ static void update_io_tick_disk(struct disk_util *du)
dus->s.ios[1] += (__dus.s.ios[1] - ldus->s.ios[1]);
dus->s.merges[0] += (__dus.s.merges[0] - ldus->s.merges[0]);
dus->s.merges[1] += (__dus.s.merges[1] - ldus->s.merges[1]);
dus->s.ticks[0] += (__dus.s.ticks[0] - ldus->s.ticks[0]);
dus->s.ticks[1] += (__dus.s.ticks[1] - ldus->s.ticks[1]);
dus->s.io_ticks += (__dus.s.io_ticks - ldus->s.io_ticks);
dus->s.time_in_queue += (__dus.s.time_in_queue - ldus->s.time_in_queue);
dus->s.ticks[0] += safe_32bit_diff(__dus.s.ticks[0], ldus->s.ticks[0]);
dus->s.ticks[1] += safe_32bit_diff(__dus.s.ticks[1], ldus->s.ticks[1]);
dus->s.io_ticks += safe_32bit_diff(__dus.s.io_ticks, ldus->s.io_ticks);
dus->s.time_in_queue +=
safe_32bit_diff(__dus.s.time_in_queue, ldus->s.time_in_queue);

fio_gettime(&t, NULL);
dus->s.msec += mtime_since(&du->time, &t);
memcpy(&du->time, &t, sizeof(t));
memcpy(&ldus->s, &__dus.s, sizeof(__dus.s));
du->time = t;
ldus->s = __dus.s;
}

int update_io_ticks(void)
Expand Down

0 comments on commit 16b9f29

Please sign in to comment.