aboutsummaryrefslogtreecommitdiffstats
path: root/include/asterisk/lock.h
blob: bde1e6adaa81dc3a7ba60af78004dad222ee5257 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
 * Asterisk -- A telephony toolkit for Linux.
 *
 * General Asterisk channel definitions.
 * 
 * Copyright (C) 1999, Mark Spencer
 *
 * Mark Spencer <markster@linux-support.net>
 *
 * This program is free software, distributed under the terms of
 * the GNU General Public License
 */

#ifndef _ASTERISK_LOCK_H
#define _ASTERISK_LOCK_H

#include <pthread.h>

#ifdef DEBUG_THREADS

#define TRIES 50

#include <errno.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>

// #define AST_MUTEX_INITIALIZER      PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP
// #define AST_MUTEX_KIND             PTHREAD_MUTEX_RECURSIVE_NP
#ifdef PTHREAD_ERRORCHECK_MUTEX_INITIALIZER_NP
#define AST_MUTEX_INITIALIZER         PTHREAD_ERRORCHECK_MUTEX_INITIALIZER_NP
#define AST_MUTEX_KIND                PTHREAD_MUTEX_ERRORCHECK_NP
#else
#define AST_MUTEX_INITIALIZER      PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP
#define AST_MUTEX_KIND             PTHREAD_MUTEX_RECURSIVE_NP
#endif

struct mutex_info {
	pthread_mutex_t *mutex;
	char *file;
	int lineno;
	char *func;
	struct mutex_info *next;
};

static inline int ast_pthread_mutex_init(pthread_mutex_t *t) {
	static pthread_mutexattr_t  attr;
	static int  init = 1;
	extern int  pthread_mutexattr_setkind_np(pthread_mutexattr_t *, int);

	if (init) {
		pthread_mutexattr_init(&attr);
		pthread_mutexattr_setkind_np(&attr, AST_MUTEX_KIND);
		init = 0;
	}
	return pthread_mutex_init(t, &attr);
}

static inline int __ast_pthread_mutex_lock(char *filename, int lineno, char *func, pthread_mutex_t *t) {
	int res;
	int tries = TRIES;
	do {
		res = pthread_mutex_trylock(t);
		/* If we can't run, yield */
		if (res) {
			sched_yield();
			usleep(1);
		}
	} while(res && tries--);
	if (res) {
		fprintf(stderr, "%s line %d (%s): Error obtaining mutex: %s\n", 
				filename, lineno, func, strerror(res));
		if ((res = pthread_mutex_lock(t)))
                        fprintf(stderr, "%s line %d (%s): Error waiting for mutex: %s\n", 
                               filename, lineno, func, strerror(res));
                else
		        fprintf(stderr, "%s line %d (%s): Got it eventually...\n",
			       filename, lineno, func);
	}
	return res;
}

#define ast_pthread_mutex_lock(a) __ast_pthread_mutex_lock(__FILE__, __LINE__, __PRETTY_FUNCTION__, a)

static inline int __ast_pthread_mutex_unlock(char *filename, int lineno, char *func, pthread_mutex_t *t) {
	int res;
	res = pthread_mutex_unlock(t);
	if (res) 
		fprintf(stderr, "%s line %d (%s): Error releasing mutex: %s\n", 
				filename, lineno, func, strerror(res));
	return res;
}

#define ast_pthread_mutex_unlock(a) __ast_pthread_mutex_unlock(__FILE__, __LINE__, __PRETTY_FUNCTION__, a)

#else

#define AST_MUTEX_INITIALIZER      PTHREAD_MUTEX_INITIALIZER
#define AST_MUTEX_KIND             PTHREAD_MUTEX_FAST_NP

#define ast_pthread_mutex_init(mutex) pthread_mutex_init(mutex, NULL)
#define ast_pthread_mutex_lock pthread_mutex_lock
#define ast_pthread_mutex_unlock pthread_mutex_unlock

#endif

#endif