Age Owner Branch data TLA Line data Source code
1 : : /*-------------------------------------------------------------------------
2 : : *
3 : : * fe-gssapi-common.c
4 : : * The front-end (client) GSSAPI common code
5 : : *
6 : : * Portions Copyright (c) 1996-2024, PostgreSQL Global Development Group
7 : : * Portions Copyright (c) 1994, Regents of the University of California
8 : : *
9 : : * IDENTIFICATION
10 : : * src/interfaces/libpq/fe-gssapi-common.c
11 : : *-------------------------------------------------------------------------
12 : : */
13 : :
14 : : #include "postgres_fe.h"
15 : :
16 : : #include "fe-gssapi-common.h"
17 : :
18 : : #include "libpq-int.h"
19 : : #include "pqexpbuffer.h"
20 : :
21 : : /*
22 : : * Fetch all errors of a specific type and append to "str".
23 : : * Each error string is preceded by a space.
24 : : */
25 : : static void
1203 tgl@sss.pgh.pa.us 26 :CBC 2 : pg_GSS_error_int(PQExpBuffer str, OM_uint32 stat, int type)
27 : : {
28 : : OM_uint32 lmin_s;
29 : : gss_buffer_desc lmsg;
1838 sfrost@snowman.net 30 : 2 : OM_uint32 msg_ctx = 0;
31 : :
32 : : do
33 : : {
1203 tgl@sss.pgh.pa.us 34 [ - + ]: 2 : if (gss_display_status(&lmin_s, stat, type, GSS_C_NO_OID,
35 : : &msg_ctx, &lmsg) != GSS_S_COMPLETE)
1203 tgl@sss.pgh.pa.us 36 :UBC 0 : break;
1026 tgl@sss.pgh.pa.us 37 :CBC 2 : appendPQExpBufferChar(str, ' ');
38 : 2 : appendBinaryPQExpBuffer(str, lmsg.value, lmsg.length);
1838 sfrost@snowman.net 39 : 2 : gss_release_buffer(&lmin_s, &lmsg);
40 [ - + ]: 2 : } while (msg_ctx);
41 : 2 : }
42 : :
43 : : /*
44 : : * GSSAPI errors contain two parts; put both into conn->errorMessage.
45 : : */
46 : : void
47 : 1 : pg_GSS_error(const char *mprefix, PGconn *conn,
48 : : OM_uint32 maj_stat, OM_uint32 min_stat)
49 : : {
1203 tgl@sss.pgh.pa.us 50 : 1 : appendPQExpBuffer(&conn->errorMessage, "%s:", mprefix);
51 : 1 : pg_GSS_error_int(&conn->errorMessage, maj_stat, GSS_C_GSS_CODE);
52 : 1 : appendPQExpBufferChar(&conn->errorMessage, ':');
53 : 1 : pg_GSS_error_int(&conn->errorMessage, min_stat, GSS_C_MECH_CODE);
54 : 1 : appendPQExpBufferChar(&conn->errorMessage, '\n');
1838 sfrost@snowman.net 55 : 1 : }
56 : :
57 : : /*
58 : : * Check if we can acquire credentials at all (and yield them if so).
59 : : */
60 : : bool
1682 peter@eisentraut.org 61 : 581 : pg_GSS_have_cred_cache(gss_cred_id_t *cred_out)
62 : : {
63 : : OM_uint32 major,
64 : : minor;
1838 sfrost@snowman.net 65 : 581 : gss_cred_id_t cred = GSS_C_NO_CREDENTIAL;
66 : :
67 : 581 : major = gss_acquire_cred(&minor, GSS_C_NO_NAME, 0, GSS_C_NO_OID_SET,
68 : : GSS_C_INITIATE, &cred, NULL, NULL);
69 [ + + ]: 581 : if (major != GSS_S_COMPLETE)
70 : : {
71 : 209 : *cred_out = NULL;
72 : 209 : return false;
73 : : }
74 : 372 : *cred_out = cred;
75 : 372 : return true;
76 : : }
77 : :
78 : : /*
79 : : * Try to load service name for a connection
80 : : */
81 : : int
82 : 478 : pg_GSS_load_servicename(PGconn *conn)
83 : : {
84 : : OM_uint32 maj_stat,
85 : : min_stat;
86 : : int maxlen;
87 : : gss_buffer_desc temp_gbuf;
88 : : char *host;
89 : :
90 [ + + ]: 478 : if (conn->gtarg_nam != NULL)
91 : : /* Already taken care of - move along */
92 : 233 : return STATUS_OK;
93 : :
94 : 245 : host = PQhost(conn);
95 [ + - - + ]: 245 : if (!(host && host[0] != '\0'))
96 : : {
516 peter@eisentraut.org 97 :UBC 0 : libpq_append_conn_error(conn, "host name must be specified");
1838 sfrost@snowman.net 98 : 0 : return STATUS_ERROR;
99 : : }
100 : :
101 : : /*
102 : : * Import service principal name so the proper ticket can be acquired by
103 : : * the GSSAPI system.
104 : : */
1203 tgl@sss.pgh.pa.us 105 :CBC 245 : maxlen = strlen(conn->krbsrvname) + strlen(host) + 2;
1838 sfrost@snowman.net 106 : 245 : temp_gbuf.value = (char *) malloc(maxlen);
107 [ - + ]: 245 : if (!temp_gbuf.value)
108 : : {
516 peter@eisentraut.org 109 :UBC 0 : libpq_append_conn_error(conn, "out of memory");
1838 sfrost@snowman.net 110 : 0 : return STATUS_ERROR;
111 : : }
1838 sfrost@snowman.net 112 :CBC 245 : snprintf(temp_gbuf.value, maxlen, "%s@%s",
113 : : conn->krbsrvname, host);
114 : 245 : temp_gbuf.length = strlen(temp_gbuf.value);
115 : :
116 : 245 : maj_stat = gss_import_name(&min_stat, &temp_gbuf,
117 : : GSS_C_NT_HOSTBASED_SERVICE, &conn->gtarg_nam);
118 : 245 : free(temp_gbuf.value);
119 : :
120 [ - + ]: 245 : if (maj_stat != GSS_S_COMPLETE)
121 : : {
1838 sfrost@snowman.net 122 :UBC 0 : pg_GSS_error(libpq_gettext("GSSAPI name import error"),
123 : : conn,
124 : : maj_stat, min_stat);
125 : 0 : return STATUS_ERROR;
126 : : }
1838 sfrost@snowman.net 127 :CBC 245 : return STATUS_OK;
128 : : }
|