base/libs/qdecoder/qCgiResponse.c

Go to the documentation of this file.

/*
 * Copyright 2008 The qDecoder Project. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE QDECODER PROJECT ``AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE QDECODER PROJECT BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef DISABLE_CGI

#ifdef ENABLE_FASTCGI
#include "fcgi_stdio.h"
#else
#include <stdio.h>
#endif
#include <stdlib.h>
#include <stdbool.h>
#include <stdarg.h>
#include <string.h>
#include <fcntl.h>
#include <unistd.h>
#include <libgen.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "qDecoder.h"
#include "qInternal.h"

bool qCgiResponseSetCookie(Q_ENTRY *request, const char *name, const char *value, int expire, const char *path, const char *domain, bool secure) {
    /* check content flag */
    if (qCgiResponseGetContentType(request) != NULL) {
        DEBUG("Should be called before qCgiResponseSetContentType().");
        return false;
    }

    /* name=value */
    char *encname = qEncodeUrl(name);
    char *encvalue = qEncodeUrl(value);
    char cookie[(4 * 1024) + 256];
    snprintf(cookie, sizeof(cookie), "%s=%s", encname, encvalue);
    free(encname), free(encvalue);

    if (expire != 0) {
        char *gmtstr = qTimeGetGmtStr(time(NULL) + expire);
        strcat(cookie, "; expires=");
        strcat(cookie, gmtstr);
        free(gmtstr);
    }

    if (path != NULL) {
        if (path[0] != '/') {
            DEBUG("Path string(%s) must start with '/' character.", path);
            return false;
        }
        strcat(cookie, "; path=");
        strcat(cookie, path);
    }

    if (domain != NULL) {
        if (strstr(domain, "/") != NULL || strstr(domain, ".") == NULL) {
            DEBUG("Invalid domain name(%s).", domain);
            return false;
        }
        strcat(cookie, "; domain=");
        strcat(cookie, domain);
    }

    if (secure == true) {
        strcat(cookie, "; secure");
    }

    printf("Set-Cookie: %s\n", cookie);

    return true;
}

bool qCgiResponseRemoveCookie(Q_ENTRY *request, const char *name, const char *path, const char *domain, bool secure) {
        return qCgiResponseSetCookie(request, name, "", -1, path, domain, secure);
}

bool qCgiResponseSetContentType(Q_ENTRY *request, const char *mimetype) {
    if(qEntryGetStr(request, "_Q_CONTENTTYPE") != NULL) {
        DEBUG("alreay set.");
        return false;
    }

    printf("Content-Type: %s\n\n", mimetype);

    qEntryPutStr(request, "_Q_CONTENTTYPE", mimetype, true);
    return true;
}

const char *qCgiResponseGetContentType(Q_ENTRY *request) {
    return qEntryGetStr(request, "_Q_CONTENTTYPE");
}

bool qCgiResponseRedirect(Q_ENTRY *request, const char *uri) {
    if(qCgiResponseGetContentType(request) != NULL) {
        DEBUG("Should be called before qCgiResponseSetContentType().");
        return false;
    }

    printf("Location: %s\n\n", uri);
    return true;
}

int qCgiResponseDownload(Q_ENTRY *request, const char *filepath, const char *mimetype) {
    if (qCgiResponseGetContentType(request) != NULL) {
        DEBUG("Should be called before qCgiResponseSetContentType().");
        return -1;
    }

    int fd;
    if (filepath == NULL || (fd = open(filepath, O_RDONLY, 0)) < 0) {
        DEBUG("Can't open file.");
        return -1;
    }

    const char *mime;
    if(mimetype == NULL) mime = "application/octet-stream";
    else mime = mimetype;

    char *disposition;
    if (!strcmp(mime, "application/octet-stream")) disposition = "attachment";
    else disposition = "inline";

    char *filename = qFileGetName(filepath);

    printf("Content-Disposition: %s;filename=\"%s\"\n", disposition, filename);
    printf("Content-Transfer-Encoding: binary\n");
    printf("Accept-Ranges: bytes\n");
    printf("Content-Length: %zu\n", (size_t)qFileGetSize(filepath));
    printf("Connection: close\n");
    qCgiResponseSetContentType(request, mime);

    free(filename);

    fflush(stdout);
    int sent = qFileSend(fileno(stdout), fd, qFileGetSize(filepath));

    close(fd);
    return sent;
}
void qCgiResponseError(Q_ENTRY *request, char *format, ...) {
    char buf[MAX_LINEBUF];
    va_list arglist;

    va_start(arglist, format);
    vsnprintf(buf, sizeof(buf), format, arglist);
    va_end(arglist);

    if (getenv("REMOTE_ADDR") == NULL)  {
        printf("Error: %s\n", buf);
    } else {
        qCgiResponseSetContentType(request, "text/html");

        printf("<html>\n");
        printf("<head>\n");
        printf("<title>Error: %s</title>\n", buf);
        printf("<script language='JavaScript'>\n");
        printf("  alert(\"%s\");\n", buf);
        printf("  history.back();\n");
        printf("</script>\n");
        printf("</head>\n");
        printf("</html>\n");
    }

    qEntryFree(request);
    exit(EXIT_FAILURE);
}

#endif /* DISABLE_CGI */

---