linux - 如何重写文件内容？ Linux x86_64、汇编、GAS

Question

嗨，我得到了这个令人兴奋的任务，实际上几乎完成了..但它在一个有趣的方面失败了。 任务是加载一个包含整数的文件，对它们进行排序并将它们写入文件。耶...好吧，我的程序正在这样做，但它保留了原始内容。即:

让我们说 45 32

应该得到有序的内容 32 45

好吧，我的程序保留了原始内容并添加了新内容: 45 32 32 45。

那么有什么解决这个问题的建议吗？虽然删除原始文件并以相同的名称创建一个新文件。但那有点失败，如果文件有非整数并且我的代码有关于那个的错误报告。

我在这里给出重要的代码:

_OpenFile:
    movq $2, %rax           # open file
    movq $inputfile, %rdi   # filename
    movq $2, %rsi           # read and write
    movq $0644, %rdx        # setting proper permissions
    syscall
    ret

和:

_PrintNumber: #needs rdi as numberholder
    movq $1, %r9            # count the number of chars to print
    push $10                # store the chars on the stack, we always have '\n'
    movq %rdi, %rax         # things are easier with it in rax
    movq $10, %rcx
    decode_loop:
    movq $0, %rdx
    idivq %rcx              # do rdx:rax / rcx
    addq $48, %rdx          # convert the remainder to an ASCII digit
    pushq %rdx              # and save it on the stack
    addq $1, %r9            # while counting the number of chars
    cmpq $0, %rax
    jne decode_loop         # loop until rax == 0
    write_loop:
    movq $1, %rax           # write
    movq $3, %rdi           # to the file
    movq %rsp, %rsi         # which is stored here
    movq $1, %rdx           # a single character/byte
    syscall

    addq $8, %rsp           # pop the character
    addq $-1, %r9           # correct the char count

    jne write_loop          # loop until r9 reaches 0
    ret

感谢所有愿意对此发表评论的人!

Answer 1

看起来您正在使用 O_APPEND 重新打开文件，或者您以读/写方式打开它并且在重写它之前没有搜索到开头。 (所以读取整个文件后，文件描述符的位置就是文件的末尾，所以新写入的数据会到那里。)

lseek(2) system call 是您移动文件位置所需要的。 lseek(fd, 0, SEEK_SET) 倒回到开头。 (参数进入 EDI、RSI、EDX，就像 x86-64 System V 系统调用约定的正常情况一样，内核接口(interface)与 libc 接口(interface)匹配。)

由于您要写出的数据具有相同的长度，因此您不需要在开始重写之前对文件进行 ftruncate(fd, len)。您将覆盖所有字节直到最后。

---

顺便说一句，您不需要分别对每个字符进行 write；您可以制作一个包含数字所有 ASCII 字节的小缓冲区，并进行一次 write 系统调用；效率更高，实际上需要更少的代码:Printing an integer as a string with AT&T syntax, with Linux system calls instead of printf。我在那里的回答还表明你可以 #include <asm/unistd.h> 并使用像
这样的代码 mov $__NR_write, %eax # SYS_write, from unistd_64.h
如果您使用 .S 文件而不是使用数字文字作为系统调用号，那么 gcc 将通过预处理器运行它。

不幸的是，像 <unistd.h> (不是 asm/unistd.h )这样的大多数 header 也有 C 声明，所以你不能像 SEEK_SET 或 O_RDWR 这样容易地获得常量的宏，而这些宏会让你做 mov $SEEK_SET, %edx 或 mov $O_WRONLY|O_CREAT|O_TRUNC, %esi 。

---

取消文件链接不会影响已打开文件的内容；要获得您在问题中所描绘的效果，您可以关闭/重新打开文件。(在 Unix 中，删除文件的目录条目不会影响已经打开它的程序。它会不过，一旦最后一个目录条目和它的文件描述符消失，就会从磁盘中释放。)

所以你会打开它进行读取，读取所有数据，然后(在检查错误之后，当你确定你有一些有效数据要写入时)，open(infile, O_CREAT|O_TRUNC|O_WRONLY, 0666) 并写出数据。你没有使用 O_APPEND ，所以新的只写 fd 的位置将在文件的前面。并且文件大小将被截断为 0。与 shell 中的 echo foo > filename 完全一样。

(它仍将具有相同的 inode 编号，并且是具有不同内容的“同一文件”，除非您在打开它之前使用 unlink(infile) 重新创建一个具有该名称的新文件。在这种情况下，O_CREAT 实际上是必要的。当重新打开一个要写入的现有文件 + 截断，当文件已经存在时不需要 O_CREAT。)

这里的关键是在做任何破坏性的事情之前检查读取错误，而不是仅仅读取它，破坏原件，然后继续。因此，在您排序时文件仍在磁盘上。