102 lines · 2.3 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0-only/* * rtmutex API */#include <linux/spinlock.h>#include <linux/export.h> #define RT_MUTEX_BUILD_MUTEX#define WW_RT#include "rtmutex.c" int ww_mutex_trylock(struct ww_mutex *lock, struct ww_acquire_ctx *ww_ctx){	struct rt_mutex *rtm = &lock->base; 	if (!ww_ctx)		return rt_mutex_trylock(rtm); 	/*	 * Reset the wounded flag after a kill. No other process can	 * race and wound us here, since they can't have a valid owner	 * pointer if we don't have any locks held.	 */	if (ww_ctx->acquired == 0)		ww_ctx->wounded = 0; 	if (__rt_mutex_trylock(&rtm->rtmutex)) {		ww_mutex_set_context_fastpath(lock, ww_ctx);		mutex_acquire_nest(&rtm->dep_map, 0, 1, &ww_ctx->dep_map, _RET_IP_);		return 1;	} 	return 0;}EXPORT_SYMBOL(ww_mutex_trylock); static int __sched__ww_rt_mutex_lock(struct ww_mutex *lock, struct ww_acquire_ctx *ww_ctx,		   unsigned int state, unsigned long ip){	struct lockdep_map __maybe_unused *nest_lock = NULL;	struct rt_mutex *rtm = &lock->base;	int ret; 	might_sleep(); 	if (ww_ctx) {		if (unlikely(ww_ctx == READ_ONCE(lock->ctx)))			return -EALREADY; 		/*		 * Reset the wounded flag after a kill. No other process can		 * race and wound us here, since they can't have a valid owner		 * pointer if we don't have any locks held.		 */		if (ww_ctx->acquired == 0)			ww_ctx->wounded = 0; #ifdef CONFIG_DEBUG_LOCK_ALLOC		nest_lock = &ww_ctx->dep_map;#endif	}	mutex_acquire_nest(&rtm->dep_map, 0, 0, nest_lock, ip); 	if (likely(rt_mutex_try_acquire(&rtm->rtmutex))) {		if (ww_ctx)			ww_mutex_set_context_fastpath(lock, ww_ctx);		return 0;	} 	ret = rt_mutex_slowlock(&rtm->rtmutex, ww_ctx, state); 	if (ret)		mutex_release(&rtm->dep_map, ip);	return ret;} int __schedww_mutex_lock(struct ww_mutex *lock, struct ww_acquire_ctx *ctx){	return __ww_rt_mutex_lock(lock, ctx, TASK_UNINTERRUPTIBLE, _RET_IP_);}EXPORT_SYMBOL(ww_mutex_lock); int __schedww_mutex_lock_interruptible(struct ww_mutex *lock, struct ww_acquire_ctx *ctx){	return __ww_rt_mutex_lock(lock, ctx, TASK_INTERRUPTIBLE, _RET_IP_);}EXPORT_SYMBOL(ww_mutex_lock_interruptible); void __sched ww_mutex_unlock(struct ww_mutex *lock)	__no_context_analysis{	struct rt_mutex *rtm = &lock->base; 	__ww_mutex_unlock(lock); 	mutex_release(&rtm->dep_map, _RET_IP_);	__rt_mutex_unlock(&rtm->rtmutex);}EXPORT_SYMBOL(ww_mutex_unlock);