return 0;
 }
 
+#ifdef CONFIG_NFS_DISABLE_UDP_SUPPORT
+static bool nfs_server_transport_udp_invalid(const struct nfs_fs_context *ctx)
+{
+       return true;
+}
+#else
+static bool nfs_server_transport_udp_invalid(const struct nfs_fs_context *ctx)
+{
+       if (ctx->version == 4)
+               return true;
+       return false;
+}
+#endif
+
 /*
  * Sanity check the NFS transport protocol.
- *
  */
-static void nfs_validate_transport_protocol(struct nfs_fs_context *ctx)
+static int nfs_validate_transport_protocol(struct fs_context *fc,
+                                          struct nfs_fs_context *ctx)
 {
        switch (ctx->nfs_server.protocol) {
        case XPRT_TRANSPORT_UDP:
+               if (nfs_server_transport_udp_invalid(ctx))
+                       goto out_invalid_transport_udp;
+               break;
        case XPRT_TRANSPORT_TCP:
        case XPRT_TRANSPORT_RDMA:
                break;
        default:
                ctx->nfs_server.protocol = XPRT_TRANSPORT_TCP;
        }
+       return 0;
+out_invalid_transport_udp:
+       return nfs_invalf(fc, "NFS: Unsupported transport protocol udp");
 }
 
 /*
  */
 static void nfs_set_mount_transport_protocol(struct nfs_fs_context *ctx)
 {
-       nfs_validate_transport_protocol(ctx);
-
        if (ctx->mount_server.protocol == XPRT_TRANSPORT_UDP ||
            ctx->mount_server.protocol == XPRT_TRANSPORT_TCP)
                        return;
        struct nfs_fh *mntfh = ctx->mntfh;
        struct sockaddr *sap = (struct sockaddr *)&ctx->nfs_server.address;
        int extra_flags = NFS_MOUNT_LEGACY_INTERFACE;
+       int ret;
 
        if (data == NULL)
                goto out_no_data;
                goto generic;
        }
 
+       ret = nfs_validate_transport_protocol(fc, ctx);
+       if (ret)
+               return ret;
+
        ctx->skip_reconfig_option_check = true;
        return 0;
 
 {
        struct nfs_fs_context *ctx = nfs_fc2context(fc);
        struct sockaddr *sap = (struct sockaddr *)&ctx->nfs_server.address;
+       int ret;
        char *c;
 
        if (!data) {
        ctx->acdirmin   = data->acdirmin;
        ctx->acdirmax   = data->acdirmax;
        ctx->nfs_server.protocol = data->proto;
-       nfs_validate_transport_protocol(ctx);
-       if (ctx->nfs_server.protocol == XPRT_TRANSPORT_UDP)
-               goto out_invalid_transport_udp;
+       ret = nfs_validate_transport_protocol(fc, ctx);
+       if (ret)
+               return ret;
 done:
        ctx->skip_reconfig_option_check = true;
        return 0;
 
 out_no_address:
        return nfs_invalf(fc, "NFS4: mount program didn't pass remote address");
-
-out_invalid_transport_udp:
-       return nfs_invalf(fc, "NFS: Unsupported transport protocol udp");
 }
 #endif
 
        if (!nfs_verify_server_address(sap))
                goto out_no_address;
 
+       ret = nfs_validate_transport_protocol(fc, ctx);
+       if (ret)
+               return ret;
+
        if (ctx->version == 4) {
                if (IS_ENABLED(CONFIG_NFS_V4)) {
                        if (ctx->nfs_server.protocol == XPRT_TRANSPORT_RDMA)
                                port = NFS_PORT;
                        max_namelen = NFS4_MAXNAMLEN;
                        max_pathlen = NFS4_MAXPATHLEN;
-                       nfs_validate_transport_protocol(ctx);
-                       if (ctx->nfs_server.protocol == XPRT_TRANSPORT_UDP)
-                               goto out_invalid_transport_udp;
                        ctx->flags &= ~(NFS_MOUNT_NONLM | NFS_MOUNT_NOACL |
                                        NFS_MOUNT_VER3 | NFS_MOUNT_LOCAL_FLOCK |
                                        NFS_MOUNT_LOCAL_FCNTL);
                }
        } else {
                nfs_set_mount_transport_protocol(ctx);
-#ifdef CONFIG_NFS_DISABLE_UDP_SUPPORT
-              if (ctx->nfs_server.protocol == XPRT_TRANSPORT_UDP)
-                      goto out_invalid_transport_udp;
-#endif
                if (ctx->nfs_server.protocol == XPRT_TRANSPORT_RDMA)
                        port = NFS_RDMA_PORT;
        }
 out_v4_not_compiled:
        nfs_errorf(fc, "NFS: NFSv4 is not compiled into kernel");
        return -EPROTONOSUPPORT;
-out_invalid_transport_udp:
-       return nfs_invalf(fc, "NFS: Unsupported transport protocol udp");
 out_no_address:
        return nfs_invalf(fc, "NFS: mount program didn't pass remote address");
 out_mountproto_mismatch: