如何将文件映射到内存

实际案例
  1. 在访问某些二进制文件时,希望能把文件映射到内存中,可以实现随机访问(如 framebuffer设备文件);
  2. 某些嵌入式设备,寄存器被编址到内存地址空间,我们可以映射/dev/mem某范围,去访问这些寄存器;
  3. 如果多个进程映射同一个文件,还能实现进程通信的目的。

解决方案:使用标准库中mmap模块的mmap()函数,它需要一个打开的文件描述符作为参数。

注:本案例在Linux系统下实验。

在shell下,我们通过dd命令创建了一个数据全为0且大小为1M的二进制文件,通过od -x命令以十六进制的方式查看该文件。

rookie@rookie ~/Desktop $ dd if=/dev/zero of=demo.bin bs=1024 count=1024
1024+0 records in
1024+0 records out
1048576 bytes (1.0 MB, 1.0 MiB) copied, 0.007549 s, 139 MB/s

rookie@rookie ~/Desktop $ od -x demo.bin 
0000000 0000 0000 0000 0000 0000 0000 0000 0000
*
4000000

我们在shell中键入如下命令将二进制文件映射到内存:

>>> f = open('demo.bin', 'r+b')
>>> import mmap
>>> m = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_WRITE)
>>> type(m)

>>> m[0]
'\x00'
>>> m[10:20]
'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
>>> m[0] = '\x88'

rookie@rookie ~/Desktop $ od -x demo.bin 
0000000 0088 0000 0000 0000 0000 0000 0000 0000
0000020 0000 0000 0000 0000 0000 0000 0000 0000
*
4000000

上述命令中,我们使用Python内置的open()函数来得到文件描述符,并使用f.fileno()来获取文件描述符。除此之外,我们也可使用os模块下的open()函数来得到文件描述符。mmap()函数的第二个参数0表示将整个文件映射到内存,第三个参数mmap.ACCESS_WRITE表示获取写权限。

通过调用mmap()函数,我们可以得到mmap的对象m。我们可以利用这个对象m进行类似列表一样的操作。

注:在mmap()函数中还有一个offset参数,其表示指定映射文件的某个区域,但其必须是内存页大小的整数倍。

>>> m = mmap.mmap(f.fileno(), mmap.PAGESIZE*8, access=mmap.ACCESS_WRITE, offset=mmap.PAGESIZE*4)
>>> m[:0x1000] = '\xaa' * 0x1000

rookie@rookie ~/Desktop $ od -x demo.bin 
0000000 0088 0000 0000 0000 0000 0000 0000 0000
0000020 0000 0000 0000 0000 0000 0000 0000 0000
*
0040000 aaaa aaaa aaaa aaaa aaaa aaaa aaaa aaaa
*
0050000 0000 0000 0000 0000 0000 0000 0000 0000
*
4000000

通过上述操作,我们在内存中修改数据的同时也将文件的数据给修改了。因此,我们达到了将文件映射到内存的目的。

注:本部分内容本人也是有点囫囵吞枣,希望路过的大佬能够指点迷津,谢谢!

你可能感兴趣的:(如何将文件映射到内存)