← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to client/linebuffer.cc

  • Committer: Lee Bieber
  • Date: 2010-11-08 02:12:25 UTC
  • mfrom: (1912.1.2 build)
  • Revision ID: kalebral@gmail.com-20101108021225-7vfdp6cn81b8i1cx
Merge Brian - Cleanup counter_thread for slap, and remove the need to track the count of threads by doing a join to harvest.
Merge Andrew - fix bug 653240: drizzle command line client reads copious amounts of data into RAM before sending commands 
Merge Andrew - fix bug 672074: buffer overrun in LineBuffer

Show diffs side-by-side

added added

removed removed

Lines of Context:
client/linebuffer.cc
27
27
LineBuffer::LineBuffer(uint32_t my_max_size,FILE *my_file)
28
28
  :
29
29
    file(my_file),
30
 
    line(),
31
 
    max_size(my_max_size),
32
 
    eof(false)
 
30
    max_size(my_max_size)
33
31
{
 
32
  if (my_file)
 
33
    file_stream = new boost::iostreams::stream<boost::iostreams::file_descriptor>(fileno(my_file), true);
 
34
  else
 
35
    file_stream = new std::stringstream;
34
36
  line.reserve(max_size);
35
37
}
36
38
 
37
39
void LineBuffer::addString(const string &str)
38
40
{
39
 
  buffer << str << endl;
 
41
  (*file_stream) << str << endl;
40
42
}
41
43
 
42
44
char *LineBuffer::readline()
43
45
{
44
 
  uint32_t read_count;
45
 
 
46
 
  if (file && !eof)
47
 
  {
48
 
    if ((read_count= read(fileno(file), (&line[0]), max_size-1)))
49
 
    {
50
 
      line[read_count+1]= '\0';
51
 
      buffer << &line[0];
52
 
    }
53
 
    else
54
 
      eof= true;
55
 
  }
56
 
 
57
 
  buffer.getline(&line[0],max_size);
58
 
 
59
 
  if (buffer.eof())
 
46
  file_stream->getline(&line[0], max_size);
 
47
 
 
48
  if (file_stream->fail())
60
49
    return 0;
61
50
  else
62
51
    return &line[0];