[asterisk-commits] twilson: tag 1.6.2.17.1 r310984 - in /tags/1.6.2.17.1/main: manager.c tcptls.c

[SVN commits to the Asterisk project]

Author: twilson
Date: Wed Mar 16 13:42:29 2011
New Revision: 310984

URL: http://svnview.digium.com/svn/asterisk?view=rev&rev=310984
Log:
AST-2011-003 and AST-2011-004

Modified:
    tags/1.6.2.17.1/main/manager.c
    tags/1.6.2.17.1/main/tcptls.c

Modified: tags/1.6.2.17.1/main/manager.c
URL: http://svnview.digium.com/svn/asterisk/tags/1.6.2.17.1/main/manager.c?view=diff&rev=310984&r1=310983&r2=310984
==============================================================================
--- tags/1.6.2.17.1/main/manager.c (original)
+++ tags/1.6.2.17.1/main/manager.c Wed Mar 16 13:42:29 2011
@@ -228,6 +228,7 @@
 	struct mansession_session *session;
 	FILE *f;
 	int fd;
+	int write_error:1;
 };
 
 static AST_LIST_HEAD_STATIC(sessions, mansession_session);
@@ -964,11 +965,15 @@
  */
 static int send_string(struct mansession *s, char *string)
 {
-	if (s->f) {
-		return ast_careful_fwrite(s->f, s->fd, string, strlen(string), s->session->writetimeout);
-	} else {
-		return ast_careful_fwrite(s->session->f, s->session->fd, string, strlen(string), s->session->writetimeout);
-	}
+	int res;
+
+	if (s->f && (res = ast_careful_fwrite(s->f, s->fd, string, strlen(string), s->session->writetimeout))) {
+		s->write_error = 1;
+	} else if ((res = ast_careful_fwrite(s->session->f, s->session->fd, string, strlen(string), s->session->writetimeout))) {
+		s->write_error = 1;
+	}
+
+	return res;
 }
 
 /*!
@@ -3272,7 +3277,7 @@
 
 	astman_append(&s, "Asterisk Call Manager/%s\r\n", AMI_VERSION);	/* welcome prompt */
 	for (;;) {
-		if ((res = do_message(&s)) < 0)
+		if ((res = do_message(&s)) < 0 || s.write_error)
 			break;
 	}
 	/* session is over, explain why and terminate */

Modified: tags/1.6.2.17.1/main/tcptls.c
URL: http://svnview.digium.com/svn/asterisk/tags/1.6.2.17.1/main/tcptls.c?view=diff&rev=310984&r1=310983&r2=310984
==============================================================================
--- tags/1.6.2.17.1/main/tcptls.c (original)
+++ tags/1.6.2.17.1/main/tcptls.c Wed Mar 16 13:42:29 2011
@@ -139,8 +139,12 @@
 	* open a FILE * as appropriate.
 	*/
 	if (!tcptls_session->parent->tls_cfg) {
-		tcptls_session->f = fdopen(tcptls_session->fd, "w+");
-		setvbuf(tcptls_session->f, NULL, _IONBF, 0);
+		if ((tcptls_session->f = fdopen(tcptls_session->fd, "w+"))) {
+			if(setvbuf(tcptls_session->f, NULL, _IONBF, 0)) {
+				fclose(tcptls_session->f);
+				tcptls_session->f = NULL;
+			}
+		}
 	}
 #ifdef DO_SSL
 	else if ( (tcptls_session->ssl = SSL_new(tcptls_session->parent->tls_cfg->ssl_ctx)) ) {