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 <mysql.h>
#include <stdio.h>
#include <string.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;
        unsigned int ssl_mode = SSL_MODE_VERIFY_CA;
        char sql[SQL_BUFFER_LEN];

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

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

        if (mysql_options(db, MYSQL_OPT_SSL_MODE, &ssl_mode) != 0)
        {
                log_error("mysql_options() error\n");
                return NULL;
        }

        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.23	#include <mysql.h>
17	sysadm	1.3	#include <stdio.h>
18	sysadm	1.23	#include <string.h>
19	sysadm	1.1
20	sysadm	1.10	// Global declaration for database
21	sysadm	1.23	char DB_ca_cert[FILE_PATH_LEN] = "conf/ca_cert.pem";
22	sysadm	1.20	char DB_host[DB_host_max_len + 1];
23			char DB_username[DB_username_max_len + 1];
24			char DB_password[DB_password_max_len + 1];
25			char DB_database[DB_database_max_len + 1];
26			char DB_timezone[DB_timezone_max_len + 1];
27	sysadm	1.10
28			MYSQL *db_open()
29	sysadm	1.1	{
30	sysadm	1.16	MYSQL *db = NULL;
31	sysadm	1.24	unsigned int ssl_mode = SSL_MODE_VERIFY_CA;
32	sysadm	1.11	char sql[SQL_BUFFER_LEN];
33	sysadm	1.5
34	sysadm	1.7	db = mysql_init(NULL);
35			if (db == NULL)
36			{
37			log_error("mysql_init() failed\n");
38			return NULL;
39			}
40
41	sysadm	1.24	if (mysql_ssl_set(db, NULL, NULL, DB_ca_cert, NULL, NULL) != 0)
42			{
43			log_error("mysql_ssl_set() error\n");
44			return NULL;
45			}
46
47			if (mysql_options(db, MYSQL_OPT_SSL_MODE, &ssl_mode) != 0)
48			{
49			log_error("mysql_options() error\n");
50			return NULL;
51			}
52	sysadm	1.23
53	sysadm	1.16	if (mysql_real_connect(db, DB_host, DB_username, DB_password, DB_database,
54			0, NULL, 0) == NULL)
55	sysadm	1.7	{
56	sysadm	1.16	log_error("mysql_real_connect() error: %s\n", mysql_error(db));
57			mysql_close(db);
58	sysadm	1.7	return NULL;
59			}
60
61	sysadm	1.17	if (mysql_set_character_set(db, "utf8") != 0)
62	sysadm	1.7	{
63	sysadm	1.16	log_error("Set character set error: %s\n", mysql_error(db));
64			mysql_close(db);
65	sysadm	1.7	return NULL;
66			}
67
68	sysadm	1.12	snprintf(sql, sizeof(sql),
69	sysadm	1.16	"SET time_zone = '%s'",
70			DB_timezone);
71	sysadm	1.8
72			if (mysql_query(db, sql) != 0)
73	sysadm	1.7	{
74	sysadm	1.14	log_error("Set timezone error: %s\n", mysql_error(db));
75	sysadm	1.16	mysql_close(db);
76	sysadm	1.7	return NULL;
77			}
78	sysadm	1.6
79	sysadm	1.7	return db;
80	sysadm	1.1	}
webmaster@leafok.com	ViewVC Help
Powered by ViewVC 1.3.0-beta1