zoukankan      html  css  js  c++  java
  • 一个多线程的隐患

        今天查看项目代码的时候发现下面这段代码:

    代码
    1 internal MoveTask Dequeue(string host)
    2 {
    3 if (host == null)
    4 throw new ArgumentNullException("host");
    5
    6 lock (m_QueueLock)
    7 {
    8 if (!m_MoveTaskQueueDictionary.ContainsKey(host) || (m_MoveTaskQueueDictionary[host].Count == 0))
    9 return null;
    10 }
    11
    12 lock (m_QueueLock)
    13 {
    14 return m_MoveTaskQueueDictionary[host].Dequeue();
    15 }
    16 }

         当时这段代码实现时分了两个Lock为的就是对锁的晚占用,早释放。没想到却带来了多线程环境下可能发生的隐患:某个时刻某个Host的队列长度刚好等于1,这时线程A拥有了锁,那么在第一个Lock检查时该Host队列长度大于0,退出锁。此时另外一个线程B在A退出锁之后也进入了锁,也在第一个Lock检查时发现同一个Host的队列长度大于0,然后退出锁,准备进入第二个Lock。这时A线程由于B线程退出了锁,从而得以进入第二个Lock,对队列执行Dequeue方法后安全退出了锁。然后B线程进入锁执行Dequeue方法,这时队列为空,从而引发InvalidOperationException。

         解决办法很简单,合并这两个Lock为一个即可:

    代码
    1 internal MoveTask Dequeue(string host)
    2 {
    3 if (host == null)
    4 throw new ArgumentNullException("host");
    5
    6 lock (m_QueueLock)
    7 {
    8 if (!m_MoveTaskQueueDictionary.ContainsKey(host) || (m_MoveTaskQueueDictionary[host].Count == 0))
    9 return null;
    10
    11 return m_MoveTaskQueueDictionary[host].Dequeue();
    12 }
    13 }
  • 相关阅读:
    任意代码执行实例
    任意代码执行漏洞
    Port Hacking
    任意命令执行漏洞
    sql注入基于错误-单引号-字符型
    与SQL注入第一次相遇
    Hibernate基础知识
    TxQueryRunner类对结果集封装成bean、map及object的操作
    map的数据创建任意类的JavaBean对象
    UUID产生的JAR
  • 原文地址:https://www.cnblogs.com/wenhx/p/Multiple.html
Copyright © 2011-2022 走看看