1 /*
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
3 *
4 * Copyright 2022 Olaf Wintermann. All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are met:
8 *
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 *
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29 #include "service.h"
30
31 #include <stdio.h>
32 #include <string.h>
33
34 #include "resource.h"
35
36 int pg_query(pblock *pb, Session *sn, Request *rq) {
37 char *respool = pblock_findval("resource", pb);
38 char *query = pblock_findval("query", pb);
39
40 if(!respool) {
41 log_ereport(LOG_MISCONFIG, "pg-query: missing resource parameter");
42 return REQ_ABORTED;
43 }
44 if(!query) {
45 log_ereport(LOG_MISCONFIG, "pg-query: missing query parameter");
46 return REQ_ABORTED;
47 }
48
49 // Get the named resource and check if the PG connection works
50 ResourceData *res = resourcepool_lookup(sn, rq, respool, 0);
51 if(!res) {
52 log_ereport(LOG_FAILURE, "pg-query: cannot get resource ''%s''", respool);
53 return REQ_ABORTED;
54 }
55
56 PGconn *connection = res->data;
57 if(pg_check_connection(LOG_FAILURE, respool, connection)) {
58 return REQ_ABORTED;
59 }
60
61 // execute query
62 PGresult *result = PQexec(connection, query);
63 if(!result) return REQ_ABORTED;
64
65 // start response
66 pblock_remove("content-type", rq->srvhdrs);
67 pblock_nvinsert("content-type", "text/html", rq->srvhdrs);
68
69 protocol_status(sn, rq, 200, NULL);
70 http_start_response(sn, rq);
71
72 // (html) header
73 net_printf(sn->csd, "%s", "<!DOCTYPE html>\n<html><head></head><body>\n");
74
75 int nfields = PQnfields(result);
76 if(nfields > 0) {
77 net_printf(sn->csd, "<table>\n<tr>\n");
78 for(int i=0;i<nfields;i++) {
79 char *fieldName = PQfname(result, i);
80 char *fieldNameEscaped = util_html_escape(fieldName);
81 if(fieldNameEscaped) {
82 net_printf(sn->csd, "<th>%s</th>\n", fieldNameEscaped);
83 FREE(fieldNameEscaped);
84 }
85 }
86 net_printf(sn->csd, "</tr>\n");
87
88 int nrows = PQntuples(result);
89 for(int r=0;r<nrows;r++) {
90 net_printf(sn->csd, "<tr>\n");
91 for(int c=0;c<nfields;c++) {
92 char *fieldValue = PQgetvalue(result, r, c);
93 char *fieldValueEscaped = util_html_escape(fieldValue);
94 if(fieldValueEscaped) {
95 net_printf(sn->csd, "<td>%s</td>\n", fieldValueEscaped);
96 FREE(fieldValueEscaped);
97 }
98 }
99 net_printf(sn->csd, "</tr>\n");
100 }
101
102
103 net_printf(sn->csd, "</table>\n");
104 } else {
105 net_printf(sn->csd, "<p>%s</p>\n", PQresStatus(PQresultStatus(result)));
106 }
107
108 net_printf(sn->csd, "%s", "</body></html>\n");
109
110 PQclear(result);
111
112 return REQ_PROCEED;
113 }
114