|
|
@ -34,12 +34,7 @@ |
|
|
|
#include <grpc/support/alloc.h> |
|
|
|
#include <grpc/support/alloc.h> |
|
|
|
#include <grpc/support/log.h> |
|
|
|
#include <grpc/support/log.h> |
|
|
|
|
|
|
|
|
|
|
|
typedef struct { |
|
|
|
user_agent_parser ua_parser[GRPC_MAX_WORKAROUND_ID]; |
|
|
|
bool enabled; |
|
|
|
|
|
|
|
user_agent_parser ua_parser; |
|
|
|
|
|
|
|
} workaround_context; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static workaround_context workarounds[GRPC_MAX_WORKAROUND_ID]; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static void destroy_user_agent_md(void *user_agent_md) { |
|
|
|
static void destroy_user_agent_md(void *user_agent_md) { |
|
|
|
gpr_free(user_agent_md); |
|
|
|
gpr_free(user_agent_md); |
|
|
@ -55,8 +50,8 @@ grpc_workaround_user_agent_md *grpc_parse_user_agent(grpc_mdelem md) { |
|
|
|
} |
|
|
|
} |
|
|
|
user_agent_md = gpr_malloc(sizeof(grpc_workaround_user_agent_md)); |
|
|
|
user_agent_md = gpr_malloc(sizeof(grpc_workaround_user_agent_md)); |
|
|
|
for (int i = 0; i < GRPC_MAX_WORKAROUND_ID; i++) { |
|
|
|
for (int i = 0; i < GRPC_MAX_WORKAROUND_ID; i++) { |
|
|
|
if (workarounds[i].enabled && workarounds[i].ua_parser) { |
|
|
|
if (ua_parser[i]) { |
|
|
|
user_agent_md->workaround_active[i] = workarounds[i].ua_parser(md); |
|
|
|
user_agent_md->workaround_active[i] = ua_parser[i](md); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
grpc_mdelem_set_user_data(md, destroy_user_agent_md, (void *)user_agent_md); |
|
|
|
grpc_mdelem_set_user_data(md, destroy_user_agent_md, (void *)user_agent_md); |
|
|
@ -66,10 +61,6 @@ grpc_workaround_user_agent_md *grpc_parse_user_agent(grpc_mdelem md) { |
|
|
|
|
|
|
|
|
|
|
|
void grpc_register_workaround(uint32_t id, user_agent_parser parser) { |
|
|
|
void grpc_register_workaround(uint32_t id, user_agent_parser parser) { |
|
|
|
GPR_ASSERT(id < GRPC_MAX_WORKAROUND_ID); |
|
|
|
GPR_ASSERT(id < GRPC_MAX_WORKAROUND_ID); |
|
|
|
workarounds[id].ua_parser = parser; |
|
|
|
ua_parser[id] = parser; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void grpc_enable_workaround(uint32_t id) { |
|
|
|
|
|
|
|
GPR_ASSERT(id < GRPC_MAX_WORKAROUND_ID); |
|
|
|
|
|
|
|
workarounds[id].enabled = true; |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|