简单的寄存器读写程序

功能点

1、支持通过文件映射完成寄存器读写；
2、支持直接读写寄存器；
3、参数个数为4时，表明为读寄存器值；
4、参数个数为5时，表明为寄存器写值。

代码实现

头文件

#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <string.h>
#include <sys/mman.h>
#include <unistd.h>

使用帮助

void usage(char *argv)
{
    printf("Usage:\n\t%s 寄存器地址 寄存器长度 偏移位置 配置的值\n",argv);
}

// 寄存器读取
static int reg_read(int fd, char *argv[])
{
    // 寄存器基地址
    u64_t reg_addr_base_addr = 0;
    u64_t reg_addr = 0;
    sscanf(argv[1], "%lX", &reg_addr_base_addr);
    sscanf(argv[2], "%lX", &reg_addr_len);
    sscanf(argv[3], "%lX", &reg_addr);
    // 通过文件映射读寄存器
    // todo 
    // 直接些寄存器;
    printf("READ ->(%lX + %lX) = %lX\n",reg_addr_base_addr, reg_addr,*(volatile u64_t *)(base + reg_addr));
    return 0;
}

// 寄存器写
void reg_write(int fd, char *argv[])
{
    u64_t value = 0;
    // 寄存器基地址
    u64_t reg_addr_base_addr = 0;
    u64_t reg_addr = 0;
    sscanf(argv[1], "%lX", &reg_addr_base_addr);
    sscanf(argv[2], "%lX", &reg_addr_len);
    sscanf(argv[3], "%lX", &reg_addr);
    sscanf(argv[4], "%lX", &value);
    // 通过文件映射写寄存器
    // todo 
    // 直接些寄存器;
    printf("WRITE->(%lX + %lX) = %lX\n", reg_addr_base_addr, reg_addr, value);

    return;
}
 

入口函数

int main(int argc,char *argv[])
{
    void *base = NULL;
    int ret = 0;

    int fd = open("/dev/mem", O_RDWR | O_SYNC);
    if (fd == -1) {
        printf("Open mem file error!\n");
        usage(argv[0]);
        exit(0);
    }

    if(argc == 4) {
        ret = reg_read(fd, argv);
    } else if(argc == 5) {
        reg_write(fd, argv);
        ret = reg_read(fd, argv);
    } else {
        usage(argv[0]);
    }

    munmap(base, reg_addr_len);
    close(fd);
    return ret;
}