这段代码是一个模仿 Linux dd 命令的工具,它用于在不同文件之间复制数据。dd 是一个非常强大的命令行工具,可以用于数据备份、转换和复制。下面我将详细解释这段代码的原理、实现方式以及如何运行和测试。
Linux dd 命令的工作原理
dd 命令是 Unix 和 Linux 系统中非常强大的文件复制和转换工具。它通过指定块大小(block size)来读取和写入数据,实现高效的文件或设备复制。dd 命令不仅能够处理普通文件,还能直接操作设备文件,这使得它在系统备份、恢复和数据克隆等任务中非常有用。
-
基本语法:
dd if=输入文件 of=输出文件 [选项]
if
:指定输入文件名或设备名。of
:指定输出文件名或设备名。[选项]
:各种可选参数,用于控制复制行为。
-
常用选项:
bs
:设置读写的块大小。例如:bs=4M
。count
:复制的块数。例如:count=1
。conv
:指定数据转换选项,如conv=sync
确保同步模式。status=progress
:显示详细的进度信息。
-
工作流程:
- 打开源和目标文件/设备: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 命令因其强大的功能和灵活性,在系统管理和运维工作中有着广泛的应用场景。以下是几个典型的应用场景及其分析:
-
系统备份与恢复:
- 场景:定期备份服务器上的硬盘或分区,以防数据丢失或系统故障。
- 分析:通过 dd 命令,可以创建整个硬盘或分区的镜像文件,方便存储和快速恢复。此外,还可以使用压缩工具(如 gzip)进一步减小镜像文件的大小。
-
数据克隆和迁移:
- 场景:在更换硬盘或迁移数据时,需要将旧硬盘上的数据完整复制到新硬盘。
- 分析:dd 命令可以直接操作设备文件,无需经过文件系统,从而提高数据复制的效率和可靠性。这对于大规模数据迁移尤其有用。
-
制作启动盘和恢复盘:
- 场景:需要制作可启动的 USB 盘或 CD/DVD,用于系统安装或故障排查。
- 分析:dd 命令可以将 ISO 镜像文件直接写入到 USB 或光盘设备,操作简单且高效。这在紧急情况下尤为重要,如系统崩溃后的恢复工作。
-
安全删除数据:
- 场景:需要彻底删除敏感数据,确保无法通过恢复工具找回。
- 分析:通过用随机数据覆盖硬盘,可以有效防止数据被恢复。这种方法比单纯的文件删除更为安全,适用于处理包含敏感信息的硬盘。
-
性能测试:
- 场景:测试磁盘的读写速度,评估存储设备的性能。
- 分析: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 次。
代码中提供了多种复制策略:
-
splice(2):使用 Linux 的
splice
系统调用来在管道和文件之间传输数据,这种方式可以有效地利用内核缓冲区,减少数据复制过程中的上下文切换。 -
mmap(2):通过内存映射的方式直接在内存中操作文件数据,这种方式适用于大块数据的复制。
-
sendfile(2):在内核层面直接将数据从一个文件描述符传输到另一个,减少了数据在用户空间的拷贝。
-
多核处理:代码还支持将数据复制任务分配到多个 CPU 核心上,以提高复制效率。
补充内容
- 安全性:在使用
dd
时,一定要小心指定正确的输入和输出文件,错误的命令可能会导致数据丢失。 - 效率:选择合适的块大小可以显著影响
dd
的性能。通常,较大的块大小可以提高复制速度。 - 错误处理:
dd
命令在执行过程中可能会遇到错误,了解如何解读错误信息对于解决问题至关重要。 - 日志记录:
dd
命令执行时可以重定向输出到日志文件,以便于事后分析。 - 进度监控:可以通过
status=progress
参数来监控dd
命令的执行进度。
结语
dd
命令是一个功能强大的工具,它在 Linux 系统中扮演着重要的角色。了解其工作原理和使用方法,可以帮助你更有效地管理和操作数据。在使用时,务必谨慎,以避免不必要的数据损失。
Welcome to follow WeChat official account【程序猿编码】