/[LeafOK_CVS]/lbbs/src/database.c
ViewVC logotype

Diff of /lbbs/src/database.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

Revision 1.6 by sysadm, Wed Sep 12 03:21:49 2007 UTC Revision 1.26 by sysadm, Wed Nov 26 14:38:27 2025 UTC
# Line 1  Line 1 
1  /***************************************************************************  /* SPDX-License-Identifier: GPL-3.0-or-later */
2                           database.c  -  description  /*
3                               -------------------   * database
4      begin                : Mon Oct 11 2004   *   - configuration and function of DB connection
5      copyright            : (C) 2004 by Leaflet   *
6      email                : leaflet@leafok.com   * Copyright (C) 2004-2025  Leaflet <leaflet@leafok.com>
7   ***************************************************************************/   */
8    
9  /***************************************************************************  #ifdef HAVE_CONFIG_H
10   *                                                                         *  #include "config.h"
11   *   This program is free software; you can redistribute it and/or modify  *  #endif
  *   it under the terms of the GNU General Public License as published by  *  
  *   the Free Software Foundation; either version 2 of the License, or     *  
  *   (at your option) any later version.                                   *  
  *                                                                         *  
  ***************************************************************************/  
12    
13  #include "common.h"  #include "common.h"
14    #include "database.h"
15    #include "log.h"
16    #include <errno.h>
17    #include <fcntl.h>
18  #include <mysql.h>  #include <mysql.h>
19  #include <stdio.h>  #include <stdio.h>
20    #include <string.h>
21    #include <sys/stat.h>
22    
23  MYSQL *  // Global declaration for database
24  db_open ()  char DB_ca_cert[FILE_PATH_LEN] = "conf/ca_cert.pem";
25  {  char DB_host[DB_host_max_len + 1];
26    MYSQL *db;  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    
31    db = mysql_init (NULL);  MYSQL *db_open()
32    if (db == NULL)  {
33      {          MYSQL *db = NULL;
34        log_error ("mysql_init() failed\n");  #ifdef HAVE_MARIADB_CLIENT
35        return NULL;          my_bool verify_server_cert = 0;
36      }  #else
37            unsigned int ssl_mode = SSL_MODE_PREFERRED;
38    db = mysql_real_connect (db, DB_host, DB_username, DB_password, DB_database,  #endif
39                             0, NULL, 0);          char sql[SQL_BUFFER_LEN];
40    if (db == NULL)          int fd;
41      {  
42        log_error ("mysql_connect() failed\n");          db = mysql_init(NULL);
43        return NULL;          if (db == NULL)
44      }          {
45                    log_error("mysql_init() failed\n");
46    if (mysql_query (db, "SET CHARACTER SET gb2312") != 0)                  return NULL;
47      {          }
48        log_error ("SET CHARACTER SET failed\n");  
49        return NULL;          fd = open(DB_ca_cert, O_RDONLY);
50      }          if (fd == -1)
51            {
52    if (mysql_query (db, "SET NAMES \'gb2312\'") != 0)                  if (errno != ENOENT)
53      {                  {
54        log_error ("SET NAMES failed\n");                          log_error("open(%s) error: %d\n", DB_ca_cert, errno);
55        return NULL;                  }
56      }          }
57            else
58            {
59                    close(fd);
60    #ifndef HAVE_MARIADB_CLIENT
61                    ssl_mode = SSL_MODE_VERIFY_CA;
62    #endif
63            }
64    
65            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    #ifdef HAVE_MARIADB_CLIENT
72            if (mysql_optionsv(db, MYSQL_OPT_SSL_VERIFY_SERVER_CERT, &verify_server_cert) != 0)
73            {
74                    log_error("mysql_optionsv() error\n");
75                    return NULL;
76            }
77    #else
78            if (mysql_options(db, MYSQL_OPT_SSL_MODE, &ssl_mode) != 0)
79            {
80                    log_error("mysql_options() error\n");
81                    return NULL;
82            }
83    #endif
84    
85            if (mysql_real_connect(db, DB_host, DB_username, DB_password, DB_database,
86                                                       0, NULL, 0) == NULL)
87            {
88                    log_error("mysql_real_connect() error: %s\n", mysql_error(db));
89                    mysql_close(db);
90                    return NULL;
91            }
92    
93            if (mysql_set_character_set(db, "utf8") != 0)
94            {
95                    log_error("Set character set error: %s\n", mysql_error(db));
96                    mysql_close(db);
97                    return NULL;
98            }
99    
100            snprintf(sql, sizeof(sql),
101                             "SET time_zone = '%s'",
102                             DB_timezone);
103    
104            if (mysql_query(db, sql) != 0)
105            {
106                    log_error("Set timezone error: %s\n", mysql_error(db));
107                    mysql_close(db);
108                    return NULL;
109            }
110    
111    return db;          return db;
112  }  }

Legend:
Removed lines/characters  
Changed lines/characters
  Added lines/characters
webmaster@leafok.com
 ViewVC Help
Powered by ViewVC 1.3.0-beta1