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

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult 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. #ifdef USE_PRAGMA_INTERFACE
  14. #pragma interface /* gcc class implementation */
  15. #endif
  16. class i_string;
  17. class THD;
  18. typedef struct st_mysql_field MYSQL_FIELD;
  19. typedef struct st_mysql_rows MYSQL_ROWS;
  20. class Protocol
  21. {
  22. protected:
  23.   THD  *thd;
  24.   String *packet;
  25.   String *convert;
  26.   uint field_pos;
  27. #ifndef DEBUG_OFF
  28.   enum enum_field_types *field_types;
  29. #endif
  30.   uint field_count;
  31. #ifndef EMBEDDED_LIBRARY
  32.   bool net_store_data(const char *from, uint length);
  33. #else
  34.   virtual bool net_store_data(const char *from, uint length);
  35.   char **next_field;
  36.   MYSQL_FIELD *next_mysql_field;
  37.   MEM_ROOT *alloc;
  38. #endif
  39.   bool store_string_aux(const char *from, uint length,
  40.                         CHARSET_INFO *fromcs, CHARSET_INFO *tocs);
  41. public:
  42.   Protocol() {}
  43.   Protocol(THD *thd_arg) { init(thd_arg); }
  44.   virtual ~Protocol() {}
  45.   void init(THD* thd_arg);
  46.   bool send_fields(List<Item> *list, uint flag);
  47.   bool send_records_num(List<Item> *list, ulonglong records);
  48.   bool store(I_List<i_string> *str_list);
  49.   bool store(const char *from, CHARSET_INFO *cs);
  50.   String *storage_packet() { return packet; }
  51.   inline void free() { packet->free(); }
  52. #ifndef EMBEDDED_LIBRARY
  53.   bool write();
  54. #else
  55.   virtual bool write();
  56. #endif
  57.   inline  bool store(uint32 from)
  58.   { return store_long((longlong) from); }
  59.   inline  bool store(longlong from)
  60.   { return store_longlong((longlong) from, 0); }
  61.   inline  bool store(ulonglong from)
  62.   { return store_longlong((longlong) from, 1); }
  63.   inline bool store(String *str)
  64.   { return store((char*) str->ptr(), str->length(), str->charset()); }
  65.   virtual bool prepare_for_send(List<Item> *item_list) 
  66.   {
  67.     field_count=item_list->elements;
  68.     return 0;
  69.   }
  70.   virtual bool flush();
  71.   virtual void prepare_for_resend()=0;
  72.   virtual bool store_null()=0;
  73.   virtual bool store_tiny(longlong from)=0;
  74.   virtual bool store_short(longlong from)=0;
  75.   virtual bool store_long(longlong from)=0;
  76.   virtual bool store_longlong(longlong from, bool unsigned_flag)=0;
  77.   virtual bool store(const char *from, uint length, CHARSET_INFO *cs)=0;
  78.   virtual bool store(const char *from, uint length, 
  79.         CHARSET_INFO *fromcs, CHARSET_INFO *tocs)=0;
  80.   virtual bool store(float from, uint32 decimals, String *buffer)=0;
  81.   virtual bool store(double from, uint32 decimals, String *buffer)=0;
  82.   virtual bool store(TIME *time)=0;
  83.   virtual bool store_date(TIME *time)=0;
  84.   virtual bool store_time(TIME *time)=0;
  85.   virtual bool store(Field *field)=0;
  86. };
  87. /* Class used for the old (MySQL 4.0 protocol) */
  88. class Protocol_simple :public Protocol
  89. {
  90. public:
  91.   Protocol_simple() {}
  92.   Protocol_simple(THD *thd_arg) :Protocol(thd_arg) {}
  93.   virtual void prepare_for_resend();
  94.   virtual bool store_null();
  95.   virtual bool store_tiny(longlong from);
  96.   virtual bool store_short(longlong from);
  97.   virtual bool store_long(longlong from);
  98.   virtual bool store_longlong(longlong from, bool unsigned_flag);
  99.   virtual bool store(const char *from, uint length, CHARSET_INFO *cs);
  100.   virtual bool store(const char *from, uint length,
  101.         CHARSET_INFO *fromcs, CHARSET_INFO *tocs);
  102.   virtual bool store(TIME *time);
  103.   virtual bool store_date(TIME *time);
  104.   virtual bool store_time(TIME *time);
  105.   virtual bool store(float nr, uint32 decimals, String *buffer);
  106.   virtual bool store(double from, uint32 decimals, String *buffer);
  107.   virtual bool store(Field *field);
  108. };
  109. class Protocol_prep :public Protocol
  110. {
  111. private:
  112.   uint bit_fields;
  113. public:
  114.   Protocol_prep() {}
  115.   Protocol_prep(THD *thd_arg) :Protocol(thd_arg) {}
  116.   virtual bool prepare_for_send(List<Item> *item_list);
  117.   virtual void prepare_for_resend();
  118. #ifdef EMBEDDED_LIBRARY
  119.   virtual bool write();
  120.   bool net_store_data(const char *from, uint length);
  121. #endif
  122.   virtual bool store_null();
  123.   virtual bool store_tiny(longlong from);
  124.   virtual bool store_short(longlong from);
  125.   virtual bool store_long(longlong from);
  126.   virtual bool store_longlong(longlong from, bool unsigned_flag);
  127.   virtual bool store(const char *from,uint length, CHARSET_INFO *cs);
  128.   virtual bool store(const char *from, uint length,
  129.         CHARSET_INFO *fromcs, CHARSET_INFO *tocs);
  130.   virtual bool store(TIME *time);
  131.   virtual bool store_date(TIME *time);
  132.   virtual bool store_time(TIME *time);
  133.   virtual bool store(float nr, uint32 decimals, String *buffer);
  134.   virtual bool store(double from, uint32 decimals, String *buffer);
  135.   virtual bool store(Field *field);
  136. };
  137. class Protocol_cursor :public Protocol_simple
  138. {
  139. public:
  140.   MEM_ROOT *alloc;
  141.   MYSQL_FIELD *fields;
  142.   MYSQL_ROWS *data;
  143.   MYSQL_ROWS **prev_record;
  144.   ulong row_count;
  145.   Protocol_cursor() {}
  146.   Protocol_cursor(THD *thd_arg, MEM_ROOT *ini_alloc) :Protocol_simple(thd_arg), alloc(ini_alloc) {}
  147.   bool prepare_for_send(List<Item> *item_list) 
  148.   {
  149.     fields= NULL;
  150.     data= NULL;
  151.     prev_record= &data;
  152.     return Protocol_simple::prepare_for_send(item_list);
  153.   }
  154.   bool send_fields(List<Item> *list, uint flag);
  155.   bool write();
  156. };
  157. void send_warning(THD *thd, uint sql_errno, const char *err=0);
  158. void net_printf(THD *thd,uint sql_errno, ...);
  159. void send_ok(THD *thd, ha_rows affected_rows=0L, ulonglong id=0L,
  160.      const char *info=0);
  161. void send_eof(THD *thd, bool no_flush=0);
  162. bool send_old_password_request(THD *thd);
  163. char *net_store_length(char *packet,uint length);
  164. char *net_store_data(char *to,const char *from, uint length);
  165. char *net_store_data(char *to,int32 from);
  166. char *net_store_data(char *to,longlong from);