Python中查询块设备文件大小

Question

我有一个 Python 脚本，它读取标记不可读扇区的文件（通常来自光学介质），以允许重新尝试在不同的光学读取器上读取所述不可读扇区。

我发现我的脚本不适用于块设备（例如 /dev/sr0），以便创建包含的 ISO9660/UDF 文件系统的副本，因为 os.stat().st_size 为零。该算法目前需要提前知道文件大小；我可以更改它，但是问题（了解块设备大小）仍然存在，并且这里没有得到解答，所以我提出这个问题。

我知道以下两个相关的SO问题：

• 确定块设备的大小（/proc/partitions，通过ctypes进行ioctl）
• 如何在Python中检查文件大小？（关于非特殊文件）

因此，我问：在Python中，如何才能我获取块设备文件的文件大小？

Answer 1

我所达到的“最干净”（即不依赖于外部卷并且最可重用）的Python解决方案是打开设备文件并在末尾查找，返回文件偏移量：

def get_file_size(filename):
    "Get the file size by seeking at end"
    fd= os.open(filename, os.O_RDONLY)
    try:
        return os.lseek(fd, 0, os.SEEK_END)
    finally:
        os.close(fd)

Answer 2

Linux 特定的基于 ioctl 的解决方案：

import fcntl
import struct

device_path = '/dev/sr0'

req = 0x80081272 # BLKGETSIZE64, result is bytes as unsigned 64-bit integer (uint64)
buf = b' ' * 8
fmt = 'L'

with open(device_path) as dev:
    buf = fcntl.ioctl(dev.fileno(), req, buf)
bytes = struct.unpack('L', buf)[0]

print device_path, 'is about', bytes / (1024 ** 2), 'megabytes'

当然，其他 Unix 系统的 req、buf、fmt 会有不同的值。

Answer 3

另一种可能的解决方案是

def blockdev_size(path):
    """Return device size in bytes.
    """
    with open(path, 'rb') as f:
        return f.seek(0, 2) or f.tell()

或 f.tell() 部分是为了 Python 2 可移植性而存在的 - file.seek() 在 Python 2 中返回 None。

魔术常量 2 可以替换为 io.SEEK_END。

Answer 4

在Linux中，即使没有sudo也可以读取/sys/block/${dev}/size。要获取 /dev/sdb 的大小，只需执行以下操作：

print( 512 * int(open('/sys/block/sdb/size','r').read()) )

另请参阅 https://unix.stackexchange .com/a/52219/384116

Answer 5

此解决方案使用 lsblk（由 util-linux 提供）并且不需要 sudo：

import json
import subprocess

def get_device_size_bytes(device: str) -> int:
    output = subprocess.run(
        ["lsblk", device, "--json", "--bytes"], capture_output=True, check=True
    )
    dev_info = json.loads(output.stdout)
    size_bytes = dev_info["blockdevices"][0]["size"]
    return size_bytes

更短的解决方案仅从 lsblk 获取大小：

import subprocess

def get_device_size_bytes(device: str) -> int:
    output = subprocess.run(
        ["lsblk", device, "--output", "SIZE", "--bytes", "--noheadings", "--nodeps"],
        capture_output=True,
        check=True,
    )
    size = int(output.stdout.decode())
    return size

Answer 6

尝试适应其他答案：

import fcntl
c = 0x00001260 ## check man ioctl_list, BLKGETSIZE
f = open('/dev/sr0', 'ro')
s = fcntl.ioctl(f, c)
print s

我手头没有合适的计算机来测试这一点。我很想知道它是否有效:)