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;

        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);
#ifndef HAVE_MARIADB_CLIENT
                ssl_mode = SSL_MODE_VERIFY_CA;
#endif
        }

        if (mysql_ssl_set(db, NULL, NULL, DB_ca_cert, 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.5
42	sysadm	1.7	db = mysql_init(NULL);
43			if (db == NULL)
44			{
45			log_error("mysql_init() failed\n");
46			return NULL;
47			}
48
49	sysadm	1.26	fd = open(DB_ca_cert, O_RDONLY);
50			if (fd == -1)
51			{
52			if (errno != ENOENT)
53			{
54			log_error("open(%s) error: %d\n", DB_ca_cert, errno);
55			}
56			}
57			else
58			{
59			close(fd);
60			#ifndef HAVE_MARIADB_CLIENT
61			ssl_mode = SSL_MODE_VERIFY_CA;
62			#endif
63			}
64
65	sysadm	1.24	if (mysql_ssl_set(db, NULL, NULL, DB_ca_cert, NULL, NULL) != 0)
66			{
67			log_error("mysql_ssl_set() error\n");
68			return NULL;
69			}
70
71	sysadm	1.25	#ifdef HAVE_MARIADB_CLIENT
72	sysadm	1.26	if (mysql_optionsv(db, MYSQL_OPT_SSL_VERIFY_SERVER_CERT, &verify_server_cert) != 0)
73	sysadm	1.25	{
74			log_error("mysql_optionsv() error\n");
75			return NULL;
76			}
77			#else
78	sysadm	1.24	if (mysql_options(db, MYSQL_OPT_SSL_MODE, &ssl_mode) != 0)
79			{
80			log_error("mysql_options() error\n");
81			return NULL;
82			}
83	sysadm	1.25	#endif
84	sysadm	1.23
85	sysadm	1.16	if (mysql_real_connect(db, DB_host, DB_username, DB_password, DB_database,
86			0, NULL, 0) == NULL)
87	sysadm	1.7	{
88	sysadm	1.16	log_error("mysql_real_connect() error: %s\n", mysql_error(db));
89			mysql_close(db);
90	sysadm	1.7	return NULL;
91			}
92
93	sysadm	1.17	if (mysql_set_character_set(db, "utf8") != 0)
94	sysadm	1.7	{
95	sysadm	1.16	log_error("Set character set error: %s\n", mysql_error(db));
96			mysql_close(db);
97	sysadm	1.7	return NULL;
98			}
99
100	sysadm	1.12	snprintf(sql, sizeof(sql),
101	sysadm	1.16	"SET time_zone = '%s'",
102			DB_timezone);
103	sysadm	1.8
104			if (mysql_query(db, sql) != 0)
105	sysadm	1.7	{
106	sysadm	1.14	log_error("Set timezone error: %s\n", mysql_error(db));
107	sysadm	1.16	mysql_close(db);
108	sysadm	1.7	return NULL;
109			}
110	sysadm	1.6
111	sysadm	1.7	return db;
112	sysadm	1.1	}
webmaster@leafok.com	ViewVC Help
Powered by ViewVC 1.3.0-beta1