Add belt and braces check of PyString_AsString return.

This commit is contained in:
Andrew McNamara 2005-01-12 01:16:35 +00:00
parent 1b6f398c98
commit cf0fd5ab29

View file

@ -1098,6 +1098,7 @@ static int
join_append_lineterminator(WriterObj *self) join_append_lineterminator(WriterObj *self)
{ {
int terminator_len; int terminator_len;
char *terminator;
terminator_len = PyString_Size(self->dialect->lineterminator); terminator_len = PyString_Size(self->dialect->lineterminator);
@ -1105,10 +1106,10 @@ join_append_lineterminator(WriterObj *self)
if (!join_check_rec_size(self, self->rec_len + terminator_len)) if (!join_check_rec_size(self, self->rec_len + terminator_len))
return 0; return 0;
memmove(self->rec + self->rec_len, terminator = PyString_AsString(self->dialect->lineterminator);
/* should not be NULL */ if (terminator == NULL)
PyString_AsString(self->dialect->lineterminator), return 0;
terminator_len); memmove(self->rec + self->rec_len, terminator, terminator_len);
self->rec_len += terminator_len; self->rec_len += terminator_len;
return 1; return 1;