powerpc/book3s64/pkeys: Simplify pkey disable branch

#include <linux/jump_label.h>

#include <asm/firmware.h>

DECLARE_STATIC_KEY_TRUE(pkey_disabled);

DECLARE_STATIC_KEY_FALSE(pkey_disabled);

extern int pkeys_total; /* total pkeys as per device tree */

extern u32 initial_allocation_mask; /*  bits set for the initially allocated keys */

extern u32 reserved_allocation_mask; /* bits set for reserved keys */

#include <linux/pkeys.h>

#include <linux/of_device.h>

DEFINE_STATIC_KEY_TRUE(pkey_disabled);

DEFINE_STATIC_KEY_FALSE(pkey_disabled);

int  pkeys_total;		/* Total pkeys as per device tree */

u32  initial_allocation_mask;   /* Bits set for the initially allocated keys */

/*

	/* scan the device tree for pkey feature */

	pkeys_total = scan_pkey_feature();

	if (pkeys_total)

		static_branch_disable(&pkey_disabled);

	else {

	if (!pkeys_total) {

		/* No support for pkey. Mark it disabled */

		static_branch_enable(&pkey_disabled);

		return 0;

	}