Mercurial > hg > orthanc
annotate Core/MultiThreading/Semaphore.h @ 3809:2bf30ef727e3
enforcing SQLiteDatabaseWrapper::GetTableRecordCount() against SQL injection
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 03 Apr 2020 14:47:37 +0200 |
parents | 94f4a18a79cc |
children | 7176ebf08765 |
rev | line source |
---|---|
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
1 /** |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
2 * Orthanc - A Lightweight, RESTful DICOM Store |
1900 | 3 * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics |
1288
6e7e5ed91c2d
upgrade to year 2015
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
995
diff
changeset
|
4 * Department, University Hospital of Liege, Belgium |
3640
94f4a18a79cc
upgrade to year 2020
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
3483
diff
changeset
|
5 * Copyright (C) 2017-2020 Osimis S.A., Belgium |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
6 * |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
7 * This program is free software: you can redistribute it and/or |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
8 * modify it under the terms of the GNU General Public License as |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
9 * published by the Free Software Foundation, either version 3 of the |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
10 * License, or (at your option) any later version. |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
11 * |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
12 * In addition, as a special exception, the copyright holders of this |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
13 * program give permission to link the code of its release with the |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
14 * OpenSSL project's "OpenSSL" library (or with modified versions of it |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
15 * that use the same license as the "OpenSSL" library), and distribute |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
16 * the linked executables. You must obey the GNU General Public License |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
17 * in all respects for all of the code used other than "OpenSSL". If you |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
18 * modify file(s) with this exception, you may extend this exception to |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
19 * your version of the file(s), but you are not obligated to do so. If |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
20 * you do not wish to do so, delete this exception statement from your |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
21 * version. If you delete this exception statement from all source files |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
22 * in the program, then also delete it here. |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
23 * |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
24 * This program is distributed in the hope that it will be useful, but |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
25 * WITHOUT ANY WARRANTY; without even the implied warranty of |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
26 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
27 * General Public License for more details. |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
28 * |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
29 * You should have received a copy of the GNU General Public License |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
30 * along with this program. If not, see <http://www.gnu.org/licenses/>. |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
31 **/ |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
32 |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
33 |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
34 #pragma once |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
35 |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
36 #include <boost/noncopyable.hpp> |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
37 #include <boost/thread.hpp> |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
38 |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
39 namespace Orthanc |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
40 { |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
41 class Semaphore : public boost::noncopyable |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
42 { |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
43 private: |
3480
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
44 unsigned int availableResources_; |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
45 boost::mutex mutex_; |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
46 boost::condition_variable condition_; |
2794
f790999a250a
Semaphore removed from graveyard, SystemToolbox::GetHardwareConcurrency()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2611
diff
changeset
|
47 |
f790999a250a
Semaphore removed from graveyard, SystemToolbox::GetHardwareConcurrency()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2611
diff
changeset
|
48 void Release(); |
f790999a250a
Semaphore removed from graveyard, SystemToolbox::GetHardwareConcurrency()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2611
diff
changeset
|
49 |
f790999a250a
Semaphore removed from graveyard, SystemToolbox::GetHardwareConcurrency()
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
2611
diff
changeset
|
50 void Acquire(); |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
51 |
3480
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
52 bool TryAcquire(); |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
53 public: |
3480
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
54 explicit Semaphore(unsigned int availableResources); |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
55 |
3483
4d26261fd05a
get semaphore count for debug purpose
Alain Mazy <alain@mazy.be>
parents:
3480
diff
changeset
|
56 unsigned int GetAvailableResourcesCount() const |
4d26261fd05a
get semaphore count for debug purpose
Alain Mazy <alain@mazy.be>
parents:
3480
diff
changeset
|
57 { |
4d26261fd05a
get semaphore count for debug purpose
Alain Mazy <alain@mazy.be>
parents:
3480
diff
changeset
|
58 return availableResources_; |
4d26261fd05a
get semaphore count for debug purpose
Alain Mazy <alain@mazy.be>
parents:
3480
diff
changeset
|
59 } |
4d26261fd05a
get semaphore count for debug purpose
Alain Mazy <alain@mazy.be>
parents:
3480
diff
changeset
|
60 |
4d26261fd05a
get semaphore count for debug purpose
Alain Mazy <alain@mazy.be>
parents:
3480
diff
changeset
|
61 |
2108 | 62 class Locker : public boost::noncopyable |
63 { | |
64 private: | |
65 Semaphore& that_; | |
66 | |
67 public: | |
2223 | 68 explicit Locker(Semaphore& that) : |
2108 | 69 that_(that) |
70 { | |
71 that_.Acquire(); | |
72 } | |
73 | |
74 ~Locker() | |
75 { | |
76 that_.Release(); | |
77 } | |
78 }; | |
3480
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
79 |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
80 class TryLocker : public boost::noncopyable |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
81 { |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
82 private: |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
83 Semaphore& that_; |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
84 bool isAcquired_; |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
85 |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
86 public: |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
87 explicit TryLocker(Semaphore& that) : |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
88 that_(that) |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
89 { |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
90 isAcquired_ = that_.TryAcquire(); |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
91 } |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
92 |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
93 ~TryLocker() |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
94 { |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
95 if (isAcquired_) |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
96 { |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
97 that_.Release(); |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
98 } |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
99 } |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
100 |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
101 bool IsAcquired() const |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
102 { |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
103 return isAcquired_; |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
104 } |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
105 }; |
10fd1b9ae044
fix Semaphore + added TryAcquire & TryLocker
Alain Mazy <alain@mazy.be>
parents:
3060
diff
changeset
|
106 |
995
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
107 }; |
8c67382f44a7
limit number of jobs in the scheduler
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
108 } |