Function wrap_tm_func

modules/compression/compression.c:288–340

}

void wrap_tm_func(struct cell* t, int type, struct tmcb_params* p)
{
	int ret = 0;
	mc_whitelist_p wh_list = NULL;
	struct mc_compact_args* mc_compact_args = NULL;
	struct mc_comp_args* args = NULL;
	struct ua_client *uac = & TM_BRANCH( t, p->code);
	char* buf = uac->request.buffer.s;
	int olen = uac->request.buffer.len;

	switch (type) {
		case COMPRESS_CB:
			if ((args = GET_GLOBAL_CTX(compress_ctx_pos)) == NULL)
				break;

			if ((ret = mc_compress_cb(&buf, args, TM_CB, &olen)) < 0)
				LM_ERR("compression failed\n");

			wh_list = args->hdr2compress_list;
			pkg_free(args);
			SET_GLOBAL_CTX(compress_ctx_pos, NULL);
			break;

		case COMPACT_CB:
			/* if not registered yet we take from global context */
			if ((mc_compact_args = GET_GLOBAL_CTX(compact_ctx_pos)) == NULL)
				break;

			if ((ret = mc_compact_cb(&buf, mc_compact_args, TM_CB, &olen)) < 0)
				LM_ERR("compaction failed\n");

			SET_GLOBAL_CTX(compact_ctx_pos, NULL);

			break;

		default:
			LM_BUG("!!! invalid CB type arg!\n");
			return;
	}

	/* free whitelists for both actions */
	if (wh_list)
		free_whitelist(wh_list);
	free_mc_compact_args(mc_compact_args);
	if (ret < 0)
		return;

	uac->request.buffer.s = buf;
	uac->request.buffer.len = olen;
	/* we also need to compute the uri so that it points within the new buffer */
	uac->uri.s = buf + t->method.len + 1;
	/* uri.len should be the same, since it is not changed by compression */
}

int wrap_msg_compress(str* buf, struct sip_msg* p_msg) {
	return wrap_msg_func( buf, p_msg, COMPRESS_CB);