blob: 8883eb18fd8af38f6746ee424f7543cbaa77ea44 (
plain) (
blame)
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
|
#ifndef _REFERENCE_H_
#define _REFERENCE_H_
#include "reftypes.h"
#include <stddef.h>
#include <stdint.h>
#define container_of(ptr, type, member) \
((type *)((char *)(ptr) - (char *)&(((type *)NULL)->member)))
void ref_init( ref *reference, void ( *freefun )( ref * ), long count );
void ref_setref( weakref *weakref, ref *ref );
_Noreturn void _ref_error( const char *message );
static inline ref *ref_get( weakref *weakref )
{
char *old_weakref = (char *)*weakref;
do {
if ( old_weakref == NULL )
return NULL;
if ( aligned_ref( old_weakref ) != aligned_ref( old_weakref + 1 ) ) {
old_weakref = (char *)*weakref;
continue;
}
} while ( !atomic_compare_exchange_weak( weakref, (void **)&old_weakref, old_weakref + 1 ) );
struct _ref_ *ref = aligned_ref( old_weakref )->ref;
if ( unlikely( ++ref->count == -1 ) ) {
_ref_error( "Reference counter overflow. Aborting." );
}
char *cur_weakref = ( char * )*weakref;
do {
if ( aligned_ref( cur_weakref ) != aligned_ref( old_weakref ) ) {
ref->count--;
break;
}
} while ( !atomic_compare_exchange_weak( weakref, (void **)&cur_weakref, cur_weakref - 1 ) );
return ref;
}
static inline void ref_put( ref *ref )
{
if ( --ref->count == 0 ) {
ref->free( ref );
}
}
#define ref_get_uplink(wr) ({ \
ref* ref = ref_get( wr ); \
ref == NULL ? NULL : container_of(ref, dnbd3_uplink_t, reference); \
})
#endif
|
