lbbs/src/database.c

/* SPDX-License-Identifier: GPL-3.0-or-later */
/*
 * database
 *   - configuration and function of DB connection
 *
 * Copyright (C) 2004-2025  Leaflet <leaflet@leafok.com>
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "common.h"
#include "database.h"
#include "log.h"
#include <errno.h>
#include <fcntl.h>
#include <mysql.h>
#include <stdio.h>
#include <string.h>
#include <sys/stat.h>

// Global declaration for database
char DB_ca_cert[FILE_PATH_LEN] = "conf/ca_cert.pem";
char DB_host[DB_host_max_len + 1];
char DB_username[DB_username_max_len + 1];
char DB_password[DB_password_max_len + 1];
char DB_database[DB_database_max_len + 1];
char DB_timezone[DB_timezone_max_len + 1];

MYSQL *db_open()
{
        MYSQL *db = NULL;
#ifdef HAVE_MARIADB_CLIENT
        my_bool verify_server_cert = 0;
#else
        unsigned int ssl_mode = SSL_MODE_PREFERRED;
#endif
        char sql[SQL_BUFFER_LEN];
        int fd;
        int have_ca_cert = 0;

        db = mysql_init(NULL);
        if (db == NULL)
        {
                log_error("mysql_init() failed\n");
                return NULL;
        }

        fd = open(DB_ca_cert, O_RDONLY);
        if (fd == -1)
        {
                if (errno != ENOENT)
                {
                        log_error("open(%s) error: %d\n", DB_ca_cert, errno);
                }
        }
        else
        {
                close(fd);
                have_ca_cert = 1;
#ifndef HAVE_MARIADB_CLIENT
                ssl_mode = SSL_MODE_VERIFY_CA;
#endif
        }

        if (mysql_ssl_set(db, NULL, NULL, (have_ca_cert ? DB_ca_cert : NULL), NULL, NULL) != 0)
        {
                log_error("mysql_ssl_set() error\n");
                return NULL;
        }

#ifdef HAVE_MARIADB_CLIENT
        if (mysql_optionsv(db, MYSQL_OPT_SSL_VERIFY_SERVER_CERT, &verify_server_cert) != 0)
        {
                log_error("mysql_optionsv() error\n");
                return NULL;
        }
#else
        if (mysql_options(db, MYSQL_OPT_SSL_MODE, &ssl_mode) != 0)
        {
                log_error("mysql_options() error\n");
                return NULL;
        }
#endif

        if (mysql_real_connect(db, DB_host, DB_username, DB_password, DB_database,
                                                   0, NULL, 0) == NULL)
        {
                log_error("mysql_real_connect() error: %s\n", mysql_error(db));
                mysql_close(db);
                return NULL;
        }

        if (mysql_set_character_set(db, "utf8") != 0)
        {
                log_error("Set character set error: %s\n", mysql_error(db));
                mysql_close(db);
                return NULL;
        }

        snprintf(sql, sizeof(sql),
                         "SET time_zone = '%s'",
                         DB_timezone);

        if (mysql_query(db, sql) != 0)
        {
                log_error("Set timezone error: %s\n", mysql_error(db));
                mysql_close(db);
                return NULL;
        }

        return db;
}
1	sysadm	1.18	/* SPDX-License-Identifier: GPL-3.0-or-later */
2			/*
3			* database
4			* - configuration and function of DB connection
5			*
6	sysadm	1.19	* Copyright (C) 2004-2025 Leaflet <leaflet@leafok.com>
7	sysadm	1.18	*/
8	sysadm	1.1
9	sysadm	1.21	#ifdef HAVE_CONFIG_H
10			#include "config.h"
11			#endif
12
13	sysadm	1.1	#include "common.h"
14	sysadm	1.11	#include "database.h"
15	sysadm	1.9	#include "log.h"
16	sysadm	1.26	#include <errno.h>
17			#include <fcntl.h>
18	sysadm	1.23	#include <mysql.h>
19	sysadm	1.3	#include <stdio.h>
20	sysadm	1.23	#include <string.h>
21	sysadm	1.26	#include <sys/stat.h>
22	sysadm	1.1
23	sysadm	1.10	// Global declaration for database
24	sysadm	1.23	char DB_ca_cert[FILE_PATH_LEN] = "conf/ca_cert.pem";
25	sysadm	1.20	char DB_host[DB_host_max_len + 1];
26			char DB_username[DB_username_max_len + 1];
27			char DB_password[DB_password_max_len + 1];
28			char DB_database[DB_database_max_len + 1];
29			char DB_timezone[DB_timezone_max_len + 1];
30	sysadm	1.10
31			MYSQL *db_open()
32	sysadm	1.1	{
33	sysadm	1.16	MYSQL *db = NULL;
34	sysadm	1.25	#ifdef HAVE_MARIADB_CLIENT
35	sysadm	1.26	my_bool verify_server_cert = 0;
36	sysadm	1.25	#else
37	sysadm	1.26	unsigned int ssl_mode = SSL_MODE_PREFERRED;
38	sysadm	1.25	#endif
39	sysadm	1.11	char sql[SQL_BUFFER_LEN];
40	sysadm	1.26	int fd;
41	sysadm	1.27	int have_ca_cert = 0;
42	sysadm	1.5
43	sysadm	1.7	db = mysql_init(NULL);
44			if (db == NULL)
45			{
46			log_error("mysql_init() failed\n");
47			return NULL;
48			}
49
50	sysadm	1.26	fd = open(DB_ca_cert, O_RDONLY);
51			if (fd == -1)
52			{
53			if (errno != ENOENT)
54			{
55			log_error("open(%s) error: %d\n", DB_ca_cert, errno);
56			}
57			}
58			else
59			{
60			close(fd);
61	sysadm	1.27	have_ca_cert = 1;
62	sysadm	1.26	#ifndef HAVE_MARIADB_CLIENT
63			ssl_mode = SSL_MODE_VERIFY_CA;
64			#endif
65			}
66
67	sysadm	1.27	if (mysql_ssl_set(db, NULL, NULL, (have_ca_cert ? DB_ca_cert : NULL), NULL, NULL) != 0)
68	sysadm	1.24	{
69			log_error("mysql_ssl_set() error\n");
70			return NULL;
71			}
72
73	sysadm	1.25	#ifdef HAVE_MARIADB_CLIENT
74	sysadm	1.26	if (mysql_optionsv(db, MYSQL_OPT_SSL_VERIFY_SERVER_CERT, &verify_server_cert) != 0)
75	sysadm	1.25	{
76			log_error("mysql_optionsv() error\n");
77			return NULL;
78			}
79			#else
80	sysadm	1.24	if (mysql_options(db, MYSQL_OPT_SSL_MODE, &ssl_mode) != 0)
81			{
82			log_error("mysql_options() error\n");
83			return NULL;
84			}
85	sysadm	1.25	#endif
86	sysadm	1.23
87	sysadm	1.16	if (mysql_real_connect(db, DB_host, DB_username, DB_password, DB_database,
88			0, NULL, 0) == NULL)
89	sysadm	1.7	{
90	sysadm	1.16	log_error("mysql_real_connect() error: %s\n", mysql_error(db));
91			mysql_close(db);
92	sysadm	1.7	return NULL;
93			}
94
95	sysadm	1.17	if (mysql_set_character_set(db, "utf8") != 0)
96	sysadm	1.7	{
97	sysadm	1.16	log_error("Set character set error: %s\n", mysql_error(db));
98			mysql_close(db);
99	sysadm	1.7	return NULL;
100			}
101
102	sysadm	1.12	snprintf(sql, sizeof(sql),
103	sysadm	1.16	"SET time_zone = '%s'",
104			DB_timezone);
105	sysadm	1.8
106			if (mysql_query(db, sql) != 0)
107	sysadm	1.7	{
108	sysadm	1.14	log_error("Set timezone error: %s\n", mysql_error(db));
109	sysadm	1.16	mysql_close(db);
110	sysadm	1.7	return NULL;
111			}
112	sysadm	1.6
113	sysadm	1.7	return db;
114	sysadm	1.1	}
webmaster@leafok.com	ViewVC Help
Powered by ViewVC 1.3.0-beta1