src/misc.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2008 by Andrzej Rybczak   *
   3  *   electricityispower@gmail.com   *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include <cstdlib>
  22 #include <fstream>
  23 #include <sstream>
  24 #include <openssl/evp.h>
  25
  26 #include "configuration.h"
  27 #include "misc.h"
  28
  29 extern ScrobbyConfig config;
  30
  31 pthread_mutex_t log_file = PTHREAD_MUTEX_INITIALIZER;
  32
  33 size_t write_data(char *buffer, size_t size, size_t nmemb, std::string data)
  34 {
  35         size_t result = size * nmemb;
  36         data.append(buffer, result);
  37         return result;
  38 }
  39
  40 bool Daemonize()
  41 {
  42         if (daemon(0, 0) < 0)
  43                 return false;
  44
  45         std::ofstream f(config.file_pid.c_str(), std::ios_base::app);
  46         if (f.is_open())
  47         {
  48                 pid_t pid = getpid();
  49                 f << pid;
  50                 f.close();
  51                 return true;
  52         }
  53         else
  54                 return false;
  55 }
  56
  57 void ClearCache()
  58 {
  59         std::ofstream f(config.file_cache.c_str(), std::ios::trunc);
  60         f.close();
  61 }
  62
  63 void GetCachedSongs(std::vector<std::string> &v)
  64 {
  65         std::ifstream f(config.file_cache.c_str());
  66         if (f.is_open())
  67         {
  68                 std::string line;
  69                 while (!f.eof())
  70                 {
  71                         getline(f, line);
  72                         if (!line.empty())
  73                                 v.push_back(line);
  74                 }
  75         }
  76 }
  77
  78 void Cache(const std::string &s)
  79 {
  80         std::ofstream f(config.file_cache.c_str(), std::ios::app);
  81         if (f.is_open())
  82         {
  83                 f << s << std::endl;
  84                 f.close();
  85         }
  86 }
  87
  88 void Log(LogLevel ll, const char *format, ...)
  89 {
  90         if (config.log_level < ll)
  91                 return;
  92         pthread_mutex_lock(&log_file);
  93         FILE *f = fopen(config.file_log.c_str(), "a");
  94         if (!f)
  95         {
  96                 perror("Cannot open log file!\n");
  97                 exit(1);
  98         }
  99         fprintf(f, "[%s] ", DateTime().c_str());
 100         va_list list;
 101         va_start(list, format);
 102         vfprintf(f, format, list);
 103         va_end(list);
 104         fprintf(f, "\n");
 105         fclose(f);
 106         pthread_mutex_unlock(&log_file);
 107 }
 108
 109 void ignore_newlines(std::string &s)
 110 {
 111         for (size_t i = s.find("\n"); i != std::string::npos; i = s.find("\n"))
 112                 s.replace(i, 1, "");
 113 }
 114
 115 std::string md5sum(const std::string &s)
 116 {
 117         unsigned char md_value[EVP_MAX_MD_SIZE];
 118         unsigned int md_len;
 119
 120         EVP_MD_CTX mdctx;
 121         EVP_DigestInit(&mdctx, EVP_md5());
 122         EVP_DigestUpdate(&mdctx, s.c_str(), s.length());
 123         EVP_DigestFinal_ex(&mdctx, md_value, &md_len);
 124         EVP_MD_CTX_cleanup(&mdctx);
 125
 126         char result[md_len*2+1];
 127         for (unsigned i = 0; i < md_len; i++)
 128                 sprintf(&result[i*2], "%02x", md_value[i]);
 129         result[md_len*2] = 0;
 130
 131         return result;
 132 }
 133
 134 std::string DateTime()
 135 {
 136         char result[32];
 137         time_t raw;
 138         tm *t;
 139         time(&raw);
 140         t = localtime(&raw);
 141         result[strftime(result, 31, "%Y/%m/%d %X", t)] = 0;
 142         return result;
 143 }
 144
 145 int StrToInt(const std::string &s)
 146 {
 147         return atoi(s.c_str());
 148 }
 149