-
-
Notifications
You must be signed in to change notification settings - Fork 108
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* move to mach messages * cleanup * mach responses * finalize xnu mach ipc * clean up memory management
- Loading branch information
1 parent
d59ce51
commit 8f3e5fa
Showing
12 changed files
with
251 additions
and
243 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,166 @@ | ||
| #include "mach.h" | ||
| #include <mach/message.h> | ||
| #include <stdint.h> | ||
|
|
||
| mach_port_t mach_get_bs_port() { | ||
| mach_port_name_t task = mach_task_self(); | ||
|
|
||
| mach_port_t bs_port; | ||
| if (task_get_special_port(task, | ||
| TASK_BOOTSTRAP_PORT, | ||
| &bs_port ) != KERN_SUCCESS) { | ||
| return 0; | ||
| } | ||
|
|
||
| mach_port_t port; | ||
| if (bootstrap_look_up(bs_port, | ||
| "git.felix.sketchybar", | ||
| &port ) != KERN_SUCCESS) { | ||
| return 0; | ||
| } | ||
|
|
||
| return port; | ||
| } | ||
|
|
||
| void mach_receive_message(mach_port_t port, struct mach_buffer* buffer, | ||
| bool timeout ) { | ||
| *buffer = (struct mach_buffer) { 0 }; | ||
| mach_msg_return_t msg_return; | ||
| if (timeout) | ||
| msg_return = mach_msg(&buffer->message.header, | ||
| MACH_RCV_MSG | MACH_RCV_TIMEOUT, | ||
| 0, | ||
| sizeof(struct mach_buffer), | ||
| port, | ||
| 10, | ||
| MACH_PORT_NULL ); | ||
| else | ||
| msg_return = mach_msg(&buffer->message.header, | ||
| MACH_RCV_MSG, | ||
| 0, | ||
| sizeof(struct mach_buffer), | ||
| port, | ||
| MACH_MSG_TIMEOUT_NONE, | ||
| MACH_PORT_NULL ); | ||
|
|
||
| if (msg_return != MACH_MSG_SUCCESS) { | ||
| buffer->message.descriptor.address = NULL; | ||
| } | ||
| } | ||
|
|
||
| bool mach_send_message(mach_port_t port, char* message, uint32_t len, | ||
| bool await_response) { | ||
| if (!message || !port) { | ||
| if (message) free(message); | ||
| return false; | ||
| } | ||
|
|
||
| mach_port_t response_port; | ||
| if (await_response) { | ||
| mach_port_name_t task = mach_task_self(); | ||
| if (mach_port_allocate(task, MACH_PORT_RIGHT_RECEIVE, | ||
| &response_port ) != KERN_SUCCESS) { | ||
| return false; | ||
| } | ||
|
|
||
| if (mach_port_insert_right(task, response_port, | ||
| response_port, | ||
| MACH_MSG_TYPE_MAKE_SEND)!= KERN_SUCCESS) { | ||
| return false; | ||
| } | ||
| } | ||
|
|
||
| struct mach_message msg = { 0 }; | ||
| msg.header.msgh_remote_port = port; | ||
| if (await_response) { | ||
| msg.header.msgh_local_port = response_port; | ||
| msg.header.msgh_id = response_port; | ||
| msg.header.msgh_bits = MACH_MSGH_BITS_SET(MACH_MSG_TYPE_COPY_SEND, | ||
| MACH_MSG_TYPE_MAKE_SEND_ONCE, | ||
| 0, | ||
| MACH_MSGH_BITS_COMPLEX ); | ||
| } else { | ||
| msg.header.msgh_bits = MACH_MSGH_BITS_SET(MACH_MSG_TYPE_MOVE_SEND_ONCE | ||
| & MACH_MSGH_BITS_REMOTE_MASK, | ||
| 0, | ||
| 0, | ||
| MACH_MSGH_BITS_COMPLEX ); | ||
| } | ||
|
|
||
| msg.header.msgh_size = sizeof(struct mach_message); | ||
|
|
||
| msg.msgh_descriptor_count = 1; | ||
| msg.descriptor.address = message; | ||
| msg.descriptor.size = len * sizeof(char); | ||
| msg.descriptor.copy = MACH_MSG_VIRTUAL_COPY; | ||
| msg.descriptor.deallocate = await_response; | ||
| msg.descriptor.type = MACH_MSG_OOL_DESCRIPTOR; | ||
|
|
||
| mach_msg_return_t msg_return = mach_msg(&msg.header, | ||
| MACH_SEND_MSG, | ||
| sizeof(struct mach_message), | ||
| 0, | ||
| MACH_PORT_NULL, | ||
| MACH_MSG_TIMEOUT_NONE, | ||
| MACH_PORT_NULL ); | ||
|
|
||
| if (await_response) { | ||
| struct mach_buffer buffer = { 0 }; | ||
| mach_receive_message(response_port, &buffer, true); | ||
| if (buffer.message.descriptor.address) | ||
| printf("%s", (char*)buffer.message.descriptor.address); | ||
| mach_msg_destroy(&buffer.message.header); | ||
| } | ||
|
|
||
| return msg_return == MACH_MSG_SUCCESS; | ||
| } | ||
|
|
||
| static void* mach_connection_handler(void *context) { | ||
| struct mach_server* server = context; | ||
| while (server->is_running) { | ||
| struct mach_buffer* buffer = malloc(sizeof(struct mach_buffer)); | ||
| mach_receive_message(server->port, buffer, false); | ||
| server->handler(buffer); | ||
| } | ||
|
|
||
| return NULL; | ||
| } | ||
|
|
||
| #pragma clang diagnostic push | ||
| #pragma clang diagnostic ignored "-Wdeprecated-declarations" | ||
| bool mach_server_begin(struct mach_server* mach_server, mach_handler handler) { | ||
| mach_server->task = mach_task_self(); | ||
|
|
||
| if (mach_port_allocate(mach_server->task, | ||
| MACH_PORT_RIGHT_RECEIVE, | ||
| &mach_server->port ) != KERN_SUCCESS) { | ||
| return false; | ||
| } | ||
|
|
||
| if (mach_port_insert_right(mach_server->task, | ||
| mach_server->port, | ||
| mach_server->port, | ||
| MACH_MSG_TYPE_MAKE_SEND) != KERN_SUCCESS) { | ||
| return false; | ||
| } | ||
|
|
||
| if (task_get_special_port(mach_server->task, | ||
| TASK_BOOTSTRAP_PORT, | ||
| &mach_server->bs_port) != KERN_SUCCESS) { | ||
| return false; | ||
| } | ||
|
|
||
| if (bootstrap_register(mach_server->bs_port, | ||
| "git.felix.sketchybar", | ||
| mach_server->port ) != KERN_SUCCESS) { | ||
| return false; | ||
| } | ||
|
|
||
| mach_server->handler = handler; | ||
| mach_server->is_running = true; | ||
| pthread_create(&mach_server->thread, NULL, &mach_connection_handler, | ||
| mach_server ); | ||
|
|
||
| return true; | ||
| } | ||
| #pragma clang diagnostic pop |
Oops, something went wrong.