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

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000-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. /*
  14.   Low level functions for storing data to be send to the MySQL client
  15.   The actual communction is handled by the net_xxx functions in net_serv.cc
  16. */
  17. #ifdef USE_PRAGMA_IMPLEMENTATION
  18. #pragma implementation // gcc: Class implementation
  19. #endif
  20. #include "mysql_priv.h"
  21. #include <mysql.h>
  22. bool Protocol_cursor::send_fields(List<Item> *list, uint flag)
  23. {
  24.   List_iterator_fast<Item> it(*list);
  25.   Item                     *item;
  26.   MYSQL_FIELD              *client_field;
  27.   
  28.   DBUG_ENTER("send_fields");
  29.   if (prepare_for_send(list))
  30.       return FALSE;
  31.   fields= (MYSQL_FIELD *)alloc_root(alloc, sizeof(MYSQL_FIELD) * field_count);
  32.   if (!fields)
  33.     goto err;
  34.   client_field= fields;
  35.   while ((item= it++))
  36.   {
  37.     Send_field server_field;
  38.     item->make_field(&server_field);
  39.     client_field->db=   strdup_root(alloc, server_field.db_name);
  40.     client_field->table=  strdup_root(alloc, server_field.table_name);
  41.     client_field->name=   strdup_root(alloc, server_field.col_name);
  42.     client_field->org_table= strdup_root(alloc, server_field.org_table_name);
  43.     client_field->org_name=  strdup_root(alloc, server_field.org_col_name);
  44.     client_field->length= server_field.length;
  45.     client_field->type=   server_field.type;
  46.     client_field->flags= server_field.flags;
  47.     client_field->decimals= server_field.decimals;
  48.     client_field->db_length= strlen(client_field->db);
  49.     client_field->table_length= strlen(client_field->table);
  50.     client_field->name_length= strlen(client_field->name);
  51.     client_field->org_name_length= strlen(client_field->org_name);
  52.     client_field->org_table_length= strlen(client_field->org_table);
  53.     client_field->charsetnr= server_field.charsetnr;
  54.     
  55.     if (INTERNAL_NUM_FIELD(client_field))
  56.       client_field->flags|= NUM_FLAG;
  57.     if (flag & 2)
  58.     {
  59.       char buff[80];
  60.       String tmp(buff, sizeof(buff), default_charset_info), *res;
  61.       if (!(res=item->val_str(&tmp)))
  62. client_field->def= (char*) "";
  63.       else
  64. client_field->def= strmake_root(alloc, res->ptr(), res->length());
  65.     }
  66.     else
  67.       client_field->def=0;
  68.     client_field->max_length= 0;
  69.     ++client_field;
  70.   }
  71.   DBUG_RETURN(FALSE);
  72.  err:
  73.   send_error(thd, ER_OUT_OF_RESOURCES); /* purecov: inspected */
  74.   DBUG_RETURN(TRUE); /* purecov: inspected */
  75. }
  76. /* Get the length of next field. Change parameter to point at fieldstart */
  77. bool Protocol_cursor::write()
  78. {
  79.   byte *cp= (byte *)packet->ptr();
  80.   byte *end_pos= (byte *)packet->ptr() + packet->length();
  81.   ulong len;
  82.   MYSQL_FIELD *cur_field= fields;
  83.   MYSQL_FIELD *fields_end= fields + field_count;
  84.   MYSQL_ROWS *new_record;
  85.   byte **data_tmp;
  86.   byte *to;
  87.   new_record= (MYSQL_ROWS *)alloc_root(alloc, 
  88.        sizeof(MYSQL_ROWS) + (field_count + 1)*sizeof(char *) + packet->length());
  89.   if (!new_record)
  90.     goto err;
  91.   data_tmp= (byte **)(new_record + 1);
  92.   new_record->data= (char **)data_tmp;
  93.   to= (byte *)(fields + field_count + 1);
  94.   for (; cur_field < fields_end; ++cur_field, ++data_tmp)
  95.   {
  96.     if ((len=net_field_length((uchar **)&cp)))
  97.     {
  98.       *data_tmp= 0;
  99.     }
  100.     else
  101.     {
  102.       if ((long)len > (end_pos - cp))
  103.       {
  104. // TODO error signal      send_error(thd, CR_MALFORMED_PACKET);
  105. return TRUE;
  106.       }
  107.       memcpy(to,(char*) cp,len);
  108.       to[len]=0;
  109.       to+=len+1;
  110.       cp+=len;
  111.       if (cur_field->max_length < len)
  112. cur_field->max_length=len;
  113.     }
  114.   }
  115.   *prev_record= new_record;
  116.   prev_record= &new_record->next;
  117.   new_record->next= NULL;
  118.   row_count++;
  119.   return FALSE;
  120.  err:
  121. // TODO error signal      send_error(thd, ER_OUT_OF_RESOURCES);
  122.   return TRUE;
  123. }