123 lines
3.6 KiB
C
123 lines
3.6 KiB
C
#pragma once
|
|
|
|
#include <mach/arm/kern_return.h>
|
|
#include <mach/mach.h>
|
|
#include <mach/mach_port.h>
|
|
#include <mach/message.h>
|
|
#include <bootstrap.h>
|
|
#include <stdlib.h>
|
|
#include <pthread.h>
|
|
#include <stdio.h>
|
|
|
|
typedef char* env;
|
|
|
|
#define MACH_HANDLER(name) void name(env env)
|
|
typedef MACH_HANDLER(mach_handler);
|
|
|
|
struct mach_message {
|
|
mach_msg_header_t header;
|
|
mach_msg_size_t msgh_descriptor_count;
|
|
mach_msg_ool_descriptor_t descriptor;
|
|
};
|
|
|
|
struct mach_buffer {
|
|
struct mach_message message;
|
|
mach_msg_trailer_t trailer;
|
|
};
|
|
|
|
static mach_port_t g_mach_port = 0;
|
|
|
|
static inline 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;
|
|
}
|
|
|
|
char* name = getenv("BAR_NAME");
|
|
if (!name) name = "sketchybar";
|
|
uint32_t lookup_len = 16 + strlen(name);
|
|
|
|
char buffer[lookup_len];
|
|
snprintf(buffer, lookup_len, "git.felix.%s", name);
|
|
|
|
mach_port_t port;
|
|
if (bootstrap_look_up(bs_port, buffer, &port) != KERN_SUCCESS) return 0;
|
|
return port;
|
|
}
|
|
|
|
static inline bool mach_send_message(mach_port_t port, char* message, uint32_t len) {
|
|
if (!message || !port) {
|
|
return false;
|
|
}
|
|
|
|
struct mach_message msg = { 0 };
|
|
msg.header.msgh_remote_port = port;
|
|
msg.header.msgh_local_port = 0;
|
|
msg.header.msgh_id = 0;
|
|
msg.header.msgh_bits = MACH_MSGH_BITS_SET(MACH_MSG_TYPE_COPY_SEND,
|
|
MACH_MSG_TYPE_MAKE_SEND,
|
|
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 = false;
|
|
msg.descriptor.type = MACH_MSG_OOL_DESCRIPTOR;
|
|
|
|
kern_return_t err = mach_msg(&msg.header,
|
|
MACH_SEND_MSG,
|
|
sizeof(struct mach_message),
|
|
0,
|
|
MACH_PORT_NULL,
|
|
MACH_MSG_TIMEOUT_NONE,
|
|
MACH_PORT_NULL );
|
|
|
|
return err == KERN_SUCCESS;
|
|
}
|
|
|
|
static inline uint32_t format_message(char* message, char* formatted_message) {
|
|
// This is not actually robust, switch to stack based messaging.
|
|
char outer_quote = 0;
|
|
uint32_t caret = 0;
|
|
uint32_t message_length = strlen(message) + 1;
|
|
for (int i = 0; i < message_length; ++i) {
|
|
if (message[i] == '"' || message[i] == '\'') {
|
|
if (outer_quote && outer_quote == message[i]) outer_quote = 0;
|
|
else if (!outer_quote) outer_quote = message[i];
|
|
continue;
|
|
}
|
|
formatted_message[caret] = message[i];
|
|
if (message[i] == ' ' && !outer_quote) formatted_message[caret] = '\0';
|
|
caret++;
|
|
}
|
|
|
|
if (caret > 0 && formatted_message[caret] == '\0'
|
|
&& formatted_message[caret - 1] == '\0') {
|
|
caret--;
|
|
}
|
|
formatted_message[caret] = '\0';
|
|
return caret + 1;
|
|
}
|
|
|
|
static inline void sketchybar(char* message) {
|
|
char formatted_message[strlen(message) + 2];
|
|
uint32_t length = format_message(message, formatted_message);
|
|
if (!length) return;
|
|
|
|
if (!g_mach_port) g_mach_port = mach_get_bs_port();
|
|
if (!mach_send_message(g_mach_port, formatted_message, length)) {
|
|
g_mach_port = mach_get_bs_port();
|
|
if (!mach_send_message(g_mach_port, formatted_message, length)) {
|
|
// No sketchybar instance running, exit.
|
|
exit(0);
|
|
}
|
|
}
|
|
}
|