1 /*
2  * Copyright (c) 2021-2022 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *    http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #include "hks_rwlock.h"
17 
18 #include <pthread.h>
19 
20 #include "hks_mem.h"
21 #include "hks_template.h"
22 
23 struct HksRwlock {
24     pthread_rwlock_t lock;
25 };
26 
HksRwlockCreate(void)27 HksRwlock *HksRwlockCreate(void)
28 {
29     HksRwlock *rwLock = (HksRwlock *)HksMalloc(sizeof(HksRwlock));
30     if (rwLock != NULL) {
31         int result = pthread_rwlock_init(&rwLock->lock, NULL);
32         if (result != 0) {
33             HKS_FREE(rwLock);
34             rwLock = NULL;
35         }
36     }
37     return rwLock;
38 }
39 
HksRwlockLockRead(HksRwlock * lock)40 int32_t HksRwlockLockRead(HksRwlock *lock)
41 {
42     HKS_IF_NULL_RETURN(lock, -1)
43 
44     return pthread_rwlock_rdlock(&lock->lock);
45 }
46 
HksRwlockUnlockRead(HksRwlock * lock)47 int32_t HksRwlockUnlockRead(HksRwlock *lock)
48 {
49     HKS_IF_NULL_RETURN(lock, -1)
50 
51     return pthread_rwlock_unlock(&lock->lock);
52 }
53 
HksRwlockLockWrite(HksRwlock * lock)54 int32_t HksRwlockLockWrite(HksRwlock *lock)
55 {
56     HKS_IF_NULL_RETURN(lock, -1)
57 
58     return pthread_rwlock_wrlock(&lock->lock);
59 }
60 
HksRwlockUnlockWrite(HksRwlock * lock)61 int32_t HksRwlockUnlockWrite(HksRwlock *lock)
62 {
63     HKS_IF_NULL_RETURN(lock, -1)
64 
65     return pthread_rwlock_unlock(&lock->lock);
66 }
67 
HksRwlockClose(HksRwlock * lock)68 void HksRwlockClose(HksRwlock *lock)
69 {
70     if (lock == NULL) {
71         return;
72     }
73 
74     pthread_rwlock_destroy(&lock->lock);
75     HKS_FREE(lock);
76 }
77