| |
| |
| |
| |
| |
| |
| #include <psxtypes/psxtypes.h> |
| #include <ntapi/nt_file.h> |
| #include <ntapi/nt_string.h> |
| #include <ntapi/nt_atomic.h> |
| #include <ntapi/nt_port.h> |
| #include <ntapi/nt_ipc.h> |
| #include <ntapi/nt_msq.h> |
| #include <ntapi/ntapi.h> |
| #include "ntapi_impl.h" |
| |
| static int32_t __msq_recv_return( |
| void * mapaddr, |
| intptr_t * hlock, |
| int32_t status) |
| { |
| if (hlock) |
| at_store(hlock,0); |
| |
| if (mapaddr) |
| __ntapi->zw_unmap_view_of_section( |
| NT_CURRENT_PROCESS_HANDLE, |
| mapaddr); |
| |
| return status; |
| } |
| |
| |
| int32_t __stdcall __ntapi_msq_recv( |
| __in nt_msq_info * msq, |
| __in void * hevent __optional, |
| __in nt_io_apc_routine * apc_routine __optional, |
| __in void * apc_context __optional, |
| __out void * buffer, |
| __in size_t len, |
| __in intptr_t rank, |
| __in uint32_t options, |
| __out nt_io_status_block * iosb) |
| { |
| int32_t status; |
| void * mapaddr; |
| void * hsection; |
| void * secaddr; |
| size_t secsize; |
| nt_msq_info_msg msg; |
| intptr_t * hlock; |
| |
| |
| if (!iosb) |
| return NT_STATUS_INVALID_PARAMETER; |
| |
| else if (!len) |
| return NT_STATUS_INVALID_PARAMETER; |
| |
| |
| if (msq->section_addr) { |
| hsection = msq->section; |
| secaddr = msq->section_addr; |
| secsize = msq->section_size; |
| mapaddr = 0; |
| |
| } else if ((status = __ntapi->ipc_init_section_by_port( |
| msq->hport,&hsection, |
| &secaddr,&secsize))) |
| return status; |
| |
| else |
| mapaddr = secaddr; |
| |
| |
| if (len > secsize) |
| len = secsize; |
| |
| |
| hlock = &(__ntapi_internals()->hlock); |
| |
| if (at_locked_cas(hlock,0,1)) |
| return __msq_recv_return( |
| mapaddr,0, |
| NT_STATUS_RESOURCE_NOT_OWNED); |
| |
| |
| __ntapi->tt_aligned_block_memset( |
| &msg,0,sizeof(msg)); |
| |
| msg.header.msg_type = NT_LPC_NEW_MESSAGE; |
| msg.header.data_size = sizeof(msg.data); |
| msg.header.msg_size = sizeof(msg); |
| msg.data.ttyinfo.opcode = NT_TTY_MSQ_RECV; |
| |
| msg.data.msqinfo.msqkey = msq->msqkey; |
| msg.data.msqinfo.msqid = msq->msqid; |
| msg.data.msqinfo.msqrpid = msq->msqrpid; |
| msg.data.msqinfo.msqrank = rank; |
| msg.data.msqinfo.section_addr = secaddr; |
| msg.data.msqinfo.section_size = len; |
| msg.data.msqinfo.ntoptions = options; |
| |
| msg.data.msqinfo.hevent = hevent; |
| msg.data.msqinfo.apc_routine = apc_routine; |
| msg.data.msqinfo.apc_context = apc_context; |
| msg.data.msqinfo.riosb = iosb; |
| |
| if ((status = __ntapi->zw_request_wait_reply_port(msq->hport,&msg,&msg))) |
| return __msq_recv_return(mapaddr,hlock,status); |
| else if (msg.data.ttyinfo.status) |
| return __msq_recv_return(mapaddr,hlock,msg.data.ttyinfo.status); |
| |
| |
| __ntapi->tt_generic_memcpy( |
| buffer,secaddr, |
| msg.data.msqinfo.ntiosb.info); |
| |
| |
| iosb->status = msg.data.msqinfo.ntiosb.status; |
| iosb->info = msg.data.msqinfo.ntiosb.info; |
| |
| return __msq_recv_return(mapaddr,hlock,NT_STATUS_SUCCESS); |
| } |