1 /* SPDX-License-Identifier: GPL-2.0 */ 2 /* 3 * Copyright (c) 2022 Paulo Alcantara <palcantara@suse.de> 4 */ 5 6 #ifndef _CIFS_DFS_H 7 #define _CIFS_DFS_H 8 9 #include "cifsglob.h" 10 #include "fs_context.h" 11 #include "cifs_unicode.h" 12 #include <linux/namei.h> 13 14 #define DFS_INTERLINK(v) \ 15 (((v) & DFSREF_REFERRAL_SERVER) && !((v) & DFSREF_STORAGE_SERVER)) 16 17 struct dfs_ref { 18 char *path; 19 char *full_path; 20 struct dfs_cache_tgt_list tl; 21 struct dfs_cache_tgt_iterator *tit; 22 }; 23 24 struct dfs_ref_walk { 25 struct dfs_ref *ref; 26 struct dfs_ref refs[MAX_NESTED_LINKS]; 27 }; 28 29 #define ref_walk_start(w) ((w)->refs) 30 #define ref_walk_end(w) (&(w)->refs[ARRAY_SIZE((w)->refs) - 1]) 31 #define ref_walk_cur(w) ((w)->ref) 32 #define ref_walk_descend(w) (--ref_walk_cur(w) >= ref_walk_start(w)) 33 34 #define ref_walk_tit(w) (ref_walk_cur(w)->tit) 35 #define ref_walk_empty(w) (!ref_walk_tit(w)) 36 #define ref_walk_path(w) (ref_walk_cur(w)->path) 37 #define ref_walk_fpath(w) (ref_walk_cur(w)->full_path) 38 #define ref_walk_tl(w) (&ref_walk_cur(w)->tl) 39 40 static inline struct dfs_ref_walk *ref_walk_alloc(void) 41 { 42 struct dfs_ref_walk *rw; 43 44 rw = kmalloc(sizeof(*rw), GFP_KERNEL); 45 if (!rw) 46 return ERR_PTR(-ENOMEM); 47 return rw; 48 } 49 50 static inline void ref_walk_init(struct dfs_ref_walk *rw) 51 { 52 memset(rw, 0, sizeof(*rw)); 53 ref_walk_cur(rw) = ref_walk_start(rw); 54 } 55 56 static inline void __ref_walk_free(struct dfs_ref *ref) 57 { 58 kfree(ref->path); 59 kfree(ref->full_path); 60 dfs_cache_free_tgts(&ref->tl); 61 memset(ref, 0, sizeof(*ref)); 62 } 63 64 static inline void ref_walk_free(struct dfs_ref_walk *rw) 65 { 66 struct dfs_ref *ref = ref_walk_start(rw); 67 68 for (; ref <= ref_walk_end(rw); ref++) 69 __ref_walk_free(ref); 70 kfree(rw); 71 } 72 73 static inline int ref_walk_advance(struct dfs_ref_walk *rw) 74 { 75 struct dfs_ref *ref = ref_walk_cur(rw) + 1; 76 77 if (ref > ref_walk_end(rw)) 78 return -ELOOP; 79 __ref_walk_free(ref); 80 ref_walk_cur(rw) = ref; 81 return 0; 82 } 83 84 static inline struct dfs_cache_tgt_iterator * 85 ref_walk_next_tgt(struct dfs_ref_walk *rw) 86 { 87 struct dfs_cache_tgt_iterator *tit; 88 struct dfs_ref *ref = ref_walk_cur(rw); 89 90 if (!ref->tit) 91 tit = dfs_cache_get_tgt_iterator(&ref->tl); 92 else 93 tit = dfs_cache_get_next_tgt(&ref->tl, ref->tit); 94 ref->tit = tit; 95 return tit; 96 } 97 98 static inline int ref_walk_get_tgt(struct dfs_ref_walk *rw, 99 struct dfs_info3_param *tgt) 100 { 101 zfree_dfs_info_param(tgt); 102 return dfs_cache_get_tgt_referral(ref_walk_path(rw) + 1, 103 ref_walk_tit(rw), tgt); 104 } 105 106 static inline int ref_walk_num_tgts(struct dfs_ref_walk *rw) 107 { 108 return dfs_cache_get_nr_tgts(ref_walk_tl(rw)); 109 } 110 111 static inline void ref_walk_set_tgt_hint(struct dfs_ref_walk *rw) 112 { 113 dfs_cache_noreq_update_tgthint(ref_walk_path(rw) + 1, 114 ref_walk_tit(rw)); 115 } 116 117 struct dfs_root_ses { 118 struct list_head list; 119 struct cifs_ses *ses; 120 }; 121 122 int dfs_parse_target_referral(const char *full_path, const struct dfs_info3_param *ref, 123 struct smb3_fs_context *ctx); 124 int dfs_mount_share(struct cifs_mount_ctx *mnt_ctx, bool *isdfs); 125 126 static inline char *dfs_get_path(struct cifs_sb_info *cifs_sb, const char *path) 127 { 128 return dfs_cache_canonical_path(path, cifs_sb->local_nls, cifs_remap(cifs_sb)); 129 } 130 131 static inline int dfs_get_referral(struct cifs_mount_ctx *mnt_ctx, const char *path, 132 struct dfs_info3_param *ref, struct dfs_cache_tgt_list *tl) 133 { 134 struct smb3_fs_context *ctx = mnt_ctx->fs_ctx; 135 struct cifs_sb_info *cifs_sb = mnt_ctx->cifs_sb; 136 137 return dfs_cache_find(mnt_ctx->xid, ctx->dfs_root_ses, cifs_sb->local_nls, 138 cifs_remap(cifs_sb), path, ref, tl); 139 } 140 141 static inline void dfs_put_root_smb_sessions(struct list_head *head) 142 { 143 struct dfs_root_ses *root, *tmp; 144 145 list_for_each_entry_safe(root, tmp, head, list) { 146 list_del_init(&root->list); 147 cifs_put_smb_ses(root->ses); 148 kfree(root); 149 } 150 } 151 152 #endif /* _CIFS_DFS_H */ 153