156 lines
5.4 KiB
C
156 lines
5.4 KiB
C
/*-
|
|
* SPDX-License-Identifier: BSD-2-Clause-FreeBSD
|
|
*
|
|
* Copyright (C) 2012-2013 Intel Corporation
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*
|
|
* $FreeBSD$
|
|
*/
|
|
|
|
#ifndef __NVMECONTROL_H__
|
|
#define __NVMECONTROL_H__
|
|
|
|
#include <sys/linker_set.h>
|
|
#include <dev/nvme/nvme.h>
|
|
|
|
struct nvme_function;
|
|
typedef void (*nvme_fn_t)(const struct nvme_function *nf, int argc, char *argv[]);
|
|
|
|
struct nvme_function {
|
|
const char *name;
|
|
nvme_fn_t fn;
|
|
const char *usage;
|
|
};
|
|
|
|
#define NVME_SETNAME(set) set
|
|
#define NVME_CMDSET(set, sym) DATA_SET(NVME_SETNAME(set), sym)
|
|
#define NVME_COMMAND(set, nam, function, usage_str) \
|
|
static struct nvme_function function ## _nvme_cmd = \
|
|
{ .name = #nam, .fn = function, .usage = usage_str }; \
|
|
NVME_CMDSET(set, function ## _nvme_cmd)
|
|
#define NVME_CMD_BEGIN(set) SET_BEGIN(NVME_SETNAME(set))
|
|
#define NVME_CMD_LIMIT(set) SET_LIMIT(NVME_SETNAME(set))
|
|
#define NVME_CMD_DECLARE(set, t) SET_DECLARE(NVME_SETNAME(set), t)
|
|
|
|
typedef void (*print_fn_t)(const struct nvme_controller_data *cdata, void *buf, uint32_t size);
|
|
|
|
struct logpage_function {
|
|
uint8_t log_page;
|
|
const char *vendor;
|
|
const char *name;
|
|
print_fn_t print_fn;
|
|
size_t size;
|
|
};
|
|
|
|
|
|
#define NVME_LOGPAGESET(sym) DATA_SET(NVME_SETNAME(logpage), sym)
|
|
#define NVME_LOGPAGE(unique, lp, vend, nam, fn, sz) \
|
|
static struct logpage_function unique ## _lpf = { \
|
|
.log_page = lp, \
|
|
.vendor = vend, \
|
|
.name = nam, \
|
|
.print_fn = fn, \
|
|
.size = sz, \
|
|
} ; \
|
|
NVME_LOGPAGESET(unique ## _lpf)
|
|
#define NVME_LOGPAGE_BEGIN SET_BEGIN(NVME_SETNAME(logpage))
|
|
#define NVME_LOGPAGE_LIMIT SET_LIMIT(NVME_SETNAME(logpage))
|
|
#define NVME_LOGPAGE_DECLARE(t) SET_DECLARE(NVME_SETNAME(logpage), t)
|
|
|
|
#define DEFAULT_SIZE (4096)
|
|
struct kv_name {
|
|
uint32_t key;
|
|
const char *name;
|
|
};
|
|
|
|
const char *kv_lookup(const struct kv_name *kv, size_t kv_count, uint32_t key);
|
|
|
|
NVME_CMD_DECLARE(top, struct nvme_function);
|
|
NVME_LOGPAGE_DECLARE(struct logpage_function);
|
|
|
|
struct set_concat {
|
|
void **begin;
|
|
void **limit;
|
|
};
|
|
void set_concat_add(struct set_concat *m, void *begin, void *end);
|
|
#define SET_CONCAT_DEF(set, t) \
|
|
static struct set_concat set ## _concat; \
|
|
static inline const t * const *set ## _begin(void) { return ((const t * const *)set ## _concat.begin); } \
|
|
static inline const t * const *set ## _limit(void) { return ((const t * const *)set ## _concat.limit); } \
|
|
void add_to_ ## set(t **b, t **e) \
|
|
{ \
|
|
set_concat_add(&set ## _concat, b, e); \
|
|
}
|
|
#define SET_CONCAT_DECL(set, t) \
|
|
void add_to_ ## set(t **b, t **e)
|
|
SET_CONCAT_DECL(top, struct nvme_function);
|
|
SET_CONCAT_DECL(logpage, struct logpage_function);
|
|
|
|
#define NVME_CTRLR_PREFIX "nvme"
|
|
#define NVME_NS_PREFIX "ns"
|
|
|
|
int open_dev(const char *str, int *fd, int show_error, int exit_on_error);
|
|
void parse_ns_str(const char *ns_str, char *ctrlr_str, uint32_t *nsid);
|
|
void read_controller_data(int fd, struct nvme_controller_data *cdata);
|
|
void read_namespace_data(int fd, uint32_t nsid, struct nvme_namespace_data *nsdata);
|
|
void print_hex(void *data, uint32_t length);
|
|
void read_logpage(int fd, uint8_t log_page, uint32_t nsid, void *payload,
|
|
uint32_t payload_size);
|
|
void print_temp(uint16_t t);
|
|
void print_intel_add_smart(const struct nvme_controller_data *cdata __unused, void *buf, uint32_t size __unused);
|
|
|
|
void usage(const struct nvme_function *f);
|
|
void dispatch_set(int argc, char *argv[], const struct nvme_function * const *tbl,
|
|
const struct nvme_function * const *tbl_limit);
|
|
|
|
#define DISPATCH(argc, argv, set) \
|
|
dispatch_set(argc, argv, \
|
|
(const struct nvme_function * const *)NVME_CMD_BEGIN(set), \
|
|
(const struct nvme_function * const *)NVME_CMD_LIMIT(set)) \
|
|
|
|
/* Utility Routines */
|
|
/*
|
|
* 128-bit integer augments to standard values. On i386 this
|
|
* doesn't exist, so we use 64-bit values. So, on 32-bit i386,
|
|
* you'll get truncated values until someone implement 128bit
|
|
* ints in sofware.
|
|
*/
|
|
#define UINT128_DIG 39
|
|
#ifdef __i386__
|
|
typedef uint64_t uint128_t;
|
|
#else
|
|
typedef __uint128_t uint128_t;
|
|
#endif
|
|
|
|
static __inline uint128_t
|
|
to128(void *p)
|
|
{
|
|
return *(uint128_t *)p;
|
|
}
|
|
|
|
uint64_t le48dec(const void *pp);
|
|
char * uint128_to_str(uint128_t u, char *buf, size_t buflen);
|
|
|
|
#endif
|