Category

Development Platform
Home > SourceCode/Document > Database system > MySQL > View Code
SocketAuthenticator.cpp
Package: mysql-4.1.16-win-src.zip　[view]
Upload User: romrleung
Upload Date: 2022-05-23
Package Size: 18897k
Code Size: 2k
Category:

MySQL

Development Platform:

Visual C++

SocketAuthenticator.cpp：Code Content
  1. /* Copyright (C) 2003 MySQL AB
  3.    This program is free software; you can redistribute it and/or modify
  4.    it under the terms of the GNU General Public License as published by
  5.    the Free Software Foundation; either version 2 of the License, or
  6.    (at your option) any later version.
  8.    This program is distributed in the hope that it will be useful,
  9.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  10.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11.    GNU General Public License for more details.
  13.    You should have received a copy of the GNU General Public License
  14.    along with this program; if not, write to the Free Software
  15.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  18. #include <ndb_global.h>
  20. #include <SocketClient.hpp>
  21. #include <SocketAuthenticator.hpp>
  22. #include <InputStream.hpp>
  23. #include <OutputStream.hpp>
  24. #include <NdbOut.hpp>
  26. SocketAuthSimple::SocketAuthSimple(const char *username, const char *passwd) {
  27.   if (username)
  28.     m_username= strdup(username);
  29.   else
  30.     m_username= 0;
  31.   if (passwd)
  32.     m_passwd= strdup(passwd);
  33.   else
  34.     m_passwd= 0;
  35. }
  37. SocketAuthSimple::~SocketAuthSimple()
  38. {
  39.   if (m_passwd)
  40.     free((void*)m_passwd);
  41.   if (m_username)
  42.     free((void*)m_username);
  43. }
  45. bool SocketAuthSimple::client_authenticate(int sockfd)
  46. {
  47.   SocketOutputStream s_output(sockfd);
  48.   SocketInputStream  s_input(sockfd);
  50.   if (m_username)
  51.     s_output.println("%s", m_username);
  52.   else
  53.     s_output.println("");
  55.   if (m_passwd)
  56.     s_output.println("%s", m_passwd);
  57.   else
  58.     s_output.println("");
  60.   char buf[16];
  61.   if (s_input.gets(buf, 16) == 0) return false;
  62.   if (strncmp("ok", buf, 2) == 0)
  63.     return true;
  65.   return false;
  66. }
  68. bool SocketAuthSimple::server_authenticate(int sockfd)
  69. {
  71.   SocketOutputStream s_output(sockfd);
  72.   SocketInputStream  s_input(sockfd);
  74.   char buf[256];
  76.   if (s_input.gets(buf, 256) == 0) return false;
  77.   buf[255]= 0;
  78.   if (m_username)
  79.     free((void*)m_username);
  80.   m_username= strdup(buf);
  82.   if (s_input.gets(buf, 256) == 0) return false;
  83.   buf[255]= 0;
  84.   if (m_passwd)
  85.     free((void*)m_passwd);
  86.   m_passwd= strdup(buf);
  88.   s_output.println("ok");
  90.   return true;
  91. }