static void opal_handle_message(void)
{
s64 ret;
/*
* TODO: pre-allocate a message buffer depending on opal-msg-size
* value in /proc/device-tree.
*/
static struct opal_msg msg;
u32 type;
ret = opal_get_msg(__pa(&msg), sizeof(msg));
/* No opal message pending. */
if (ret == OPAL_RESOURCE)
return;
/* check for errors. */
if (ret) {
pr_warning("%s: Failed to retrive opal message, err=%lld\n",
__func__, ret);
return;
}
type = be32_to_cpu(msg.msg_type);
/* Sanity check */
if (type > OPAL_MSG_TYPE_MAX) {
pr_warning("%s: Unknown message type: %u\n", __func__, type);
return;
}
opal_message_do_notify(type, (void *)&msg);
}
int main(void)
{
struct opal_msg_entry* entry;
int free_size = OPAL_MAX_MSGS;
int nfree = free_size;
int npending = 0;
int r;
static struct opal_msg m;
uint64_t *m_ptr = (uint64_t *)&m;
zalloc_should_fail = true;
zalloc_should_fail_after = 3;
opal_init_msg();
zalloc_should_fail = false;
opal_init_msg();
assert(list_count(&msg_pending_list) == npending);
assert(list_count(&msg_free_list) == nfree);
/* Callback. */
r = opal_queue_msg(0, &magic, callback, (u64)0, (u64)1, (u64)2);
assert(r == 0);
assert(list_count(&msg_pending_list) == ++npending);
assert(list_count(&msg_free_list) == --nfree);
r = opal_get_msg(m_ptr, sizeof(m));
assert(r == 0);
assert(m.params[0] == 0);
assert(m.params[1] == 1);
assert(m.params[2] == 2);
assert(list_count(&msg_pending_list) == --npending);
assert(list_count(&msg_free_list) == ++nfree);
/* No params. */
r = opal_queue_msg(0, NULL, NULL);
assert(r == 0);
assert(list_count(&msg_pending_list) == ++npending);
assert(list_count(&msg_free_list) == --nfree);
r = opal_get_msg(m_ptr, sizeof(m));
assert(r == 0);
assert(list_count(&msg_pending_list) == --npending);
assert(list_count(&msg_free_list) == ++nfree);
/* > 8 params (ARRAY_SIZE(entry->msg.params) */
r = opal_queue_msg(0, NULL, NULL, 0, 1, 2, 3, 4, 5, 6, 7, 0xBADDA7A);
assert(r == 0);
assert(list_count(&msg_pending_list) == ++npending);
assert(list_count(&msg_free_list) == --nfree);
r = opal_get_msg(m_ptr, sizeof(m));
assert(r == 0);
assert(list_count(&msg_pending_list) == --npending);
assert(list_count(&msg_free_list) == ++nfree);
assert(m.params[0] == 0);
assert(m.params[1] == 1);
assert(m.params[2] == 2);
assert(m.params[3] == 3);
assert(m.params[4] == 4);
assert(m.params[5] == 5);
assert(m.params[6] == 6);
assert(m.params[7] == 7);
/* 8 params (ARRAY_SIZE(entry->msg.params) */
r = opal_queue_msg(0, NULL, NULL, 0, 10, 20, 30, 40, 50, 60, 70);
assert(r == 0);
assert(list_count(&msg_pending_list) == ++npending);
assert(list_count(&msg_free_list) == --nfree);
r = opal_get_msg(m_ptr, sizeof(m));
assert(r == 0);
assert(list_count(&msg_pending_list) == --npending);
assert(list_count(&msg_free_list) == ++nfree);
assert(m.params[0] == 0);
assert(m.params[1] == 10);
assert(m.params[2] == 20);
assert(m.params[3] == 30);
assert(m.params[4] == 40);
assert(m.params[5] == 50);
assert(m.params[6] == 60);
assert(m.params[7] == 70);
/* Full list (no free nodes in pending). */
while (nfree > 0) {
r = opal_queue_msg(OPAL_MSG_ASYNC_COMP, NULL, NULL);
assert(r == 0);
assert(list_count(&msg_pending_list) == ++npending);
assert(list_count(&msg_free_list) == --nfree);
}
assert(list_count(&msg_free_list) == 0);
assert(nfree == 0);
assert(npending == OPAL_MAX_MSGS);
r = opal_queue_msg(OPAL_MSG_ASYNC_COMP, NULL, NULL);
assert(r == 0);
assert(list_count(&msg_pending_list) == OPAL_MAX_MSGS+1);
assert(list_count(&msg_pending_list) == ++npending);
assert(list_count(&msg_free_list) == nfree);
/* Make zalloc fail to test error handling. */
zalloc_should_fail = true;
r = opal_queue_msg(OPAL_MSG_ASYNC_COMP, NULL, NULL);
assert(r == OPAL_RESOURCE);
assert(list_count(&msg_pending_list) == OPAL_MAX_MSGS+1);
assert(list_count(&msg_pending_list) == npending);
assert(list_count(&msg_free_list) == nfree);
/* Empty list (no nodes). */
while(!list_empty(&msg_pending_list)) {
r = opal_get_msg(m_ptr, sizeof(m));
assert(r == 0);
npending--;
nfree++;
}
assert(list_count(&msg_pending_list) == npending);
assert(list_count(&msg_free_list) == nfree);
assert(npending == 0);
assert(nfree == OPAL_MAX_MSGS+1);
r = opal_queue_msg(OPAL_MSG_ASYNC_COMP, NULL, NULL);
assert(r == 0);
assert(list_count(&msg_pending_list) == ++npending);
assert(list_count(&msg_free_list) == --nfree);
/* Request invalid size. */
r = opal_get_msg(m_ptr, sizeof(m) - 1);
assert(r == OPAL_PARAMETER);
/* Pass null buffer. */
r = opal_get_msg(NULL, sizeof(m));
assert(r == OPAL_PARAMETER);
/* Get msg when none are pending. */
r = opal_get_msg(m_ptr, sizeof(m));
assert(r == 0);
r = opal_get_msg(m_ptr, sizeof(m));
assert(r == OPAL_RESOURCE);
#define test_queue_num(type, val) \
r = opal_queue_msg(0, NULL, NULL, \
(type)val, (type)val, (type)val, (type)val, \
(type)val, (type)val, (type)val, (type)val); \
assert(r == 0); \
opal_get_msg(m_ptr, sizeof(m)); \
assert(r == OPAL_SUCCESS); \
assert(m.params[0] == (type)val); \
assert(m.params[1] == (type)val); \
assert(m.params[2] == (type)val); \
assert(m.params[3] == (type)val); \
assert(m.params[4] == (type)val); \
assert(m.params[5] == (type)val); \
assert(m.params[6] == (type)val); \
assert(m.params[7] == (type)val)
/* Test types of various widths */
test_queue_num(u64, -1);
test_queue_num(s64, -1);
test_queue_num(u32, -1);
test_queue_num(s32, -1);
test_queue_num(u16, -1);
test_queue_num(s16, -1);
test_queue_num(u8, -1);
test_queue_num(s8, -1);
/* Clean up the list to keep valgrind happy. */
while(!list_empty(&msg_free_list)) {
entry = list_pop(&msg_free_list, struct opal_msg_entry, link);
assert(entry);
free(entry);
}
while(!list_empty(&msg_pending_list)) {
entry = list_pop(&msg_pending_list, struct opal_msg_entry, link);
assert(entry);
free(entry);
}
return 0;
}
