pcpu_ref: add percpu_ref_tryget_many() (4e5ef023) · Commits · 戴 / test

include/linux/percpu-refcount.h

+21 −5

Original line number	Diff line number	Diff line
		@@ -210,15 +210,17 @@ static inline void percpu_ref_get(struct percpu_ref *ref)
		}

		/**
		* percpu_ref_tryget - try to increment a percpu refcount
		* percpu_ref_tryget_many - try to increment a percpu refcount
		* @ref: percpu_ref to try-get
		* @nr: number of references to get
		*
		* Increment a percpu refcount unless its count already reached zero.
		* Increment a percpu refcount by @nr unless its count already reached zero.
		* Returns %true on success; %false on failure.
		*
		* This function is safe to call as long as @ref is between init and exit.
		*/
		static inline bool percpu_ref_tryget(struct percpu_ref *ref)
		static inline bool percpu_ref_tryget_many(struct percpu_ref *ref,
		unsigned long nr)
		{
		unsigned long __percpu *percpu_count;
		bool ret;
		@@ -226,10 +228,10 @@ static inline bool percpu_ref_tryget(struct percpu_ref *ref)
		rcu_read_lock();

		if (__ref_is_percpu(ref, &percpu_count)) {
		this_cpu_inc(*percpu_count);
		this_cpu_add(*percpu_count, nr);
		ret = true;
		} else {
		ret = atomic_long_inc_not_zero(&ref->count);
		ret = atomic_long_add_unless(&ref->count, nr, 0);
		}

		rcu_read_unlock();
		@@ -237,6 +239,20 @@ static inline bool percpu_ref_tryget(struct percpu_ref *ref)
		return ret;
		}

		/**
		* percpu_ref_tryget - try to increment a percpu refcount
		* @ref: percpu_ref to try-get
		*
		* Increment a percpu refcount unless its count already reached zero.
		* Returns %true on success; %false on failure.
		*
		* This function is safe to call as long as @ref is between init and exit.
		*/
		static inline bool percpu_ref_tryget(struct percpu_ref *ref)
		{
		return percpu_ref_tryget_many(ref, 1);
		}

		/**
		* percpu_ref_tryget_live - try to increment a live percpu refcount
		* @ref: percpu_ref to try-get

Admin message