123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387 |
- #include <stdio.h>
- #include <string.h>
- #include <strings.h>
- #include <stdlib.h>
- #include <stdint.h>
- #include <stdarg.h>
- #include <fcntl.h>
- #include <unistd.h>
- #include <sys/ioctl.h>
- #include <sys/mman.h>
- #include <arpa/inet.h>
- #include <errno.h>
- #include <assert.h>
- #include <alloca.h>
- #include "pci.h"
- #include "bank.h"
- #include "tools.h"
- #include "error.h"
- #include "property.h"
- int pcilib_add_registers(pcilib_t *ctx, pcilib_model_modification_flags_t flags, size_t n, const pcilib_register_description_t *registers, pcilib_register_t *ids) {
- // DS: Overrride existing registers
- // Registers identified by addr + offset + size + type or name
- int err;
- size_t size;
- pcilib_register_t i;
- pcilib_register_description_t *regs;
- pcilib_register_context_t *reg_ctx;
- pcilib_register_bank_t bank = PCILIB_REGISTER_BANK_INVALID;
- pcilib_register_bank_addr_t bank_addr = (pcilib_register_bank_addr_t)-1;
- pcilib_register_bank_t *banks;
- if (!n) {
- for (n = 0; registers[n].bits; n++);
- }
- if ((ctx->num_reg + n + 1) > ctx->alloc_reg) {
- for (size = ctx->alloc_reg; size < 2 * (n + ctx->num_reg + 1); size<<=1);
- regs = (pcilib_register_description_t*)realloc(ctx->registers, size * sizeof(pcilib_register_description_t));
- if (!regs) return PCILIB_ERROR_MEMORY;
- ctx->registers = regs;
- ctx->model_info.registers = regs;
- reg_ctx = (pcilib_register_context_t*)realloc(ctx->registers, size * sizeof(pcilib_register_context_t));
- if (!reg_ctx) return PCILIB_ERROR_MEMORY;
-
- memset(reg_ctx + ctx->alloc_reg, 0, (size - ctx->alloc_reg) * sizeof(pcilib_register_context_t));
- if (ctx->register_ctx != reg_ctx) {
- // We need to recreate cache if context is moved...
- HASH_CLEAR(hh, ctx->reg_hash);
- for (i = 0; i < ctx->num_reg; i++) {
- pcilib_register_context_t *cur = &ctx->register_ctx[ctx->num_reg + i];
- HASH_ADD_KEYPTR(hh, ctx->reg_hash, cur->name, strlen(cur->name), cur);
- }
- }
- ctx->register_ctx = reg_ctx;
- ctx->alloc_reg = size;
- }
- banks = (pcilib_register_bank_t*)alloca(n * sizeof(pcilib_register_bank_t));
- if (!banks) return PCILIB_ERROR_MEMORY;
- for (i = 0; i < n; i++) {
- if (registers[i].bank != bank_addr) {
- bank_addr = registers[i].bank;
- bank = pcilib_find_register_bank_by_addr(ctx, bank_addr);
- if (bank == PCILIB_REGISTER_BANK_INVALID) {
- pcilib_error("Invalid bank address (0x%lx) is specified for register %s", bank_addr, registers[i].name);
- return PCILIB_ERROR_INVALID_BANK;
- }
- }
- /*
- // No hash so far, will iterate.
- pcilib_register_t reg = pcilib_find_register(ctx, ctx->banks[bank].name, registers[i].name);
- if (reg != PCILIB_REGISTER_INVALID) {
- pcilib_error("Register %s is already defined in the model", registers[i].name);
- return PCILIB_ERROR_EXIST;
- }
- */
- banks[i] = bank;
- }
- err = pcilib_add_register_properties(ctx, n, banks, registers);
- if (err) return err;
- for (i = 0; i < n; i++) {
- pcilib_register_context_t *cur = &ctx->register_ctx[ctx->num_reg + i];
- cur->reg = ctx->num_reg + i;
- cur->name = registers[i].name;
- cur->bank = banks[i];
- HASH_ADD_KEYPTR(hh, ctx->reg_hash, cur->name, strlen(cur->name), cur);
- }
- memcpy(ctx->registers + ctx->num_reg, registers, n * sizeof(pcilib_register_description_t));
- memset(ctx->registers + ctx->num_reg + n, 0, sizeof(pcilib_register_description_t));
- if (ids) {
- size_t i;
-
- for (i = 0; i < n; i++)
- ids[i] = ctx->num_reg + i;
- }
- ctx->num_reg += n;
- return 0;
- }
- void pcilib_clean_registers(pcilib_t *ctx, pcilib_register_t start) {
- pcilib_register_t reg;
- pcilib_register_context_t *reg_ctx, *tmp;
- if (start) {
- HASH_ITER(hh, ctx->reg_hash, reg_ctx, tmp) {
- if (reg_ctx->reg >= start) {
- HASH_DEL(ctx->reg_hash, reg_ctx);
- }
- }
- } else {
- HASH_CLEAR(hh, ctx->reg_hash);
- }
- for (reg = start; reg < ctx->num_reg; reg++) {
- if (ctx->register_ctx[reg].views)
- free(ctx->register_ctx[reg].views);
- }
- if (ctx->registers)
- memset(&ctx->registers[start], 0, sizeof(pcilib_register_description_t));
- if (ctx->register_ctx)
- memset(&ctx->register_ctx[start], 0, (ctx->alloc_reg - start) * sizeof(pcilib_register_context_t));
- ctx->num_reg = start;
- }
- static int pcilib_read_register_space_internal(pcilib_t *ctx, pcilib_register_bank_t bank, pcilib_register_addr_t addr, size_t n, pcilib_register_size_t offset, pcilib_register_size_t bits, pcilib_register_value_t *buf) {
- int err;
- size_t i;
- pcilib_register_bank_context_t *bctx = ctx->bank_ctx[bank];
- const pcilib_register_protocol_api_description_t *bapi = bctx->api;
- const pcilib_register_bank_description_t *b = bctx->bank;
- int access = b->access / 8;
- assert(bits < 8 * sizeof(pcilib_register_value_t));
- if (!bapi->read) {
- pcilib_error("Used register protocol does not define a way to read register value");
- return PCILIB_ERROR_NOTSUPPORTED;
- }
- if (((addr + n) > b->size)||(((addr + n) == b->size)&&(bits))) {
- if ((b->format)&&(strchr(b->format, 'x')))
- pcilib_error("Accessing register (%u regs at addr 0x%x) out of register space (%u registers total)", bits?(n+1):n, addr, b->size);
- else
- pcilib_error("Accessing register (%u regs at addr %u) out of register space (%u registers total)", bits?(n+1):n, addr, b->size);
- return PCILIB_ERROR_OUTOFRANGE;
- }
- //err = pcilib_init_register_banks(ctx);
- //if (err) return err;
-
- //n += bits / b->access;
- //bits %= b->access;
-
- for (i = 0; i < n; i++) {
- err = bapi->read(ctx, bctx, addr + i * access, buf + i);
- if (err) break;
- }
-
- if ((bits > 0)&&(!err)) {
- pcilib_register_value_t val = 0;
- err = bapi->read(ctx, bctx, addr + n * access, &val);
- val = (val >> offset)&BIT_MASK(bits);
- memcpy(buf + n, &val, sizeof(pcilib_register_value_t));
- }
-
- return err;
- }
- int pcilib_read_register_space(pcilib_t *ctx, const char *bank, pcilib_register_addr_t addr, size_t n, pcilib_register_value_t *buf) {
- pcilib_register_bank_t bank_id = pcilib_find_register_bank(ctx, bank);
- if (bank_id == PCILIB_REGISTER_BANK_INVALID) {
- if (bank) pcilib_error("Invalid register bank is specified (%s)", bank);
- else pcilib_error("Register bank should be specified");
- return PCILIB_ERROR_INVALID_BANK;
- }
-
- return pcilib_read_register_space_internal(ctx, bank_id, addr, n, 0, 0, buf);
- }
- int pcilib_read_register_by_id(pcilib_t *ctx, pcilib_register_t reg, pcilib_register_value_t *value) {
- int err;
- size_t i, n;
- pcilib_register_size_t bits;
- pcilib_register_value_t res;
- pcilib_register_bank_t bank;
- const pcilib_register_description_t *r;
- const pcilib_register_bank_description_t *b;
- const pcilib_model_description_t *model_info = pcilib_get_model_description(ctx);
- r = model_info->registers + reg;
-
- bank = pcilib_find_register_bank_by_addr(ctx, r->bank);
- if (bank == PCILIB_REGISTER_BANK_INVALID) return PCILIB_ERROR_INVALID_BANK;
-
- b = model_info->banks + bank;
-
- n = r->bits / b->access;
- bits = r->bits % b->access;
- pcilib_register_value_t buf[n + 1];
- err = pcilib_read_register_space_internal(ctx, bank, r->addr, n, r->offset, bits, buf);
- if ((b->endianess == PCILIB_BIG_ENDIAN)||((b->endianess == PCILIB_HOST_ENDIAN)&&(ntohs(1) == 1))) {
- pcilib_error("Big-endian byte order support is not implemented");
- return PCILIB_ERROR_NOTSUPPORTED;
- } else {
- res = 0;
- if (bits) ++n;
- for (i = 0; i < n; i++) {
- res |= buf[i] << (i * b->access);
- }
- }
-
- *value = res;
-
- return err;
- }
- int pcilib_read_register(pcilib_t *ctx, const char *bank, const char *regname, pcilib_register_value_t *value) {
- int reg;
- reg = pcilib_find_register(ctx, bank, regname);
- if (reg == PCILIB_REGISTER_INVALID) {
- pcilib_error("Register (%s) is not found", regname);
- return PCILIB_ERROR_NOTFOUND;
- }
- return pcilib_read_register_by_id(ctx, reg, value);
- }
- static int pcilib_write_register_space_internal(pcilib_t *ctx, pcilib_register_bank_t bank, pcilib_register_addr_t addr, size_t n, pcilib_register_size_t offset, pcilib_register_size_t bits, pcilib_register_value_t rwmask, pcilib_register_value_t *buf) {
- int err;
- size_t i;
- pcilib_register_bank_context_t *bctx = ctx->bank_ctx[bank];
- const pcilib_register_protocol_api_description_t *bapi = bctx->api;
- const pcilib_register_bank_description_t *b = bctx->bank;
- int access = b->access / 8;
- assert(bits < 8 * sizeof(pcilib_register_value_t));
- if (!bapi->write) {
- pcilib_error("Used register protocol does not define a way to write value into the register");
- return PCILIB_ERROR_NOTSUPPORTED;
- }
- if (((addr + n) > b->size)||(((addr + n) == b->size)&&(bits))) {
- if ((b->format)&&(strchr(b->format, 'x')))
- pcilib_error("Accessing register (%u regs at addr 0x%x) out of register space (%u registers total)", bits?(n+1):n, addr, b->size);
- else
- pcilib_error("Accessing register (%u regs at addr %u) out of register space (%u registers total)", bits?(n+1):n, addr, b->size);
- return PCILIB_ERROR_OUTOFRANGE;
- }
- //err = pcilib_init_register_banks(ctx);
- //if (err) return err;
-
- //n += bits / b->access;
- //bits %= b->access;
-
- for (i = 0; i < n; i++) {
- err = bapi->write(ctx, bctx, addr + i * access, buf[i]);
- if (err) break;
- }
-
- if ((bits > 0)&&(!err)) {
- pcilib_register_value_t val = (buf[n]&BIT_MASK(bits))<<offset;
- pcilib_register_value_t mask = BIT_MASK(bits)<<offset;
- if (~mask&rwmask) {
- pcilib_register_value_t rval;
- if (!bapi->read) {
- pcilib_error("Used register protocol does not define a way to read register. Therefore, it is only possible to write a full bank word, not partial as required by the accessed register");
- return PCILIB_ERROR_NOTSUPPORTED;
- }
- err = bapi->read(ctx, bctx, addr + n * access, &rval);
- if (err) return err;
- val |= (rval & rwmask & ~mask);
- }
-
- err = bapi->write(ctx, bctx, addr + n * access, val);
- }
-
- return err;
- }
- int pcilib_write_register_space(pcilib_t *ctx, const char *bank, pcilib_register_addr_t addr, size_t n, pcilib_register_value_t *buf) {
- pcilib_register_bank_t bank_id = pcilib_find_register_bank(ctx, bank);
- if (bank_id == PCILIB_REGISTER_BANK_INVALID) {
- if (bank) pcilib_error("Invalid register bank is specified (%s)", bank);
- else pcilib_error("Register bank should be specified");
- return PCILIB_ERROR_INVALID_BANK;
- }
-
- return pcilib_write_register_space_internal(ctx, bank_id, addr, n, 0, 0, 0, buf);
- }
- int pcilib_write_register_by_id(pcilib_t *ctx, pcilib_register_t reg, pcilib_register_value_t value) {
- int err;
- size_t i, n;
- pcilib_register_size_t bits;
- pcilib_register_bank_t bank;
- pcilib_register_value_t res;
- const pcilib_register_description_t *r;
- const pcilib_register_bank_description_t *b;
- const pcilib_model_description_t *model_info = pcilib_get_model_description(ctx);
- r = model_info->registers + reg;
- bank = pcilib_find_register_bank_by_addr(ctx, r->bank);
- if (bank == PCILIB_REGISTER_BANK_INVALID) return PCILIB_ERROR_INVALID_BANK;
- b = model_info->banks + bank;
-
- n = r->bits / b->access;
- bits = r->bits % b->access;
- pcilib_register_value_t buf[n + 1];
- memset(buf, 0, (n + 1) * sizeof(pcilib_register_value_t));
-
- if ((b->endianess == PCILIB_BIG_ENDIAN)||((b->endianess == PCILIB_HOST_ENDIAN)&&(ntohs(1) == 1))) {
- pcilib_error("Big-endian byte order support is not implemented");
- return PCILIB_ERROR_NOTSUPPORTED;
- } else {
- if (b->access == sizeof(pcilib_register_value_t) * 8) {
- buf[0] = value;
- } else {
- for (i = 0, res = value; (res > 0)&&(i <= n); ++i) {
- buf[i] = res & BIT_MASK(b->access);
- res >>= b->access;
- }
-
- if (res) {
- pcilib_error("Value %i is too big to fit in the register %s", value, r->name);
- return PCILIB_ERROR_OUTOFRANGE;
- }
- }
- }
- err = pcilib_write_register_space_internal(ctx, bank, r->addr, n, r->offset, bits, r->rwmask, buf);
- return err;
- }
- int pcilib_write_register(pcilib_t *ctx, const char *bank, const char *regname, pcilib_register_value_t value) {
- int reg;
-
- reg = pcilib_find_register(ctx, bank, regname);
- if (reg == PCILIB_REGISTER_INVALID) {
- pcilib_error("Register (%s) is not found", regname);
- return PCILIB_ERROR_NOTFOUND;
- }
- return pcilib_write_register_by_id(ctx, reg, value);
- }
|