Switch to unified view

a b/src/ptmutex.hxx
1
/* Copyright (C) 2011 J.F.Dockes
2
 *   This program is free software; you can redistribute it and/or modify
3
 *   it under the terms of the GNU General Public License as published by
4
 *   the Free Software Foundation; either version 2 of the License, or
5
 *   (at your option) any later version.
6
 *
7
 *   This program is distributed in the hope that it will be useful,
8
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
9
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10
 *   GNU General Public License for more details.
11
 *
12
 *   You should have received a copy of the GNU General Public License
13
 *   along with this program; if not, write to the
14
 *   Free Software Foundation, Inc.,
15
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
16
 */
17
#ifndef _PTMUTEX_H_INCLUDED_
18
#define _PTMUTEX_H_INCLUDED_
19
20
#include <pthread.h>
21
22
/// A trivial wrapper/helper for pthread mutex locks
23
24
namespace UPnPP {
25
26
/// Lock storage with auto-initialization. Must be created before any
27
/// lock-using thread of course (possibly as a static object).
28
class PTMutexInit {
29
public:
30
    pthread_mutex_t m_mutex;
31
    int m_status;
32
    PTMutexInit() {
33
        m_status = pthread_mutex_init(&m_mutex, 0);
34
    }
35
};
36
37
/// Take the lock when constructed, release when deleted. Can be disabled
38
/// by constructor params for conditional use.
39
class PTMutexLocker {
40
public:
41
    // The nolock arg enables conditional locking
42
    PTMutexLocker(PTMutexInit& l, bool nolock = false)
43
        : m_lock(l), m_status(-1) {
44
        if (!nolock) {
45
            m_status = pthread_mutex_lock(&m_lock.m_mutex);
46
        }
47
    }
48
    ~PTMutexLocker() {
49
        if (m_status == 0) {
50
            pthread_mutex_unlock(&m_lock.m_mutex);
51
        }
52
    }
53
    int ok() {
54
        return m_status == 0;
55
    }
56
    // For pthread_cond_wait etc.
57
    pthread_mutex_t *getMutex() {
58
        return &m_lock.m_mutex;
59
    }
60
private:
61
    PTMutexInit& m_lock;
62
    int m_status;
63
};
64
65
} // namespace UPnPP
66
67
#endif /* _PTMUTEX_H_INCLUDED_ */