driver: tpax_get_unit_ctx(): implemented the non-recursive bits.

1 files changed, 28 insertions, 1 deletions

diff --git a/src/driver/tpax_unit_ctx.c b/src/driver/tpax_unit_ctx.c
index a6a91b6..a94bf05 100644
--- a/src/driver/tpax_unit_ctx.c
+++ b/src/driver/tpax_unit_ctx.c
@@ -8,11 +8,13 @@
 #include <stddef.h>
 #include <stdlib.h>
 #include <string.h>
-#include <sys/mman.h>
+#include <fcntl.h>
+#include <sys/stat.h>
 
 #include <tpax/tpax.h>
 #include "tpax_driver_impl.h"
 #include "tpax_errinfo_impl.h"
+#include "tpax_readlink_impl.h"
 
 static int tpax_free_unit_ctx_impl(struct tpax_unit_ctx_impl * ctx, int ret)
 {
@@ -28,6 +30,7 @@ int tpax_get_unit_ctx(
 	const char *			path,
 	struct tpax_unit_ctx **		pctx)
 {
+	int				ret;
 	struct tpax_unit_ctx_impl *	ctx;
 
 	if (!dctx)
@@ -40,8 +43,32 @@ int tpax_get_unit_ctx(
 	tpax_driver_set_ectx(
 		dctx,0,path);
 
+	if (dctx->cctx->drvflags & TPAX_DRIVER_EXEC_MODE_WRITE_COPY) {
+		ret = fstatat(
+			tpax_driver_fdcwd(dctx),path,
+			&ctx->st,AT_SYMLINK_NOFOLLOW);
+
+		if (ret < 0) {
+			free(ctx);
+			return TPAX_SYSTEM_ERROR(dctx);
+		}
+	}
+
+	if (S_ISLNK(ctx->st.st_mode)) {
+		if (tpax_readlink(
+				path,ctx->linkbuf,
+				sizeof(ctx->linkbuf)) < 0) {
+			free(ctx);
+			return TPAX_SYSTEM_ERROR(dctx);
+		}
+	}
+
 	ctx->path	= path;
+	ctx->link	= ctx->linkbuf;
+
 	ctx->uctx.path	= &ctx->path;
+	ctx->uctx.link	= &ctx->link;
+	ctx->uctx.st	= &ctx->st;
 
 	*pctx = &ctx->uctx;