57 lines
1.9 KiB
C
57 lines
1.9 KiB
C
/*
|
|
+ * Copyright (C) 2016 Freescale Semiconductor, Inc.
|
|
+ * Copyright 2017 NXP
|
|
+ *
|
|
+ * SPDX-License-Identifier: GPL-2.0+
|
|
+ */
|
|
#include <common.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "debug.h"
|
|
#include "utils.h"
|
|
|
|
/* get margin_pos struct from offset [to the partition start/end] and num_bytes to read/write */
|
|
int get_margin_pos(uint64_t part_start, uint64_t part_end, unsigned long blksz,
|
|
margin_pos_t *margin, int64_t offset, size_t num_bytes, bool allow_partial) {
|
|
long off;
|
|
assert(margin != NULL);
|
|
if (blksz == 0 || part_start > part_end)
|
|
return -1;
|
|
|
|
if (offset < 0) {
|
|
margin->blk_start = (offset + 1) / (uint64_t)blksz + part_end;
|
|
margin->start = (off = offset % (uint64_t)blksz) == 0 ? 0 : blksz + off; // offset == -1 means the last byte?, or start need -1
|
|
if (offset + num_bytes - 1 >= 0) {
|
|
if (!allow_partial)
|
|
return -1;
|
|
margin->blk_end = part_end;
|
|
margin->end = blksz - 1;
|
|
} else {
|
|
margin->blk_end = (num_bytes + offset) / (uint64_t)blksz + part_end; // which blk the last byte is in
|
|
margin->end = (off = (num_bytes + offset - 1) % (uint64_t)blksz) == 0 ?
|
|
0 : blksz + off; // last byte
|
|
}
|
|
} else {
|
|
margin->blk_start = offset / (uint64_t)blksz + part_start;
|
|
margin->start = offset % (uint64_t)blksz;
|
|
margin->blk_end = (offset + num_bytes - 1) / (uint64_t)blksz + part_start ;
|
|
margin->end = (offset + num_bytes - 1) % (uint64_t)blksz;
|
|
if (margin->blk_end > part_end) {
|
|
if (!allow_partial)
|
|
return -1;
|
|
margin->blk_end = part_end;
|
|
margin->end = blksz - 1;
|
|
}
|
|
}
|
|
VDEBUG("bs=%ld, be=%ld, s=%ld, e=%ld\n",
|
|
margin->blk_start, margin->blk_end, margin->start, margin->end);
|
|
|
|
if (margin->blk_start > part_end || margin->blk_start < part_start)
|
|
return -1;
|
|
long multi = margin->blk_end - margin->blk_start - 1 +
|
|
(margin->start == 0) + (margin->end == blksz -1);
|
|
margin->multi = multi > 0 ? multi : 0;
|
|
VDEBUG("bm=%ld\n", margin->multi);
|
|
return 0;
|
|
}
|