Also, use xstrndup() instead of the custom xmemdup_add_zero() as we're
working with strings anyway.

Signed-off-by: Aleksander Morgado <aleksan...@aleksander.es>
---
 commands/Makefile      |  1 +
 commands/ratp-getenv.c | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 common/ratp.c          | 34 ----------------------------------
 3 files changed, 51 insertions(+), 34 deletions(-)
 create mode 100644 commands/ratp-getenv.c

diff --git a/commands/Makefile b/commands/Makefile
index 012a3ad68..5e5464ec3 100644
--- a/commands/Makefile
+++ b/commands/Makefile
@@ -127,3 +127,4 @@ obj-$(CONFIG_CMD_ZODIAC_PIC)        += pic.o zii-pic-esb.o \
                                zii-pic-mezz.o zii-pic-niu.o \
                                zii-pic-rdu.o zii-pic-rdu2.o
 obj-$(CONFIG_RATP)             += ratp-ping.o
+obj-$(CONFIG_RATP)             += ratp-getenv.o
diff --git a/commands/ratp-getenv.c b/commands/ratp-getenv.c
new file mode 100644
index 000000000..2daaa63d8
--- /dev/null
+++ b/commands/ratp-getenv.c
@@ -0,0 +1,50 @@
+/*
+ * Copyright (c) 2018 Sascha Hauer <s.ha...@pengutronix.de>, Pengutronix
+ *
+ * See file CREDITS for list of people who contributed to this
+ * project.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ */
+
+/*
+ * RATP getenv
+ */
+
+#include <common.h>
+#include <ratp_bb.h>
+#include <malloc.h>
+#include <environment.h>
+
+static int ratp_cmd_getenv(const struct ratp_bb *req, int req_len,
+                          struct ratp_bb **rsp, int *rsp_len)
+{
+       int dlen = req_len - sizeof (struct ratp_bb);
+       char *varname;
+       const char *value;
+
+       varname = xstrndup ((const char *)req->data, dlen);
+       value = getenv (varname);
+       free (varname);
+
+       dlen = strlen (value);
+
+       *rsp_len = sizeof(struct ratp_bb) + dlen;
+       *rsp = xzalloc(*rsp_len);
+       (*rsp)->type = cpu_to_be16(BB_RATP_TYPE_GETENV);
+       (*rsp)->flags = cpu_to_be16(BB_RATP_FLAG_RESPONSE);
+       memcpy ((*rsp)->data, value, dlen);
+       return 0;
+}
+
+BAREBOX_RATP_CMD_START(GETENV)
+       .cmd = ratp_cmd_getenv
+BAREBOX_RATP_CMD_END
diff --git a/common/ratp.c b/common/ratp.c
index a880e8e03..e1ecb314c 100644
--- a/common/ratp.c
+++ b/common/ratp.c
@@ -24,7 +24,6 @@
 #include <ratp.h>
 #include <command.h>
 #include <byteorder.h>
-#include <environment.h>
 #include <kfifo.h>
 #include <poller.h>
 #include <linux/sizes.h>
@@ -146,30 +145,6 @@ static int ratp_bb_send_command_return(struct ratp_ctx 
*ctx, uint32_t errno)
        return ret;
 }
 
-static int ratp_bb_send_getenv_return(struct ratp_ctx *ctx, const char *val)
-{
-       void *buf;
-       struct ratp_bb *rbb;
-       int len, ret;
-
-       if (!val)
-           val = "";
-
-       len = sizeof(*rbb) + strlen(val);
-       buf = xzalloc(len);
-       rbb = buf;
-       strcpy(rbb->data, val);
-
-       rbb->type = cpu_to_be16(BB_RATP_TYPE_GETENV);
-       rbb->flags = cpu_to_be16(BB_RATP_FLAG_RESPONSE);
-
-       ret = ratp_send(&ctx->ratp, buf, len);
-
-       free(buf);
-
-       return ret;
-}
-
 static char *ratp_command;
 static struct ratp_ctx *ratp_ctx;
 
@@ -355,7 +330,6 @@ static int dispatch_ratp_message(struct ratp_ctx *ctx, 
const void *buf, int len)
        const struct ratp_bb *rbb = buf;
        struct ratp_bb_pkt *pkt;
        int dlen = len - sizeof(struct ratp_bb);
-       char *varname;
        int ret = 0;
        uint16_t flags;
        uint16_t type = be16_to_cpu(rbb->type);
@@ -394,14 +368,6 @@ static int dispatch_ratp_message(struct ratp_ctx *ctx, 
const void *buf, int len)
                pr_debug("got command: %s\n", ratp_command);
                break;
 
-       case BB_RATP_TYPE_GETENV:
-               if (flags & BB_RATP_FLAG_RESPONSE)
-                       break;
-
-               varname = xmemdup_add_zero(&rbb->data, dlen);
-               ret = ratp_bb_send_getenv_return(ctx, getenv(varname));
-               break;
-
        case BB_RATP_TYPE_FS:
                /* Only responses expected */
                if (!(flags & BB_RATP_FLAG_RESPONSE))
-- 
2.15.1


_______________________________________________
barebox mailing list
barebox@lists.infradead.org
http://lists.infradead.org/mailman/listinfo/barebox

Reply via email to