讲师中心 微信公众号
AI工具推荐 视频效率加速

Java Socket 文件传输中因未校验剩余字节数导致内容重复写入的解决方案

秋静酱_1810

秋静酱_1810

发布时间:2026-02-11 22:08:56

|

848人浏览过

|

来源于php中文网

原创

Java Socket 文件传输中因未校验剩余字节数导致内容重复写入的解决方案

本文详解 java 原生 socket 文件传输时因循环读写逻辑缺陷(未严格校验剩余待传字节数)引发的内容重复问题,提供可直接复用的修复代码、关键原理说明及健壮性增强建议。

在基于 DataInputStream/DataOutputStream 的 Java Socket 文件传输实现中,一个常见却隐蔽的 Bug 是:客户端发送端循环条件未充分约束“剩余待传输字节数”为零的情况,导致最后一次 read() 返回 0 字节后仍继续执行写入操作,从而向服务端重复发送缓冲区残留数据——这正是题中“content duplicates”的根本原因。

问题核心在于客户端 sendFile() 方法中的 while 循环判断:

while(size > 0 && (bytes = input.read(buffer,0,(int)Math.min(buffer.length, size))) != -1)

该条件看似合理,但存在致命漏洞:当 size > 0 为真,而 input.read(...) 因文件末尾恰好读取到 0 字节(即 bytes == 0)时,read() 不会返回 -1(仅在流真正结束且无更多数据时才返回 -1),而是合法返回 0。此时循环体仍会执行 out.write(buffer, 0, bytes) —— 即 out.write(buffer, 0, 0),虽不写入新数据,但若服务端接收逻辑未同步校验 bytes > 0,或缓冲区含历史残留内容,就极易造成重复解析或写入。

✅ 正确做法是:显式检查每次 read() 的返回值 bytes,仅当 bytes > 0 时才进行写入和字节计数更新。同时,服务端接收逻辑也必须遵循相同原则,避免对 bytes == 0 的无效读取做误处理。

立即学习“Java免费学习笔记(深入)”;

Java注释规范Skills
Java注释规范Skills

当用户要求为 Java 代码添加注释、修改注释、审查注释、整理注释格式,或要求按照代码注释规范处理 Java 文件时,使用本技能。本技能用于统一生成和维护 Java 类注释模板、按 Git 对比 release 判断版本号、补充每行代码上方中文块注释、调整注释空行规则,并保持 Java 文件 UTF-8 无 BOM 编码。典型触发场景包括:补充类注释、更新 @version、检查 @author 来源、整理行尾注释、把注释放到代码上一行、检查注释是否符合规范、构建前检查 Java 注释风格。

下载

以下是修复后的客户端关键逻辑(sendFile 方法):

private void sendFile(String path) {
    File file = new File(path);
    try (FileInputStream input = new FileInputStream(file)) {
        long size = file.length();
        System.out.println("Sending file size: " + size);
        out.writeLong(size); // 先发送文件总长度

        byte[] buffer = new byte[1024];
        int bytes;
        while (size > 0) {
            // 每次最多读取 min(buffer.length, 剩余字节数)
            int toRead = (int) Math.min(buffer.length, size);
            bytes = input.read(buffer, 0, toRead);

            // 关键修复:仅当实际读取到正数字节时才写入
            if (bytes > 0) {
                out.write(buffer, 0, bytes);
                out.flush(); // 确保立即发送(小文件可省略,大文件建议保留)
                size -= bytes;
            } else if (bytes == 0) {
                // read() 返回 0 是合法但罕见情况(如底层流阻塞后暂无数据),此处可加日志或 break 防死循环
                System.err.println("Warning: read() returned 0 bytes unexpectedly.");
                break;
            } else { // bytes == -1,流已结束
                break;
            }
        }
        System.out.println("File sent successfully.");
    } catch (IOException e) {
        e.printStackTrace();
    }
}

对应地,服务端 receiveFile() 方法也需同步强化:

private void receiveFile(String path) {
    try (FileOutputStream output = new FileOutputStream(new File(path))) {
        long size = in.readLong();
        System.out.println("Expecting file size: " + size);

        byte[] buffer = new byte[1024];
        int bytes;
        while (size > 0) {
            int toRead = (int) Math.min(buffer.length, size);
            bytes = in.read(buffer, 0, toRead);

            if (bytes > 0) {
                output.write(buffer, 0, bytes);
                size -= bytes;
            } else if (bytes == 0) {
                System.err.println("Warning: server read() returned 0 bytes.");
                break;
            } else {
                throw new IOException("Unexpected end of stream while receiving file");
            }
        }
        System.out.println("File received successfully.");
    } catch (IOException e) {
        e.printStackTrace();
    }
}

? 重要注意事项与最佳实践:

  • 资源管理:使用 try-with-resources(如示例所示)自动关闭 FileInputStream/FileOutputStream,避免资源泄漏;
  • flush() 的权衡:频繁 flush() 降低性能,但能确保数据及时到达;若传输大文件,可考虑每 N KB 刷一次,或依赖 TCP 自身可靠性,移除 flush()(本例保留以保证教学清晰性);
  • 字符打印调试慎用:原代码中遍历 buffer 强制转 (char) 并打印,易因编码/乱码导致控制台异常,且干扰二进制文件(如图片、jar)传输——生产环境应移除所有非必要字符输出逻辑;
  • 异常处理完整性:捕获 IOException 后不应静默吞掉,至少记录日志;NullPointerException 的空 catch 块是严重反模式,必须删除;
  • 扩展性提示:若需传输多个文件或带元数据,建议在协议层增加文件名、MIME 类型等字段,而非依赖单一长度前缀。

通过以上修正,即可彻底解决因 read() 返回 0 引发的重复写入问题,构建出稳定、可预测的 Socket 文件传输通道。

热门AI工具

更多
立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

相关专题

更多
while的用法
while的用法

while的用法是“while 条件: 代码块”,条件是一个表达式,当条件为真时,执行代码块,然后再次判断条件是否为真,如果为真则继续执行代码块,直到条件为假为止。本专题为大家提供while相关的文章、下载、课程内容,供大家免费下载体验。

354

2023.09.25

点击input框没有光标怎么办
点击input框没有光标怎么办

点击input框没有光标的解决办法:1、确认输入框焦点;2、清除浏览器缓存;3、更新浏览器;4、使用JavaScript;5、检查硬件设备;6、检查输入框属性;7、调试JavaScript代码;8、检查页面其他元素;9、考虑浏览器兼容性。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

874

2023.11.24

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

100

2026.10.22

Gradle Wrapper配置教程
Gradle Wrapper配置教程

本专题介绍Gradle Wrapper的基本作用,涵盖文件组成、生成方法、版本配置和任务调用,帮助你理解项目如何使用指定的Gradle版本,减少不同开发环境带来的构建差异。

60

2026.10.22

Golangci-lint检查器使用教程
Golangci-lint检查器使用教程

本专题整理Golangci-lint常用Linters的用途与配置方法,覆盖错误处理、无效代码、安全、复杂度、命名和依赖检查,帮助项目合理选规则。

60

2026.10.22

Gradle怎么构建Java项目
Gradle怎么构建Java项目

本专题介绍Gradle Java项目构建方法,涵盖Java插件配置、依赖声明、源码编译、测试执行和打包检查,帮助你理解各阶段的任务关系,逐步完成项目构建。

20

2026.10.22

PHP时间戳漏洞修复教程
PHP时间戳漏洞修复教程

本专题汇总PHP常见时间戳漏洞,包括2038年溢出、时区漂移、时间戳伪造、重放攻击、时间盲注与日志审计失效等,并给出升级64位环境、使用 DateTimeImmutable、统一UTC、严格超时校验、预处理绑定、hash_equals 比对等修复方案,帮助开发者快速排查和加固时间相关代码。

60

2026.10.22

Golangci-lint配置文件与规则设置
Golangci-lint配置文件与规则设置

本专题聚焦Golangci-lint YAML配置文件,讲清运行参数、规则启停、目录排除、问题过滤及严重级别设置,便于团队统一检查标准。

20

2026.10.22

Golangci-lint安装与入门教程
Golangci-lint安装与入门教程

本专题讲解Golangci-lint的安装、初始化、基础命令和首次代码检查,覆盖本地环境、Docker及多平台用法,帮助Go开发者快速上手。

40

2026.10.22

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn