← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to libdrizzle/handshake.cc

  • Committer: Brian Aker
  • Date: 2011-12-19 16:28:27 UTC
  • mto: (2478.1.1 drizzle-build)
  • mto: This revision was merged to the branch mainline in revision 2478.
  • Revision ID: brian@tangent.org-20111219162827-bfwc9zzfk10omyil
Null safety fix for libdrizzle

Show diffs side-by-side

added added

removed removed

Lines of Context:
libdrizzle/handshake.cc
104
104
  int extra_length;
105
105
  unsigned char* packet_end;
106
106
 
 
107
  if (con == NULL)
 
108
  {
 
109
    return DRIZZLE_RETURN_INVALID_ARGUMENT;
 
110
  }
107
111
  drizzle_log_debug(con->drizzle, "drizzle_state_handshake_server_read");
108
112
 
109
113
  /* Assume the entire handshake packet will fit in the buffer. */
231
235
{
232
236
  uint8_t *ptr;
233
237
 
 
238
  if (con == NULL)
 
239
  {
 
240
    return DRIZZLE_RETURN_INVALID_ARGUMENT;
 
241
  }
234
242
  drizzle_log_debug(con->drizzle, "drizzle_state_handshake_server_write");
235
243
 
236
244
  /* Calculate max packet size. */
328
336
  size_t real_size;
329
337
  uint8_t scramble_size;
330
338
 
 
339
  if (con == NULL)
 
340
  {
 
341
    return DRIZZLE_RETURN_INVALID_ARGUMENT;
 
342
  }
331
343
  drizzle_log_debug(con->drizzle, "drizzle_state_handshake_client_read");
332
344
 
333
345
  /* Assume the entire handshake packet will fit in the buffer. */
477
489
  int capabilities;
478
490
  drizzle_return_t ret;
479
491
 
 
492
  if (con == NULL)
 
493
  {
 
494
    return DRIZZLE_RETURN_INVALID_ARGUMENT;
 
495
  }
480
496
  drizzle_log_debug(con->drizzle, "drizzle_state_handshake_client_write");
481
497
 
482
498
  /* Calculate max packet size. */
566
582
 
567
583
drizzle_return_t drizzle_state_handshake_result_read(drizzle_con_st *con)
568
584
{
569
 
  drizzle_return_t ret;
 
585
  if (con == NULL)
 
586
  {
 
587
    return DRIZZLE_RETURN_INVALID_ARGUMENT;
 
588
  }
 
589
  drizzle_log_debug(con->drizzle, "drizzle_state_handshake_result_read");
 
590
 
570
591
  drizzle_result_st result;
571
 
 
572
 
  drizzle_log_debug(con->drizzle, "drizzle_state_handshake_result_read");
573
 
 
574
592
  if (drizzle_result_create(con, &result) == NULL)
 
593
  {
575
594
    return DRIZZLE_RETURN_MEMORY;
 
595
  }
576
596
 
577
597
  con->result= &result;
578
598
 
579
 
  ret= drizzle_state_result_read(con);
 
599
  drizzle_return_t ret= drizzle_state_result_read(con);
580
600
  if (drizzle_state_none(con))
581
601
  {
582
602
    if (ret == DRIZZLE_RETURN_OK)
588
608
        ret= DRIZZLE_RETURN_AUTH_FAILED;
589
609
      }
590
610
      else
 
611
      {
591
612
        con->options|= DRIZZLE_CON_READY;
 
613
      }
592
614
    }
593
615
  }
594
616
 
595
617
  drizzle_result_free(&result);
596
618
 
597
619
  if (ret == DRIZZLE_RETURN_ERROR_CODE)
 
620
  {
598
621
    return DRIZZLE_RETURN_HANDSHAKE_FAILED;
 
622
  }
599
623
 
600
624
  return ret;
601
625
}