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 <cerrno>
  22 #include <cstdlib>
  23 #include <cstring>
  24 #include <fstream>
  25 #include <iostream>
  26 #include <sstream>
  27 #include <openssl/evp.h>
  28 #include <pwd.h>
  29
  30 #include "configuration.h"
  31 #include "misc.h"
  32
  33 extern ScrobbyConfig config;
  34
  35 pthread_mutex_t log_file = PTHREAD_MUTEX_INITIALIZER;
  36
  37 size_t write_data(char *buffer, size_t size, size_t nmemb, std::string data)
  38 {
  39         size_t result = size * nmemb;
  40         data.append(buffer, result);
  41         return result;
  42 }
  43
  44 void ChangeToUser()
  45 {
  46         if (config.dedicated_user.empty() || getuid() != 0)
  47                 return;
  48
  49         passwd *userinfo;
  50         if (!(userinfo = getpwnam(config.dedicated_user.c_str())))
  51         {
  52                 std::cerr << "user " << config.dedicated_user << " not found!\n";
  53                 exit(1);
  54         }
  55         if (setgid(userinfo->pw_gid) == -1)
  56         {
  57                 std::cerr << "cannot set gid for user " << config.dedicated_user << ": " << strerror(errno) << std::endl;
  58                 exit(1);
  59         }
  60         if (setuid(userinfo->pw_uid) == -1)
  61         {
  62                 std::cerr << "cannot change to uid of user " << config.dedicated_user << ": " << strerror(errno) << std::endl;
  63                 exit(1);
  64         }
  65 }
  66
  67 bool Daemonize()
  68 {
  69         if (daemon(0, 0) < 0)
  70                 return false;
  71
  72         std::ofstream f(config.file_pid.c_str(), std::ios_base::app);
  73         if (f.is_open())
  74         {
  75                 pid_t pid = getpid();
  76                 f << pid;
  77                 f.close();
  78                 return true;
  79         }
  80         else
  81                 return false;
  82 }
  83
  84 void ClearCache()
  85 {
  86         std::ofstream f(config.file_cache.c_str(), std::ios::trunc);
  87         f.close();
  88 }
  89
  90 void GetCachedSongs(std::vector<std::string> &v)
  91 {
  92         std::ifstream f(config.file_cache.c_str());
  93         if (f.is_open())
  94         {
  95                 std::string line;
  96                 while (!f.eof())
  97                 {
  98                         getline(f, line);
  99                         if (!line.empty())
 100                                 v.push_back(line);
 101                 }
 102         }
 103 }
 104
 105 void Cache(const std::string &s)
 106 {
 107         std::ofstream f(config.file_cache.c_str(), std::ios::app);
 108         if (f.is_open())
 109         {
 110                 f << s << std::endl;
 111                 f.close();
 112         }
 113 }
 114
 115 void Log(LogLevel ll, const char *format, ...)
 116 {
 117         if (config.log_level < ll)
 118                 return;
 119         pthread_mutex_lock(&log_file);
 120         FILE *f = fopen(config.file_log.c_str(), "a");
 121         if (!f)
 122         {
 123                 perror("Cannot open log file!\n");
 124                 exit(1);
 125         }
 126         fprintf(f, "[%s] ", DateTime().c_str());
 127         va_list list;
 128         va_start(list, format);
 129         vfprintf(f, format, list);
 130         va_end(list);
 131         fprintf(f, "\n");
 132         fclose(f);
 133         pthread_mutex_unlock(&log_file);
 134 }
 135
 136 void ignore_newlines(std::string &s)
 137 {
 138         for (size_t i = s.find("\n"); i != std::string::npos; i = s.find("\n"))
 139                 s.replace(i, 1, "");
 140 }
 141
 142 std::string md5sum(const std::string &s)
 143 {
 144         unsigned char md_value[EVP_MAX_MD_SIZE];
 145         unsigned int md_len;
 146
 147         EVP_MD_CTX mdctx;
 148         EVP_DigestInit(&mdctx, EVP_md5());
 149         EVP_DigestUpdate(&mdctx, s.c_str(), s.length());
 150         EVP_DigestFinal_ex(&mdctx, md_value, &md_len);
 151         EVP_MD_CTX_cleanup(&mdctx);
 152
 153         char result[md_len*2+1];
 154         for (unsigned i = 0; i < md_len; i++)
 155                 sprintf(&result[i*2], "%02x", md_value[i]);
 156         result[md_len*2] = 0;
 157
 158         return result;
 159 }
 160
 161 std::string DateTime()
 162 {
 163         char result[32];
 164         time_t raw;
 165         tm *t;
 166         time(&raw);
 167         t = localtime(&raw);
 168         result[strftime(result, 31, "%Y/%m/%d %X", t)] = 0;
 169         return result;
 170 }
 171
 172 int StrToInt(const std::string &s)
 173 {
 174         return atoi(s.c_str());
 175 }
 176