linux系统编程：IO读写过程的原子性操作实验

所谓原子性操作指的是：内核保证某系统调用中的所有步骤（操作）作为独立操作而一次性加以执行，其间不会被其他进程或线程所中断。

举个通俗点的例子：你和女朋友OOXX的时候，突然来了个电话，势必会打断你们高潮的兴致，最好的办法就是，你们做这事的时候，把通讯设备关机，就能确保，这次的事情很圆满的完成，这就是一次原子性操作。

在多进程IO过程中，如果操作不具有原子性，就可能会导致数据混乱，相互覆盖等情况。这种现象也叫竞争状态。

所谓竞争状态指的是：操作共享资源的两个进程（或线程），其结果取决于一个无法预期的顺序，因为进程获取的cpu执行时间是不确定的。

1,假想的，以独占方式创建一个文件

下面这段代码，用open和O_CREAT标志演示一个独占方式创建文件， 什么叫独占方式创建文件？  就是该进程始终认为这个文件是他打开的，或者是他创建的

 /*================================================================
 *   Copyright (C) 2018 . All rights reserved.
 *
 *   文件名称：bad_exclusive_open.c
 *   创 建 者：ghostwu(吴华)
 *   创建日期：2018年01月11日
 *   描    述：
 *
 ================================================================*/

 #include <stdio.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/types.h>
 #include <unistd.h>
 #include <errno.h>

 int main(int argc, char *argv[])
 {
     if( argc <  || strcmp( argv[], "--help" ) ==  ){
         printf( "usage:%s filename\n", argv[] );
         exit( - );
     }

     printf( "pid=%d, %s文件不存在\n", getpid(), argv[] );

     int fd = -;

     fd = open( argv[], O_WRONLY );
     if( fd <  ){
         sleep(  );
         printf( "pid=%d, 结束睡眠\n", getpid() );
         //其他错误原因，导致文件打开失败
         if( errno != ENOENT ) {
             perror( "open" );
             exit( - );
         }else {
             //文件不存在　导致文件打开失败
             fd = open( argv[], O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR );
             if( fd <  ) {
                 printf( "文件%s创建失败\n", argv[] );
                 exit( - );
             }
             printf( "文件%s,创建并打开成功:fd=%d\n", argv[], fd );
             printf( "进程id=%d\n", getpid() );
             close( fd );
         }
     }else {
         printf( "文件%s,打开成功:fd=%d\n", argv[], fd );
         printf( "进程id=%d\n", getpid() );
         close( fd );
     }

     return ;
 }

假如，我们要创建一个不存在的test.txt文件。

为了演示方便，在程序第一次判断文件不存在的情况下，让进程挂起( sleep 5 )交出cpu的执行时间，这个时候，我们可以这样测试，两种方法：

1，在另一个终端，登录另一个账户(如root账户)，创建test.txt文件

2，在另一个终端，再开启一个进程

方法一：用shell脚本创建一个test.txt，并赋予其他组的权限为rw

createfile.sh

 #!/bin/bash
 #创建文件,并改变权限配合测试

 touch test.txt
 sudo chmod a+rw test.txt

实验结果：左边的进程依然认为这个文件是他创建并打开的!

方法二，在另一个终端，再开一个进程测试

两个进程都认为，test.txt是他们自己创建并打开的

2，如何保证独占方式创建一个文件?

非常简单，只需要把加一个标志O_EXCL，结合O_CREAT

fd = open( argv[], O_WRONLY | O_CREAT | O_EXCL, S_IRUSR | S_IWUSR );

再次按照上面的2种方式测试，得到的结果就是：

如果在sleep期间，别的进程创建了文件，那么该进程会报错

3，seek与write结合，产生相互覆盖

 /*================================================================
 *   Copyright (C) 2018 . All rights reserved.
 *
 *   文件名称：seek_file.c
 *   创 建 者：ghostwu(吴华)
 *   创建日期：2018年01月11日
 *   描    述：
 *
 ================================================================*/

 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <sys/types.h>
 #include <unistd.h>

 #ifndef BUFSIZE
 #define BUFSIZE 50
 #endif

 int main(int argc, char *argv[])
 {
     if( argc <  || strcmp( argv[], "--help" ) ==  ) {
         printf( "usage:%s filename w<string>\n", argv[] );
         exit( - );
     }

     if( argv[][] != 'w' ) {
         printf( "必须以w开头\n" );
         exit( - );
     }

     int fd = -;
     fd = open( argv[], O_RDWR );

     if( fd <  ) {
         printf( "文件%s打开失败\n", argv[] );
         exit( - );
     }

     if ( - == lseek( fd, , SEEK_END ) ) {
         printf( "指针移动到尾部失败\n" );
         exit( - );
     }

     sleep(  );

     char buf[BUFSIZE];
     ssize_t nwrite;

     strcpy( buf, &argv[][] );

     nwrite = write( fd, buf, strlen( buf ) );
     if( - == nwrite ) {
         printf( "文件写入失败\n" );
         exit( - );
     }
     printf( "pid=%d,向文件%s写入了%ld个字节\n", getpid(), argv[], nwrite );

     return ;
 }

如果第一个进程执行到seek与write之间，交出 cpu, 被执行相同代码的第二个进程中断，那么这两个进程在写入数据前都把指针移动到相同的位置，如果一个进程先完成，那么后一个进程会覆盖前面进程写入的数据

试验结果：

第二个进程后结束： 第一个进程写入的123被第二个进程的4567覆盖，产生结果 4567

第一个进程后结束：第一个进程写入的4567被第二个进程的123覆盖，产生结果 1237

如何避免数据覆盖？打开文件时候，加入O_APPEND标志

fd = open( argv[], O_RDWR | O_APPEND );

总结：

1）理解原子性操作

2）理解标志O_CREAT与O_EXCL结合的意义

3）理解O_APPEND标志

4）理解竞争状态