// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (c) 2012-2014, The Linux Foundation. All rights reserved.
 * Copyright (c) 2017, Linaro Ltd.
 */
#ifndef __QMI_HELPERS_H__
#define __QMI_HELPERS_H__

#include <linux/completion.h>
#include <linux/idr.h>
#include <linux/list.h>
#include <linux/qrtr.h>
#include <linux/types.h>
#include <linux/workqueue.h>

struct socket;

/**
 * struct qmi_header - wireformat header of QMI messages
 * @type: type of message
 * @txn_id: transaction id
 * @msg_id: message id
 * @msg_len: length of message payload following header
 */
struct qmi_header {
 u8 type;
 __le16 txn_id;
 __le16 msg_id;
 __le16 msg_len;
} __packed;

#define QMI_REQUEST 0
#define QMI_RESPONSE 2
#define QMI_INDICATION 4

#define QMI_COMMON_TLV_TYPE 0

enum qmi_elem_type {
 QMI_EOTI,
 QMI_OPT_FLAG,
 QMI_DATA_LEN,
 QMI_UNSIGNED_1_BYTE,
 QMI_UNSIGNED_2_BYTE,
 QMI_UNSIGNED_4_BYTE,
 QMI_UNSIGNED_8_BYTE,
 QMI_SIGNED_2_BYTE_ENUM,
 QMI_SIGNED_4_BYTE_ENUM,
 QMI_STRUCT,
 QMI_STRING,
};

enum qmi_array_type {
 NO_ARRAY,
 STATIC_ARRAY,
 VAR_LEN_ARRAY,
};

/**
 * struct qmi_elem_info - describes how to encode a single QMI element
 * @data_type: Data type of this element.
 * @elem_len: Array length of this element, if an array.
 * @elem_size: Size of a single instance of this data type.
 * @array_type: Array type of this element.
 * @tlv_type: QMI message specific type to identify which element
 *  is present in an incoming message.
 * @offset: Specifies the offset of the first instance of this
 *  element in the data structure.
 * @ei_array: Null-terminated array of @qmi_elem_info to describe nested
 *  structures.
 */
struct qmi_elem_info {
 enum qmi_elem_type data_type;
 u32 elem_len;
 u32 elem_size;
 enum qmi_array_type array_type;
 u8 tlv_type;
 u32 offset;
 const struct qmi_elem_info *ei_array;
};

#define QMI_RESULT_SUCCESS_V01   0
#define QMI_RESULT_FAILURE_V01   1

#define QMI_ERR_NONE_V01   0
#define QMI_ERR_MALFORMED_MSG_V01  1
#define QMI_ERR_NO_MEMORY_V01   2
#define QMI_ERR_INTERNAL_V01   3
#define QMI_ERR_CLIENT_IDS_EXHAUSTED_V01 5
#define QMI_ERR_INVALID_ID_V01   41
#define QMI_ERR_ENCODING_V01   58
#define QMI_ERR_DISABLED_V01                    69
#define QMI_ERR_INCOMPATIBLE_STATE_V01  90
#define QMI_ERR_NOT_SUPPORTED_V01  94

/**
 * struct qmi_response_type_v01 - common response header (decoded)
 * @result: result of the transaction
 * @error: error value, when @result is QMI_RESULT_FAILURE_V01
 */
struct qmi_response_type_v01 {
 u16 result;
 u16 error;
};

extern const struct qmi_elem_info qmi_response_type_v01_ei[];

/**
 * struct qmi_service - context to track lookup-results
 * @service: service type
 * @version: version of the @service
 * @instance: instance id of the @service
 * @node: node of the service
 * @port: port of the service
 * @priv: handle for client's use
 * @list_node: list_head for house keeping
 */
struct qmi_service {
 unsigned int service;
 unsigned int version;
 unsigned int instance;

 unsigned int node;
 unsigned int port;

 void *priv;
 struct list_head list_node;
};

struct qmi_handle;

/**
 * struct qmi_ops - callbacks for qmi_handle
 * @new_server:  inform client of a new_server lookup-result, returning
 *                      successfully from this call causes the library to call
 *                      @del_server as the service is removed from the
 *                      lookup-result. @priv of the qmi_service can be used by
 *                      the client
 * @del_server:  inform client of a del_server lookup-result
 * @net_reset:  inform client that the name service was restarted and
 *                      that and any state needs to be released
 * @msg_handler: invoked for incoming messages, allows a client to
 *                      override the usual QMI message handler
 * @bye:                inform a client that all clients from a node are gone
 * @del_client:         inform a client that a particular client is gone
 */
struct qmi_ops {
 int (*new_server)(struct qmi_handle *qmi, struct qmi_service *svc);
 void (*del_server)(struct qmi_handle *qmi, struct qmi_service *svc);
 void (*net_reset)(struct qmi_handle *qmi);
 void (*msg_handler)(struct qmi_handle *qmi, struct sockaddr_qrtr *sq,
       const void *data, size_t count);
 void (*bye)(struct qmi_handle *qmi, unsigned int node);
 void (*del_client)(struct qmi_handle *qmi,
      unsigned int node, unsigned int port);
};

/**
 * struct qmi_txn - transaction context
 * @qmi: QMI handle this transaction is associated with
 * @id:  transaction id
 * @lock: for synchronization between handler and waiter of messages
 * @completion: completion object as the transaction receives a response
 * @result: result code for the completed transaction
 * @ei:  description of the QMI encoded response (optional)
 * @dest: destination buffer to decode message into (optional)
 */
struct qmi_txn {
 struct qmi_handle *qmi;

 u16 id;

 struct mutex lock;
 struct completion completion;
 int result;

 const struct qmi_elem_info *ei;
 void *dest;
};

/**
 * struct qmi_msg_handler - description of QMI message handler
 * @type: type of message
 * @msg_id: message id
 * @ei:  description of the QMI encoded message
 * @decoded_size: size of the decoded object
 * @fn:  function to invoke as the message is decoded
 */
struct qmi_msg_handler {
 unsigned int type;
 unsigned int msg_id;

 const struct qmi_elem_info *ei;

 size_t decoded_size;
 void (*fn)(struct qmi_handle *qmi, struct sockaddr_qrtr *sq,
     struct qmi_txn *txn, const void *decoded);
};

/**
 * struct qmi_handle - QMI context
 * @sock: socket handle
 * @sock_lock: synchronization of @sock modifications
 * @sq:  sockaddr of @sock
 * @work: work for handling incoming messages
 * @wq:  workqueue to post @work on
 * @recv_buf: scratch buffer for handling incoming messages
 * @recv_buf_size: size of @recv_buf
 * @lookups:  list of registered lookup requests
 * @lookup_results: list of lookup-results advertised to the client
 * @services:  list of registered services (by this client)
 * @ops: reference to callbacks
 * @txns: outstanding transactions
 * @txn_lock: lock for modifications of @txns
 * @handlers: list of handlers for incoming messages
 */
struct qmi_handle {
 struct socket *sock;
 struct mutex sock_lock;

 struct sockaddr_qrtr sq;

 struct work_struct work;
 struct workqueue_struct *wq;

 void *recv_buf;
 size_t recv_buf_size;

 struct list_head lookups;
 struct list_head lookup_results;
 struct list_head services;

 struct qmi_ops ops;

 struct idr txns;
 struct mutex txn_lock;

 const struct qmi_msg_handler *handlers;
};

int qmi_add_lookup(struct qmi_handle *qmi, unsigned int service,
     unsigned int version, unsigned int instance);
int qmi_add_server(struct qmi_handle *qmi, unsigned int service,
     unsigned int version, unsigned int instance);

int qmi_handle_init(struct qmi_handle *qmi, size_t max_msg_len,
      const struct qmi_ops *ops,
      const struct qmi_msg_handler *handlers);
void qmi_handle_release(struct qmi_handle *qmi);

ssize_t qmi_send_request(struct qmi_handle *qmi, struct sockaddr_qrtr *sq,
    struct qmi_txn *txn, int msg_id, size_t len,
    const struct qmi_elem_info *ei, const void *c_struct);
ssize_t qmi_send_response(struct qmi_handle *qmi, struct sockaddr_qrtr *sq,
     struct qmi_txn *txn, int msg_id, size_t len,
     const struct qmi_elem_info *ei, const void *c_struct);
ssize_t qmi_send_indication(struct qmi_handle *qmi, struct sockaddr_qrtr *sq,
       int msg_id, size_t len, const struct qmi_elem_info *ei,
       const void *c_struct);

void *qmi_encode_message(int type, unsigned int msg_id, size_t *len,
    unsigned int txn_id, const struct qmi_elem_info *ei,
    const void *c_struct);

int qmi_decode_message(const void *buf, size_t len,
         const struct qmi_elem_info *ei, void *c_struct);

int qmi_txn_init(struct qmi_handle *qmi, struct qmi_txn *txn,
   const struct qmi_elem_info *ei, void *c_struct);
int qmi_txn_wait(struct qmi_txn *txn, unsigned long timeout);
void qmi_txn_cancel(struct qmi_txn *txn);

#endif