gint l_quvi_http_header(lua_State *l)
{
gboolean croak_if_error;
const gchar *s;
GSList *opts;
CURLcode cc;
_quvi_t q;
/* quvi handle */
q = (_quvi_t) l_get_reg_userdata(l, USERDATA_QUVI_T);
g_assert(q != NULL);
/* arg1 */
s = luaL_checkstring(l, 1);
lua_pop(l, 1);
/* options */
opts = l_quvi_object_opts_new(l, 2);
croak_if_error = l_quvi_object_opts_croak_if_error(l, opts);
l_quvi_object_opts_free(opts);
/* apply */
if (strlen(s) >0)
{
CURL *c = q->handle.curl;
q->http.headers = curl_slist_append(q->http.headers, s);
cc = curl_easy_setopt(c, CURLOPT_HTTPHEADER, q->http.headers);
}
else
cc = c_reset_headers(q);
if (cc != CURLE_OK)
{
g_string_printf(q->status.errmsg, "%s", curl_easy_strerror(cc));
q->status.rc = QUVI_ERROR_CALLBACK;
if (croak_if_error == TRUE)
luaL_error(l, "%s", q->status.errmsg->str);
}
/* Return a table of results. */
lua_newtable(l);
l_setfield_s(l, QO_ERROR_MESSAGE, q->status.errmsg->str, -1);
l_setfield_n(l, QO_QUVI_CODE, q->status.rc);
return (1); /* no. of returned values (a table) */
}
gint l_quvi_http_cookie(lua_State *l)
{
struct _cookie_opts_s co;
gboolean croak_if_error;
CURLoption copt;
GSList *opts;
_quvi_t q;
/* quvi handle */
q = (_quvi_t) l_get_reg_userdata(l, USERDATA_QUVI_T);
g_assert(q != NULL);
if (q->opt.allow_cookies == QUVI_FALSE)
return (_ret(l, q)); /* Do nothing if cookies have been disabled. */
/* arg1 */
memset(&co, 0, sizeof(struct _cookie_opts_s));
co.s = luaL_checkstring(l, 1);
lua_pop(l, 1);
/* options */
opts = l_quvi_object_opts_new(l, 2);
croak_if_error = l_quvi_object_opts_croak_if_error(l, opts);
_chk_cookie_opts(l, opts, &co);
l_quvi_object_opts_free(opts);
/* mode */
switch (co.mode)
{
case COOKIE_MODE_SESSION:
copt = CURLOPT_COOKIESESSION;
break;
case COOKIE_MODE_FILE:
copt = CURLOPT_COOKIEFILE;
break;
case COOKIE_MODE_LIST:
copt = CURLOPT_COOKIELIST;
break;
case COOKIE_MODE_JAR:
copt = CURLOPT_COOKIEJAR;
break;
default:
g_string_printf(q->status.errmsg,
"[%s] invalid cookie function `0x%02x'",
__func__, co.mode);
q->status.rc = QUVI_ERROR_CALLBACK;
copt = CURLOPT_COOKIESESSION;
g_warning("%s", q->status.errmsg->str);
}
return (_setopt(l, q, copt, &co, croak_if_error));
}
