| /* |
| + * 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; |
| } |