#include <OS.h>
#include <poll.h>
#include <event_queue_defs.h>
#include "strace.h"
#include "Syscall.h"
#include "Context.h"
#include "MemoryReader.h"
#include "TypeHandler.h"
struct enum_info {
unsigned int index;
const char *name;
};
#define ENUM_INFO_ENTRY(name) \
{ name, #name }
#define FLAG_INFO_ENTRY(name) \
{ name, #name }
static const FlagsTypeHandler::FlagInfo kPollFlagInfos[] = {
FLAG_INFO_ENTRY(POLLIN),
FLAG_INFO_ENTRY(POLLOUT),
FLAG_INFO_ENTRY(POLLRDBAND),
FLAG_INFO_ENTRY(POLLWRBAND),
FLAG_INFO_ENTRY(POLLPRI),
FLAG_INFO_ENTRY(POLLERR),
FLAG_INFO_ENTRY(POLLHUP),
FLAG_INFO_ENTRY(POLLNVAL),
{ 0, NULL }
};
static const FlagsTypeHandler::FlagInfo kEventFlagInfos[] = {
FLAG_INFO_ENTRY(B_EVENT_READ),
FLAG_INFO_ENTRY(B_EVENT_WRITE),
FLAG_INFO_ENTRY(B_EVENT_ERROR),
FLAG_INFO_ENTRY(B_EVENT_PRIORITY_READ),
FLAG_INFO_ENTRY(B_EVENT_PRIORITY_WRITE),
FLAG_INFO_ENTRY(B_EVENT_HIGH_PRIORITY_READ),
FLAG_INFO_ENTRY(B_EVENT_HIGH_PRIORITY_WRITE),
FLAG_INFO_ENTRY(B_EVENT_DISCONNECTED),
FLAG_INFO_ENTRY(B_EVENT_INVALID),
FLAG_INFO_ENTRY(B_EVENT_LEVEL_TRIGGERED),
FLAG_INFO_ENTRY(B_EVENT_ONE_SHOT),
{ 0, NULL }
};
static const char* kObjectTypes[] = {
"fd",
"sem",
"port",
"thread",
};
static FlagsTypeHandler::FlagsList kPollFlags;
static FlagsTypeHandler sPollFlagsHandler(kPollFlags);
static FlagsTypeHandler::FlagsList kEventFlags;
static FlagsTypeHandler sEventFlagsHandler(kEventFlags);
static string
read_fdset(Context &context, void *data)
{
unsigned long tmp[1024 / (sizeof(unsigned long) * 8)];
int32 bytesRead;
status_t err = context.Reader().Read(data, &tmp, sizeof(tmp), bytesRead);
if (err != B_OK)
return context.FormatPointer(data);
int count = bytesRead / sizeof(unsigned long);
int added = 0;
string r;
r.reserve(16);
r = "[";
for (int i = 0; i < count && added < 8; i++) {
for (int j = 0;
j < (int)(sizeof(unsigned long) * 8) && added < 8; j++) {
if (tmp[i] & (1UL << j)) {
if (added > 0)
r += " ";
unsigned int fd = i * sizeof(unsigned long) * 8 + j;
r += context.FormatUnsigned(fd);
added++;
}
}
}
if (added >= 8)
r += " ...";
r += "]";
return r;
}
template<>
string
TypeHandlerImpl<fd_set *>::GetParameterValue(Context &context, Parameter *,
const void *address)
{
void *data = *(void **)address;
if (data != NULL && context.GetContents(Context::SIMPLE_STRUCTS))
return read_fdset(context, data);
return context.FormatPointer(data);
}
template<>
string
TypeHandlerImpl<fd_set *>::GetReturnValue(Context &context, uint64 value)
{
return context.FormatPointer((void *)value);
}
static string
read_pollfd(Context &context, void *data)
{
nfds_t numfds = context.ReadValue<nfds_t>(context.GetSibling(1));
if ((int64)numfds <= 0)
return string();
pollfd tmp[numfds];
int32 bytesRead;
status_t err = context.Reader().Read(data, &tmp, sizeof(tmp), bytesRead);
if (err != B_OK)
return context.FormatPointer(data);
string r;
r.reserve(16);
r = "[";
int added = 0;
for (nfds_t i = 0; i < numfds && added < 8; i++) {
if ((tmp[i].fd == -1 || tmp[i].revents == 0)
&& context.GetContents(Context::OUTPUT_VALUES)) {
continue;
}
if (added > 0)
r += ", ";
r += "{fd=" + context.FormatSigned(tmp[i].fd);
if (tmp[i].fd != -1 && context.GetContents(Context::INPUT_VALUES)) {
r += ", events=";
r += sPollFlagsHandler.RenderValue(context, tmp[i].events);
}
if (context.GetContents(Context::OUTPUT_VALUES)) {
r += ", revents=";
r += sPollFlagsHandler.RenderValue(context, tmp[i].revents);
}
added++;
r += "}";
}
if (added >= 8)
r += " ...";
r += "]";
return r;
}
template<>
string
TypeHandlerImpl<pollfd *>::GetParameterValue(Context &context, Parameter *,
const void *address)
{
void *data = *(void **)address;
if (data != NULL && context.GetContents(Context::SIMPLE_STRUCTS))
return read_pollfd(context, data);
return context.FormatPointer(data);
}
template<>
string
TypeHandlerImpl<pollfd *>::GetReturnValue(Context &context, uint64 value)
{
return context.FormatPointer((void *)value);
}
static string
read_object_wait_infos(Context &context, Parameter *param, void *data)
{
int numInfos = context.ReadValue<int>(context.GetNextSibling(param));
if (numInfos <= 0)
return string();
object_wait_info tmp[numInfos];
int32 bytesRead;
status_t err = context.Reader().Read(data, &tmp, sizeof(tmp), bytesRead);
if (err != B_OK)
return context.FormatPointer(data);
string r;
r.reserve(16);
r = "[";
for (int i = 0; i < numInfos; i++) {
if (i > 0)
r += ", ";
if (i >= 8) {
r += "...";
break;
}
r += "{";
r += (tmp[i].type < sizeof(kObjectTypes)) ?
kObjectTypes[tmp[i].type] : context.FormatUnsigned(tmp[i].type);
r += "=";
r += context.FormatSigned(tmp[i].object);
r += ", events=";
r += sEventFlagsHandler.RenderValue(context, tmp[i].events);
r += "}";
}
r += "]";
return r;
}
template<>
string
TypeHandlerImpl<object_wait_info *>::GetParameterValue(Context &context, Parameter *param,
const void *address)
{
void *data = *(void **)address;
if (data != NULL && context.GetContents(Context::SIMPLE_STRUCTS))
return read_object_wait_infos(context, param, data);
return context.FormatPointer(data);
}
template<>
string
TypeHandlerImpl<object_wait_info *>::GetReturnValue(Context &context, uint64 value)
{
return context.FormatPointer((void *)value);
}
static string
read_event_wait_infos(Context &context, Parameter *param, void *data)
{
int numInfos = 0;
if (param->Out())
numInfos = context.GetReturnValue();
else
numInfos = context.ReadValue<int>(context.GetNextSibling(param));
if (numInfos <= 0)
return context.FormatPointer(data);
event_wait_info tmp[numInfos];
int32 bytesRead;
status_t err = context.Reader().Read(data, &tmp, sizeof(tmp), bytesRead);
if (err != B_OK)
return context.FormatPointer(data);
string r;
r.reserve(16);
r = "[";
for (int i = 0; i < numInfos; i++) {
if (i > 0)
r += ", ";
if (i >= 8) {
r += "...";
break;
}
r += "{";
r += (tmp[i].type < sizeof(kObjectTypes)) ?
kObjectTypes[tmp[i].type] : context.FormatUnsigned(tmp[i].type);
r += "=";
r += context.FormatSigned(tmp[i].object);
r += ", events=";
if (tmp[i].events == -1)
r += "-1";
else if (tmp[i].events < 0)
r += strerror(tmp[i].events);
else
r += sEventFlagsHandler.RenderValue(context, tmp[i].events);
if (tmp[i].user_data != NULL) {
r += ", user_data=";
r += context.FormatPointer(tmp[i].user_data);
}
r += "}";
}
r += "]";
return r;
}
template<>
string
TypeHandlerImpl<event_wait_info *>::GetParameterValue(Context &context, Parameter *param,
const void *address)
{
void *data = *(void **)address;
if (data != NULL && context.GetContents(Context::SIMPLE_STRUCTS))
return read_event_wait_infos(context, param, data);
return context.FormatPointer(data);
}
template<>
string
TypeHandlerImpl<event_wait_info *>::GetReturnValue(Context &context, uint64 value)
{
return context.FormatPointer((void *)value);
}
DEFINE_TYPE(fdset_ptr, fd_set *)
DEFINE_TYPE(pollfd_ptr, pollfd *)
DEFINE_TYPE(object_wait_infos_ptr, object_wait_info *)
DEFINE_TYPE(event_wait_infos_ptr, event_wait_info *)
void
patch_events()
{
for (int i = 0; kPollFlagInfos[i].name != NULL; i++)
kPollFlags.push_back(kPollFlagInfos[i]);
for (int i = 0; kEventFlagInfos[i].name != NULL; i++)
kEventFlags.push_back(kEventFlagInfos[i]);
Syscall *poll = get_syscall("_kern_poll");
poll->ParameterAt(0)->SetInOut(true);
Syscall *select = get_syscall("_kern_select");
select->ParameterAt(1)->SetInOut(true);
select->ParameterAt(2)->SetInOut(true);
select->ParameterAt(3)->SetInOut(true);
Syscall *wait_for_objects = get_syscall("_kern_wait_for_objects");
wait_for_objects->ParameterAt(0)->SetInOut(true);
Syscall *event_queue_select = get_syscall("_kern_event_queue_select");
event_queue_select->ParameterAt(1)->SetInOut(true);
Syscall *event_queue_wait = get_syscall("_kern_event_queue_wait");
event_queue_wait->ParameterAt(1)->SetOut(true);
Syscall *wait_for_child = get_syscall("_kern_wait_for_child");
wait_for_child->ParameterAt(2)->SetOut(true);
wait_for_child->ParameterAt(3)->SetOut(true);
Syscall *wait_for_thread_etc = get_syscall("_kern_wait_for_thread_etc");
wait_for_thread_etc->ParameterAt(3)->SetOut(true);
}
