SocketAuthenticator.cpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:2k
源码类别:

MySQL数据库

开发平台:

Visual C++

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