Linux dd 命令详解：工作原理与实用指南（C/C++代码实现）

这段代码是一个模仿 Linux dd 命令的工具，它用于在不同文件之间复制数据。dd 是一个非常强大的命令行工具，可以用于数据备份、转换和复制。下面我将详细解释这段代码的原理、实现方式以及如何运行和测试。

Linux dd 命令的工作原理

dd 命令是 Unix 和 Linux 系统中非常强大的文件复制和转换工具。它通过指定块大小（block size）来读取和写入数据，实现高效的文件或设备复制。dd 命令不仅能够处理普通文件，还能直接操作设备文件，这使得它在系统备份、恢复和数据克隆等任务中非常有用。

1. 基本语法：dd if=输入文件 of=输出文件 [选项]

   • if：指定输入文件名或设备名。
   • of：指定输出文件名或设备名。
   • [选项]：各种可选参数，用于控制复制行为。

2. 常用选项：

   • bs：设置读写的块大小。例如：bs=4M。
   • count：复制的块数。例如：count=1。
   • conv：指定数据转换选项，如 conv=sync 确保同步模式。
   • status=progress：显示详细的进度信息。

3. 工作流程：

   • 打开源和目标文件/设备：dd 命令会先尝试打开指定的输入和输出文件或设备。
   • 按块读取和写入：根据指定的块大小（bs），dd 从输入文件或设备读取数据，并写入到输出文件或设备中。
   • 数据处理：在读写过程中，dd 可以根据指定的转换选项对数据进行处理，如大小写转换、字节顺序转换等。
   • 重复操作：上述过程会根据指定的块数（count）反复进行，直到完成所有数据的复制或转换。

命令行使用

dd 的基本命令行格式如下：

dd if=<input_file> of=<output_file> [其他参数]

• if：指定输入文件。
• of：指定输出文件。

以下是一些常用的参数示例：

• bs：设置块大小，例如 bs=1024。
• count：设置复制的块数，例如 count=10。
• skip：设置跳过的块数，例如 skip=5。
• seek：设置输出文件中跳过的块数，例如 seek=5。

示例

复制一个文件的前 10MB 到另一个文件：

dd if=/dev/zero of=example.img bs=1M count=10

将一个磁盘分区备份到另一个磁盘分区：

dd if=/dev/sda of=/dev/sdb

基本文件复制：

dd if=inputfile of of=outputfile bs=64K count=1

将 inputfile 复制到 outputfile，每次读取和写入 64KB 的数据块，只复制一个块。

备份和还原硬盘：

dd if=/dev/sda of=/path/to/backup.img bs=4M 
dd if=/path/to/backup.img of=/dev/sdb bs=4M 

将整个硬盘 /dev/sda 备份到 backup.img 文件中，然后将 backup.img 还原到 /dev/sdb。

创建镜像文件：

dd if=/dev/zero of=imagefile.img bs=1G count=10

创建一个名为 imagefile.img 的 10GB 镜像文件，内容全为零。

制作启动盘：

dd if=boot.img of=/dev/sdb bs=4M 

将 boot.img 写入到 USB 设备 /dev/sdb，制作可启动的 USB 盘。

擦除硬盘数据：

dd if=/dev/urandom of of=/dev/sda bs=4M 

使用随机数据覆盖整个硬盘，确保数据无法恢复。

日常定位分析

dd 命令在系统恢复、数据恢复和磁盘克隆等场景中非常有用。例如，当你需要从一个损坏的文件系统中恢复数据时，可以使用 dd 来复制文件系统的一部分到另一个健康的磁盘上，然后对复制的数据进行分析和恢复。

三、dd 命令在实际工作中的定位与分析

dd 命令因其强大的功能和灵活性，在系统管理和运维工作中有着广泛的应用场景。以下是几个典型的应用场景及其分析：

1. 系统备份与恢复：

   • 场景：定期备份服务器上的硬盘或分区，以防数据丢失或系统故障。
   • 分析：通过 dd 命令，可以创建整个硬盘或分区的镜像文件，方便存储和快速恢复。此外，还可以使用压缩工具（如 gzip）进一步减小镜像文件的大小。

2. 数据克隆和迁移：

   • 场景：在更换硬盘或迁移数据时，需要将旧硬盘上的数据完整复制到新硬盘。
   • 分析：dd 命令可以直接操作设备文件，无需经过文件系统，从而提高数据复制的效率和可靠性。这对于大规模数据迁移尤其有用。

3. 制作启动盘和恢复盘：

   • 场景：需要制作可启动的 USB 盘或 CD/DVD，用于系统安装或故障排查。
   • 分析：dd 命令可以将 ISO 镜像文件直接写入到 USB 或光盘设备，操作简单且高效。这在紧急情况下尤为重要，如系统崩溃后的恢复工作。

4. 安全删除数据：

   • 场景：需要彻底删除敏感数据，确保无法通过恢复工具找回。
   • 分析：通过用随机数据覆盖硬盘，可以有效防止数据被恢复。这种方法比单纯的文件删除更为安全，适用于处理包含敏感信息的硬盘。

5. 性能测试：

   • 场景：测试磁盘的读写速度，评估存储设备的性能。
   • 分析：dd 命令可以生成大规模的测试数据，并通过计时等方式测量磁盘的读写速度。这对于存储设备的选型和性能优化具有重要参考价值。

Linux dd 命令详解：工作原理与实用指南（C/C++代码实现）

size_t free_mem()
{uint64_t n = 0;char buf[1024], found = 0;FILE *f = fopen("/proc/meminfo", "r");if (!f)return 1024*1024;memset(buf, 0, sizeof(buf));for (;!feof(f);) {fgets(buf, sizeof(buf), f);if (strstr(buf, "MemFree:")) {found = 1;break;}}fclose(f);if (!found)return 1024*1024;n = strtoul(buf + 9, NULL, 10);if (!n)return 1024*1024;/* kB? */if (strchr(buf + 9, 'k'))n <<= 10;else if (strchr(buf + 9, 'M'))n <<= 20;return n/2;
}...#ifdef ANDROIDint copy_splice(struct dd_config *);int copy_splice_cores(struct dd_config *ddc)
{return copy_splice(ddc);
}#else
int copy_splice_cores(struct dd_config *ddc)
{int ifd, ofd, p[2] = {-1, -1};ssize_t r = 0, cpu_size = 0;size_t n = 0, min_bs = 4096;cpu_set_t *cpu_set = NULL;if (prepare_copy(ddc, &ifd, &ofd) < 0)return -1;if ((cpu_set = CPU_ALLOC(2)) == NULL) {close(ifd); close(ofd);return -1;}cpu_size = CPU_ALLOC_SIZE(2);CPU_ZERO_S(cpu_size, cpu_set);if (pipe(p) < 0) {ddc->saved_errno = errno;close(ifd); close(ofd);close(p[0]); close(p[1]);return -1;}#ifdef F_SETPIPE_SZfor (n = 29; n >= 20; --n) {if (fcntl(p[0], F_SETPIPE_SZ, 1<<n) != -1)break;}
#endifn = ddc->bs;if (fork() == 0) {/* bind to CPU#0 */CPU_SET_S(ddc->cores - 1, cpu_size, cpu_set);sched_setaffinity(0, cpu_size, cpu_set);close(p[0]);for (;ddc->b_in != ddc->count && !sigint;) {if (n > ddc->count - ddc->b_in)n = ddc->count - ddc->b_in;r = splice(ifd, NULL, p[1], NULL, n, SPLICE_F_MORE|SPLICE_F_NONBLOCK);if (r == 0)break;if (r < 0) {if (errno != EAGAIN)break;/* If running out of pipe buffer, decrease bs */r = 0;n = min_bs;}ddc->b_in += r;}exit(0);}/* bind to CPU#1 */CPU_SET_S(ddc->cores - 2, cpu_size, cpu_set);sched_setaffinity(0, cpu_size, cpu_set);for (;ddc->b_out != ddc->count;) {r = splice(p[0], NULL, ofd, NULL, n, SPLICE_F_MORE);if (r <= 0) {ddc->saved_errno = errno;break;}ddc->b_out += r;++ddc->rec_out;}ddc->rec_in = ddc->rec_out;close(ifd);close(ofd);close(p[0]);close(p[1]);wait(NULL);if (r < 0)return -1;return 0;
}
#endifint copy_splice(struct dd_config *ddc)
{
...if (prepare_copy(ddc, &ifd, &ofd) < 0)return -1;if (pipe(p) < 0) {ddc->saved_errno = errno;close(ifd); close(ofd);close(p[0]); close(p[1]);return -1;}#ifdef F_SETPIPE_SZfor (n = 29; n >= 20; --n) {if (fcntl(p[0], F_SETPIPE_SZ, 1<<n) != -1)break;}
#endifn = ddc->bs;for (;ddc->b_out != ddc->count && !sigint;) {if (n > ddc->count - ddc->b_out)n = ddc->count - ddc->b_out;r = splice(ifd, NULL, p[1], NULL, n, SPLICE_F_MORE);if (r <= 0) {ddc->saved_errno = errno;break;}++ddc->rec_in;r = splice(p[0], NULL, ofd, NULL, r, SPLICE_F_MORE);if (r <= 0) {ddc->saved_errno = errno;break;}ddc->b_out += r;++ddc->rec_out;}close(ifd);close(ofd);close(p[0]);close(p[1]);if (r < 0)return -1;return 0;
}int copy_mmap(struct dd_config *ddc)
{
...if (prepare_copy(ddc, &ifd, &ofd) < 0)return -1;if (ddc->fsize != (off_t)-1) {if (ftruncate(ofd, ddc->fsize) < 0) {ddc->saved_errno = errno;close(ifd);close(ofd);return -1;}}for (;ddc->b_out != ddc->count && !sigint;) {n = ddc->mmap;bs = ddc->bs;if (n > ddc->count - ddc->b_out)n = ddc->count - ddc->b_out;if (bs > n)bs = n;if (ddc->fsize == (off_t)-1) {if (ftruncate(ofd, ddc->b_out + n) < 0) {ddc->saved_errno = errno;break;}}addr = mmap(NULL, n, PROT_WRITE, MAP_SHARED, ofd, ddc->b_out + ddc->skip);if (addr == MAP_FAILED) {ddc->saved_errno = errno;break;}for (i = 0; i < n; i += r) {if (i + bs > n)bs = n - i;r = read(ifd, addr + i, bs);if (r <= 0) {ddc->saved_errno = errno;munmap(addr, n);break;}ddc->b_out += r;++ddc->rec_in;}...}if (ddc->fsize != ddc->b_out)ftruncate(ofd, ddc->b_out);close(ifd);close(ofd);if (r < 0 || addr == MAP_FAILED)return -1;return 0;
}int copy_sendfile(struct dd_config *ddc)
{
...off = ddc->skip;n = ddc->bs;for (;ddc->b_out < ddc->count && !sigint;) {if (n > ddc->count - ddc->b_out)n = ddc->count - ddc->b_out;r = sendfile(ofd, ifd, &off, n);if (r < 0) {ddc->saved_errno = errno;ret = -1;break;}++ddc->rec_in; ++ddc->rec_out;ddc->b_in += r;ddc->b_out += r;}close(ifd);close(ofd);return ret;
}int copy(struct dd_config *ddc)
{
...if (ddc->cores)r = copy_splice_cores(ddc);else if (ddc->mmap)r = copy_mmap(ddc);else if (ddc->sf)r = copy_sendfile(ddc);elser = copy_splice(ddc);ddc->t_end = time(NULL);/* 避免div为零 */if (ddc->t_start == ddc->t_end)++ddc->t_end;return r;
}void print_stat(const struct dd_config *ddc)
{
...#ifdef ANDROIDfprintf(stderr, "%llu records in\n%llu records out\n%llu bytes (%llu MB) copied, %lu s, %f MB/s [%f mB/s]\n",ddc->rec_in, ddc->rec_out, ddc->b_out, ddc->b_out/(1<<20),ddc->t_end - ddc->t_start,((double)(ddc->b_out/(1<<20)))/(ddc->t_end - ddc->t_start),((double)(ddc->b_out/(1000*1000)))/(ddc->t_end - ddc->t_start));
#elsefprintf(stderr, "%lu records in\n%lu records out\n%lu bytes (%lu MB) copied, %lu s, %f MB/s [%f mB/s]\n",ddc->rec_in, ddc->rec_out, ddc->b_out, ddc->b_out/(1<<20),ddc->t_end - ddc->t_start,((double)(ddc->b_out/(1<<20)))/(ddc->t_end - ddc->t_start),((double)(ddc->b_out/(1000*1000)))/(ddc->t_end - ddc->t_start));
#endif}void sig_int(int x)
{fprintf(stderr, "SIGINT! Aborting ...\n");sigint = 1;return;
}int main(int argc, char **argv)
{
...config.bs = 1<<16;config.in = "/dev/stdin";config.out = "/dev/stdout";/* 模拟“dd”参数解析 */for (i = 1; i < argc; ++i) {if (strcmp(argv[i], "-h") == 0 ||strcmp(argv[i], "--help") == 0)usage(argv[0]);memset(buf, 0, sizeof(buf));if (sscanf(argv[i], "if=%1023c", buf) == 1)config.in = strdup(buf);else if (sscanf(argv[i], "of=%1023c", buf) == 1)config.out = strdup(buf);else if (sscanf(argv[i], "skip=%1023c", buf) == 1)config.skip = strtoul(buf, NULL, 10);else if (sscanf(argv[i], "seek=%1023c", buf) == 1)config.seek = strtoul(buf, NULL, 10);else if (sscanf(argv[i], "count=%1023c", buf) == 1)config.count = strtoul(buf, NULL, 10);else if (sscanf(argv[i], "mmap=%1023c", buf) == 1) {if (!config.cores) {/* Size in MB */config.mmap = strtoul(buf, NULL, 10);config.mmap <<= 20;}} else if (sscanf(argv[i], "cores=%1023c", buf) == 1) {config.cores = strtoul(buf, NULL, 10);if (config.cores < 2)config.cores = 2;config.mmap = 0;} else if (strcmp(argv[i], "send") == 0) {config.sf = 1;} else if (strcmp(argv[i], "direct") == 0) {config.direct = 1;} else if (sscanf(argv[i], "bs=%1023c", buf) == 1) {config.bs = strtoul(buf, NULL, 10);} else if (strcmp(argv[i], "bs") == 0) {config.bs = 0;} else if (strcmp(argv[i], "quiet") == 0) {config.quiet = 1;} else if (strcmp(argv[i], "nosync") == 0) {config.nosync = 1;}}...return 0;
}

If you need the complete source code, please add the WeChat number (c17865354792)

编译完成后，你可以使用以下命令行参数来测试它：

./linux_dd if=<input_file> of=<output_file> [其他参数]

例如，你可以使用以下命令来复制一个文件：

./linux_dd if=/dev/stdin of=/dev/stdout bs=1024 count=1024

这个命令会从标准输入读取数据，并将数据写入到标准输出，每次复制 1024 字节，总共复制 1024 次。

代码中提供了多种复制策略：

1. splice(2)：使用 Linux 的 splice 系统调用来在管道和文件之间传输数据，这种方式可以有效地利用内核缓冲区，减少数据复制过程中的上下文切换。

2. mmap(2)：通过内存映射的方式直接在内存中操作文件数据，这种方式适用于大块数据的复制。

3. sendfile(2)：在内核层面直接将数据从一个文件描述符传输到另一个，减少了数据在用户空间的拷贝。

4. 多核处理：代码还支持将数据复制任务分配到多个 CPU 核心上，以提高复制效率。

补充内容

1. 安全性：在使用 dd 时，一定要小心指定正确的输入和输出文件，错误的命令可能会导致数据丢失。
2. 效率：选择合适的块大小可以显著影响 dd 的性能。通常，较大的块大小可以提高复制速度。
3. 错误处理：dd 命令在执行过程中可能会遇到错误，了解如何解读错误信息对于解决问题至关重要。
4. 日志记录：dd 命令执行时可以重定向输出到日志文件，以便于事后分析。
5. 进度监控：可以通过 status=progress 参数来监控 dd 命令的执行进度。

结语

dd 命令是一个功能强大的工具，它在 Linux 系统中扮演着重要的角色。了解其工作原理和使用方法，可以帮助你更有效地管理和操作数据。在使用时，务必谨慎，以避免不必要的数据损失。

Welcome to follow WeChat official account【程序猿编码】