Check return value from krb5_net_read()
git-svn-id: svn://svn.h5l.se/heimdal/trunk/heimdal@3834 ec53bebd-3082-4978-b11e-865c3cabbd6b
This commit is contained in:
@@ -63,10 +63,8 @@ static struct getargs args[] = {
|
|||||||
{ "debug", 'd', arg_flag, &debug_flag,
|
{ "debug", 'd', arg_flag, &debug_flag,
|
||||||
"enable debugging"
|
"enable debugging"
|
||||||
},
|
},
|
||||||
{ "debug", 'd', arg_flag, &debug_flag,
|
{ "debug-port", 'p', arg_integer,&debug_port,
|
||||||
"port to use with debug", "port"
|
"port to use with debug", "port" },
|
||||||
},
|
|
||||||
{ "debug-port", 'p', arg_integer,&debug_port },
|
|
||||||
{ "help", 'h', arg_flag, &help_flag },
|
{ "help", 'h', arg_flag, &help_flag },
|
||||||
{ "version", 'v', arg_flag, &version_flag }
|
{ "version", 'v', arg_flag, &version_flag }
|
||||||
};
|
};
|
||||||
@@ -120,13 +118,22 @@ kadm5_server_recv(krb5_context context, krb5_auth_context ac,
|
|||||||
krb5_data in, out;
|
krb5_data in, out;
|
||||||
kadm5_ret_t ret;
|
kadm5_ret_t ret;
|
||||||
|
|
||||||
if(krb5_net_read(context, fd, buf, 4) != 4)
|
ret = krb5_net_read(context, fd, buf, 4);
|
||||||
|
if(ret == 0)
|
||||||
|
exit(1);
|
||||||
|
if(ret < 0)
|
||||||
krb5_err(context, 1, errno, "krb5_net_read");
|
krb5_err(context, 1, errno, "krb5_net_read");
|
||||||
|
if(ret != 4)
|
||||||
|
krb5_errx(context, 1, "krb5_net_read(4) = %d", ret);
|
||||||
len = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
|
len = (buf[0] << 24) | (buf[1] << 16) | (buf[2] << 8) | buf[3];
|
||||||
if(len > sizeof(buf))
|
if(len > sizeof(buf))
|
||||||
return ENOMEM;
|
return ENOMEM;
|
||||||
if(krb5_net_read(context, fd, buf, len) != len)
|
ret = krb5_net_read(context, fd, buf, len);
|
||||||
|
if(ret < 0)
|
||||||
krb5_err(context, 1, errno, "krb5_net_read");
|
krb5_err(context, 1, errno, "krb5_net_read");
|
||||||
|
if(ret != len)
|
||||||
|
krb5_errx(context, 1, "krb5_net_read(%d) = %d", len, ret);
|
||||||
|
|
||||||
in.data = buf;
|
in.data = buf;
|
||||||
in.length = len;
|
in.length = len;
|
||||||
ret = krb5_rd_priv(context, ac, &in, &out, NULL);
|
ret = krb5_rd_priv(context, ac, &in, &out, NULL);
|
||||||
|
Reference in New Issue
Block a user