linux信号锁的封装与实现

信号量的值与相应资源的使用情况有关，当它的值大于 0 时，表示当前可用的资源数的数量；当它的值小于 0 时，其绝对值表示等待使用该资源的进程个数。信号量的值仅能由 PV 操作来改变。

     在 Linux 下，PV 操作通过调用semop函数来实现。该函数定义在头文件 sys/sem.h中，原型如下：

     int  semop（int  semid，struct sembuf  *sops，size_t nsops）；

     函数的参数 semid 为信号量集的标识符；参数 sops 指向进行操作的结构体数组的首地址；参数 nsops 指出将要进行操作的信号的个数。semop 函数调用成功返回 0，失败返回 -1。

     semop 的第二个参数 sops 指向的结构体数组中，每个 sembuf 结构体对应一个特定信号的操作。因此对信号量进行操作必须熟悉该数据结构，该结构定义在 linux/sem.h，如下所示：

     struct  sembuf{

         unsigned short   sem_num;      //信号在信号集中的索引，0代表第一个信号，1代表第二个信号  

         short            sem_op;      //操作类型

         short            sem_flg;    //操作标志

     };

    下面详细介绍一下 sembuf 的几个参数：

--------------------------------------------------------------------------------------------------

• sem_op 参数：

                    sem_op > 0          信号加上 sem_op 的值，表示进程释放控制的资源；

                    sem_op = 0          如果没有设置 IPC_NOWAIT，则调用进程进入睡眠状态，直到信号                                         量的值为0；否则进程不回睡眠，直接返回 EAGAIN

                    sem_op < 0          信号加上 sem_op 的值。也就是需要几个信号值，若没有设置 IPC_NOWAIT ，则调用进程阻

                                        塞，直到资源可用；否则进程直接返回EAGAIN

• sem_flg 参数：

         该参数可设置为 IPC_NOWAIT 或 SEM_UNDO 两种状态。只有将 sem_flg 指定为 SEM_UNDO 标志后，semadj （所指定信号量针对调用进程的调整值）才会更新。   此外，如果此操作指定SEM_UNDO，系统更新过程中会撤消此信号灯的计数（semadj）。此操作可以随时进行---它永远不会强制等待的过程。调用进程必须有改变信号量集的权限。

         sem_flg公认的标志是 IPC_NOWAIT 和 SEM_UNDO。如果操作指定SEM_UNDO，它将会自动撤消该进程终止时。

利用信号量实现信号锁：

[cpp] view plain copy print ?

1. class TC_SemMutex  
2. {  
3. public:  
4.     /** 
5.      * 构造函数 
6.      */  
7.     TC_SemMutex();  
8.   
9.     /** 
10.     * 构造函数 
11.     * @param iKey, key 
12.     * @throws TC_SemMutex_Exception 
13.     */  
14.     TC_SemMutex(key_t iKey);  
15.   
16.     /** 
17.     * 初始化 
18.     * @param iKey, key 
19.     * @throws TC_SemMutex_Exception 
20.     * @return 无 
21.      */  
22.     void init(key_t iKey);  
23.   
24.     /** 
25.     * 获取共享内存Key 
26.     * @return key_t ,共享内存key 
27.     */  
28.     key_t getkey() const {return _semKey;}  
29.   
30.     /** 
31.     * 获取共享内存ID 
32.     * @return int ,共享内存Id 
33.     */  
34.     int getid() const   {return _semID;}  
35.   
36.     /** 
37.     * 加读锁 
38.     *@return int 
39.     */  
40.     int rlock() const;  
41.   
42.     /** 
43.     * 解读锁 
44.     * @return int 
45.     */  
46.     int unrlock() const;  
47.   
48.     /** 
49.     * 尝试读锁 
50.     * @return bool : 加锁成功则返回false, 否则返回false 
51.     */  
52.     bool tryrlock() const;  
53.   
54.     /** 
55.     * 加写锁 
56.     * @return int 
57.     */  
58.     int wlock() const;  
59.   
60.     /** 
61.     * 解写锁 
62.     */  
63.     int unwlock() const;  
64.   
65.     /** 
66.     * 尝试写锁 
67.     * @throws TC_SemMutex_Exception 
68.     * @return bool : 加锁成功则返回false, 否则返回false 
69.     */  
70.     bool trywlock() const;  
71.   
72.     /** 
73.     * 写锁 
74.     * @return int, 0 正确 
75.     */  
76.     int lock() const        {return wlock();};  
77.   
78.     /** 
79.     * 解写锁 
80.     */  
81.     int unlock() const      {return unwlock();};  
82.   
83.     /** 
84.     * 尝试解锁 
85.     * @throws TC_SemMutex_Exception 
86.     * @return int, 0 正确 
87.     */  
88.     bool trylock() const    {return trywlock();};  
89.   
90. protected:  
91.     /** 
92.      * 信号量ID 
93.      */  
94.     int _semID;  
95.   
96.     /** 
97.      * 信号量key 
98.      */  
99.     key_t _semKey;  
100. };  

class TC_SemMutex
{
public:
    /**
     * 构造函数
     */
    TC_SemMutex();

    /**
    * 构造函数
    * @param iKey, key
    * @throws TC_SemMutex_Exception
    */
    TC_SemMutex(key_t iKey);

    /**
    * 初始化
    * @param iKey, key
    * @throws TC_SemMutex_Exception
    * @return 无
     */
    void init(key_t iKey);

    /**
    * 获取共享内存Key
    * @return key_t ,共享内存key
    */
    key_t getkey() const {return _semKey;}

    /**
    * 获取共享内存ID
    * @return int ,共享内存Id
    */
    int getid() const   {return _semID;}

    /**
    * 加读锁
    *@return int
    */
    int rlock() const;

    /**
    * 解读锁
    * @return int
    */
    int unrlock() const;

    /**
    * 尝试读锁
    * @return bool : 加锁成功则返回false, 否则返回false
    */
    bool tryrlock() const;

    /**
    * 加写锁
    * @return int
    */
    int wlock() const;

    /**
    * 解写锁
    */
    int unwlock() const;

    /**
    * 尝试写锁
    * @throws TC_SemMutex_Exception
    * @return bool : 加锁成功则返回false, 否则返回false
    */
    bool trywlock() const;

    /**
    * 写锁
    * @return int, 0 正确
    */
    int lock() const        {return wlock();};

    /**
    * 解写锁
    */
    int unlock() const      {return unwlock();};

    /**
    * 尝试解锁
    * @throws TC_SemMutex_Exception
    * @return int, 0 正确
    */
    bool trylock() const    {return trywlock();};

protected:
    /**
     * 信号量ID
     */
    int _semID;

    /**
     * 信号量key
     */
    key_t _semKey;
};

[cpp] view plain copy print ?

1. TC_SemMutex::TC_SemMutex()  
2. {  
3.   
4. }  
5.   
6. TC_SemMutex::TC_SemMutex(key_t iKey)  
7. {  
8.     init(iKey);  
9. }  
10.   
11. void TC_SemMutex::init(key_t iKey)  
12. {  
13.     #if defined(__GNU_LIBRARY__) && !defined(_SEM_SEMUN_UNDEFINED)   
14.     /* union semun is defined by including <sys/sem.h> */  
15.     #else   
16.     /* according to X/OPEN we have to define it ourselves */  
17.     union semun  
18.     {  
19.          int val;                  /* value for SETVAL */  
20.          struct semid_ds *buf;     /* buffer for IPC_STAT, IPC_SET */  
21.          unsigned short *array;    /* array for GETALL, SETALL */  
22.                                    /* Linux specific part: */  
23.          struct seminfo *__buf;    /* buffer for IPC_INFO */  
24.     };  
25.     #endif   
26.   
27.     int  iSemID;  
28.     union semun arg;  
29.     u_short array[2] = { 0, 0 };  
30.   
31.     //生成信号量集, 包含两个信号量   
32.     if ( (iSemID = semget( iKey, 2, IPC_CREAT | IPC_EXCL | 0666)) != -1 )  
33.     {  
34.         arg.array = &array[0];  
35.   
36.         //将所有信号量的值设置为0   
37.         if ( semctl( iSemID, 0, SETALL, arg ) == -1 )  
38.         {  
39.             throw TC_SemMutex_Exception("[TC_SemMutex::init] semctl error:" + string(strerror(errno)));  
40.         }  
41.     }  
42.     else  
43.     {  
44.         //信号量已经存在   
45.         if ( errno != EEXIST )  
46.         {  
47.             throw TC_SemMutex_Exception("[TC_SemMutex::init] sem has exist error:" + string(strerror(errno)));  
48.         }  
49.   
50.         //连接信号量   
51.         if ( (iSemID = semget( iKey, 2, 0666 )) == -1 )  
52.         {  
53.             throw TC_SemMutex_Exception("[TC_SemMutex::init] connect sem error:" + string(strerror(errno)));  
54.         }  
55.     }  
56.   
57.     _semKey = iKey;  
58.     _semID  = iSemID;  
59. }  
60.   
61. int TC_SemMutex::rlock() const  
62. {  
63.     //进入共享锁, 第二个信号量的值表示当前使用信号量的进程个数   
64.     //等待第一个信号量变为0(排他锁没有使用)   
65.     //占用第二个信号量(第二个信号量值+1, 表示被共享锁使用)   
66.     struct sembuf sops[2] = { {0, 0, SEM_UNDO}, {1, 1, SEM_UNDO} };  
67.     size_t nsops = 2;  
68.   
69.     return semop( _semID, &sops[0], nsops);  
70. }  
71.   
72. int TC_SemMutex::unrlock( ) const  
73. {  
74.     //解除共享锁, 有进程使用过第二个信号量   
75.     //等到第二个信号量可以使用(第二个信号量的值>=1)   
76.     struct sembuf sops[1] = { {1, -1, SEM_UNDO} };  
77.     size_t nsops = 1;  
78.   
79.     return semop( _semID, &sops[0], nsops);  
80. }  
81.   
82. bool TC_SemMutex::tryrlock() const  
83. {  
84.     struct sembuf sops[2] = { {0, 0, SEM_UNDO|IPC_NOWAIT}, {1, 1, SEM_UNDO|IPC_NOWAIT}};  
85.     size_t nsops = 2;  
86.   
87.     int iRet = semop( _semID, &sops[0], nsops );  
88.     if(iRet == -1)  
89.     {  
90.         if(errno == EAGAIN)  
91.         {  
92.             //无法获得锁   
93.             return false;  
94.         }  
95.         else  
96.         {  
97.             throw TC_SemMutex_Exception("[TC_SemMutex::tryrlock] semop error : " + string(strerror(errno)));  
98.         }  
99.     }  
100.     return true;  
101. }  
102.   
103. int TC_SemMutex::wlock() const  
104. {  
105.     //进入排他锁, 第一个信号量和第二个信号都没有被使用过(即, 两个锁都没有被使用)   
106.     //等待第一个信号量变为0   
107.     //等待第二个信号量变为0   
108.     //释放第一个信号量(第一个信号量+1, 表示有一个进程使用第一个信号量)   
109.     struct sembuf sops[3] = { {0, 0, SEM_UNDO}, {1, 0, SEM_UNDO}, {0, 1, SEM_UNDO} };  
110.     size_t nsops = 3;  
111.   
112.     return semop( _semID, &sops[0], nsops);  
113. }  
114.   
115. int TC_SemMutex::unwlock() const  
116. {  
117.     //解除排他锁, 有进程使用过第一个信号量   
118.     //等待第一个信号量(信号量值>=1)   
119.     struct sembuf sops[1] = { {0, -1, SEM_UNDO} };  
120.     size_t nsops = 1;  
121.   
122.     return semop( _semID, &sops[0], nsops);  
123.   
124. }  
125.   
126. bool TC_SemMutex::trywlock() const  
127. {  
128.     struct sembuf sops[3] = { {0, 0, SEM_UNDO|IPC_NOWAIT}, {1, 0, SEM_UNDO|IPC_NOWAIT}, {0, 1, SEM_UNDO|IPC_NOWAIT} };  
129.     size_t nsops = 3;  
130.   
131.     int iRet = semop( _semID, &sops[0], nsops );  
132.     if(iRet == -1)  
133.     {  
134.         if(errno == EAGAIN)  
135.         {  
136.             //无法获得锁   
137.             return false;  
138.         }  
139.         else  
140.         {  
141.             throw TC_SemMutex_Exception("[TC_SemMutex::trywlock] semop error : " + string(strerror(errno)));  
142.         }  
143.     }  
144.   
145.     return true;  
146. }  

TC_SemMutex::TC_SemMutex()
{

}

TC_SemMutex::TC_SemMutex(key_t iKey)
{
    init(iKey);
}

void TC_SemMutex::init(key_t iKey)
{
    #if defined(__GNU_LIBRARY__) && !defined(_SEM_SEMUN_UNDEFINED)
    /* union semun is defined by including <sys/sem.h> */
    #else
    /* according to X/OPEN we have to define it ourselves */
    union semun
    {
         int val;                  /* value for SETVAL */
         struct semid_ds *buf;     /* buffer for IPC_STAT, IPC_SET */
         unsigned short *array;    /* array for GETALL, SETALL */
                                   /* Linux specific part: */
         struct seminfo *__buf;    /* buffer for IPC_INFO */
    };
    #endif

    int  iSemID;
    union semun arg;
    u_short array[2] = { 0, 0 };

    //生成信号量集, 包含两个信号量
    if ( (iSemID = semget( iKey, 2, IPC_CREAT | IPC_EXCL | 0666)) != -1 )
    {
        arg.array = &array[0];

        //将所有信号量的值设置为0
        if ( semctl( iSemID, 0, SETALL, arg ) == -1 )
        {
            throw TC_SemMutex_Exception("[TC_SemMutex::init] semctl error:" + string(strerror(errno)));
        }
    }
    else
    {
        //信号量已经存在
        if ( errno != EEXIST )
        {
            throw TC_SemMutex_Exception("[TC_SemMutex::init] sem has exist error:" + string(strerror(errno)));
        }

        //连接信号量
        if ( (iSemID = semget( iKey, 2, 0666 )) == -1 )
        {
            throw TC_SemMutex_Exception("[TC_SemMutex::init] connect sem error:" + string(strerror(errno)));
        }
    }

    _semKey = iKey;
    _semID  = iSemID;
}

int TC_SemMutex::rlock() const
{
    //进入共享锁, 第二个信号量的值表示当前使用信号量的进程个数
    //等待第一个信号量变为0(排他锁没有使用)
    //占用第二个信号量(第二个信号量值+1, 表示被共享锁使用)
    struct sembuf sops[2] = { {0, 0, SEM_UNDO}, {1, 1, SEM_UNDO} };
    size_t nsops = 2;

    return semop( _semID, &sops[0], nsops);
}

int TC_SemMutex::unrlock( ) const
{
    //解除共享锁, 有进程使用过第二个信号量
    //等到第二个信号量可以使用(第二个信号量的值>=1)
    struct sembuf sops[1] = { {1, -1, SEM_UNDO} };
    size_t nsops = 1;

    return semop( _semID, &sops[0], nsops);
}

bool TC_SemMutex::tryrlock() const
{
    struct sembuf sops[2] = { {0, 0, SEM_UNDO|IPC_NOWAIT}, {1, 1, SEM_UNDO|IPC_NOWAIT}};
    size_t nsops = 2;

    int iRet = semop( _semID, &sops[0], nsops );
    if(iRet == -1)
    {
        if(errno == EAGAIN)
        {
            //无法获得锁
            return false;
        }
        else
        {
            throw TC_SemMutex_Exception("[TC_SemMutex::tryrlock] semop error : " + string(strerror(errno)));
        }
    }
    return true;
}

int TC_SemMutex::wlock() const
{
    //进入排他锁, 第一个信号量和第二个信号都没有被使用过(即, 两个锁都没有被使用)
    //等待第一个信号量变为0
    //等待第二个信号量变为0
    //释放第一个信号量(第一个信号量+1, 表示有一个进程使用第一个信号量)
    struct sembuf sops[3] = { {0, 0, SEM_UNDO}, {1, 0, SEM_UNDO}, {0, 1, SEM_UNDO} };
    size_t nsops = 3;

    return semop( _semID, &sops[0], nsops);
}

int TC_SemMutex::unwlock() const
{
    //解除排他锁, 有进程使用过第一个信号量
    //等待第一个信号量(信号量值>=1)
    struct sembuf sops[1] = { {0, -1, SEM_UNDO} };
    size_t nsops = 1;

    return semop( _semID, &sops[0], nsops);

}

bool TC_SemMutex::trywlock() const
{
    struct sembuf sops[3] = { {0, 0, SEM_UNDO|IPC_NOWAIT}, {1, 0, SEM_UNDO|IPC_NOWAIT}, {0, 1, SEM_UNDO|IPC_NOWAIT} };
    size_t nsops = 3;

    int iRet = semop( _semID, &sops[0], nsops );
    if(iRet == -1)
    {
        if(errno == EAGAIN)
        {
            //无法获得锁
            return false;
        }
        else
        {
            throw TC_SemMutex_Exception("[TC_SemMutex::trywlock] semop error : " + string(strerror(errno)));
        }
    }

    return true;
}

在这个锁实现中，涉及到两个信号量：

信号量1：表示排他锁的信号数目。每次read lock的时候，会要求该信号量为0，每次write lock的时候会要求该信号量为0，并且新增一次，每次write unlock都会减去1。

信号量2：表示共享锁的进程数目。每次read lock会增加1，每次read unlock会减去1，同时，write lock会要求该信号量为0。

因为opt数组之间都是原子操作，保证SEM_UNDO，就可以保证一致性和回滚。这是该信号锁的实现原理。