仿C语言fread的无符号整数泛型函数为何总是返回0？

首先，咱们来拆解你遇到的问题核心：你的泛型fread函数看似返回了正确的字节数，但实际读取不到有效数据，导致结果始终为0x00，进而触发无限循环。下面是具体的错误原因和修复方案：

一、核心错误原因 #

1. Vec初始化逻辑错误 #

你用Vec::with_capacity(element_count * type_size)创建缓冲区时，只预分配了内存空间，但Vec的实际长度还是0。这意味着buf.as_mut_slice()得到的是空切片，调用stream.read_exact(buf_slice)时实际读取了0字节，却被错误地判定为读取了element_count * type_size字节（也就是2字节）。没有数据流入，最终buffer自然被初始化为0。

2. 字节序解析逻辑错误 #

你的移位计算((type_size - i) * 8)存在溢出问题：以u16为例，当i=0时，移位值是16，但u16只有16位，左移16位会直接溢出为0，导致第一个字节完全被丢弃。即使读取到数据，解析出的结果也是错误的。

3. 返回值与C的fread行为不一致 #

C的fread返回的是成功读取的元素个数，而你的函数返回的是字节数，这导致调用逻辑中判断>0时，即使只读到1个无效字节（无法组成完整的u16），也会被判定为读取成功，进一步加剧问题。

二、正确的泛型fread实现 #

要完全复刻C的fread行为，我们需要做到：正确分配缓冲区、按内存布局直接复制字节、返回元素个数而非字节数。以下是两种实现方案：

方案1：使用unsafe内存复制（与C行为完全一致） #

这种方式直接将文件字节复制到目标类型的内存中，和C的fread逻辑完全相同，依赖主机的内存字节序：

extern crate num;

use num::{PrimInt, Unsigned};
use std::fs::File;
use std::io::{BufReader, Read};
use std::mem;
use std::ptr;

fn fread<T>(buffer: &mut [T], stream: &mut BufReader<File>) -> Result<usize, std::io::Error>
where
    T: PrimInt + Unsigned + Copy,
{
    let type_size = mem::size_of::<T>();
    let total_bytes = buffer.len() * type_size;
    // 创建有实际长度的u8缓冲区，而非仅预分配容量
    let mut buf = vec![0u8; total_bytes];
    
    let bytes_read = stream.read(&mut buf)?;
    // 计算能组成的完整元素个数
    let elements_read = bytes_read / type_size;
    
    // 直接将字节复制到目标缓冲区（unsafe操作，确保T是Plain Old Data类型）
    unsafe {
        let src = buf.as_ptr() as *const T;
        let dst = buffer.as_mut_ptr();
        ptr::copy_nonoverlapping(src, dst, elements_read);
    }
    
    // 返回成功读取的元素个数，与C的fread行为一致
    Ok(elements_read)
}

fn main() -> Result<(), std::io::Error> {
    let cmd_line_args = std::env::args().collect::<Vec<_>>();
    let f = File::open(&cmd_line_args[1])?;
    let mut reader = BufReader::new(f);
    
    let mut base_address = 0u16;
    // 读取一个u16元素，返回1表示成功
    if fread(&mut [base_address], &mut reader)? != 1 {
        panic!("Failed to read base address");
    }
    
    let mut instructions = Vec::new();
    let mut buf = [0u16; 1];
    // 判断是否成功读取一个完整元素
    while fread(&mut buf, &mut reader)? == 1 {
        instructions.push(buf[0]);
    }
    
    println!("{:#04x}", base_address);
    for instr in instructions {
        println!("0x{:04x}", instr);
    }
    
    Ok(())
}

方案2：安全的字节序解析（无需unsafe） #

如果你不想使用unsafe代码，可以手动按指定字节序解析字节。比如针对小端字节序（常见于x86等16位架构）：

extern crate num;

use num::{PrimInt, Unsigned};
use std::fs::File;
use std::io::{BufReader, Read};
use std::mem;

// 将字节切片转换为指定类型（小端字节序）
fn bytes_to_value<T: PrimInt + Unsigned>(bytes: &[u8]) -> T {
    bytes.iter().enumerate().fold(num::zero(), |acc, (i, &b)| {
        acc | T::from(b).expect("Failed to cast u8 to target type") << (i * 8)
    })
}

fn fread<T>(buffer: &mut [T], stream: &mut BufReader<File>) -> Result<usize, std::io::Error>
where
    T: PrimInt + Unsigned,
{
    let type_size = mem::size_of::<T>();
    let total_bytes = buffer.len() * type_size;
    let mut buf = vec![0u8; total_bytes];
    
    let bytes_read = stream.read(&mut buf)?;
    let elements_read = bytes_read / type_size;
    
    // 逐个解析完整元素
    for i in 0..elements_read {
        let start = i * type_size;
        let end = start + type_size;
        buffer[i] = bytes_to_value(&buf[start..end]);
    }
    
    Ok(elements_read)
}

// 主函数与方案1一致，此处省略

三、关键修复点总结 #

• 用vec![0u8; n]创建有实际长度的缓冲区，而非仅预分配容量
• 返回成功读取的元素个数，而非字节数，与C的fread行为对齐
• 要么用unsafe直接内存复制（完全复刻C逻辑），要么手动按正确字节序解析字节

内容的提问来源于stack exchange，提问作者Jarak