#include "includes.h"
extern pstring global_myname;
extern uint32 global_client_caps;
#ifdef WITH_MSDFS
static BOOL parse_dfs_path(char* pathname, struct dfs_path* pdp)
{
pstring pathname_local;
char* p,*temp;
pstrcpy(pathname_local,pathname);
p = temp = pathname_local;
ZERO_STRUCTP(pdp);
trim_string(temp,"\\","\\");
DEBUG(10,("temp in parse_dfs_path: .%s. after trimming \\'s\n",temp));
p = strchr(temp,'\\');
if(p == NULL)
return False;
*p = '\0';
pstrcpy(pdp->hostname,temp);
DEBUG(10,("hostname: %s\n",pdp->hostname));
temp = p+1;
p = strchr(temp,'\\');
if(p == NULL) {
pstrcpy(pdp->servicename,temp);
pdp->reqpath[0] = '\0';
return True;
}
*p = '\0';
pstrcpy(pdp->servicename,temp);
DEBUG(10,("servicename: %s\n",pdp->servicename));
pstrcpy(pdp->reqpath, p+1);
p = pdp->reqpath;
while (*p) {
if (*p == '\\') *p = '/';
p++;
}
DEBUG(10,("rest of the path: %s\n",pdp->reqpath));
return True;
}
static BOOL create_conn_struct( connection_struct *conn, int snum)
{
ZERO_STRUCTP(conn);
conn->service = snum;
conn->connectpath = lp_pathname(snum);
if (!vfs_init(conn)) {
DEBUG(0,("create_conn_struct: vfs init failed.\n"));
return False;
}
return True;
}
static inline BOOL parse_symlink(char* buf,struct referral** preflist,
int* refcount)
{
pstring temp;
char* prot;
char* alt_path[MAX_REFERRAL_COUNT];
int count=0, i;
struct referral* reflist;
pstrcpy(temp,buf);
prot = strtok(temp,":");
if (!strequal(prot, "msdfs"))
return False;
if (!preflist)
return True;
while(((alt_path[count] = strtok(NULL,",")) != NULL) && count<MAX_REFERRAL_COUNT)
count++;
DEBUG(10,("parse_symlink: count=%d\n", count));
reflist = *preflist = (struct referral*) malloc(count * sizeof(struct referral));
if(reflist == NULL) {
DEBUG(0,("parse_symlink: Malloc failed!\n"));
return False;
}
for(i=0;i<count;i++) {
char* p = strchr(alt_path[i],'/');
if(p)
*p = '\\';
pstrcpy(reflist[i].alternate_path, "\\");
pstrcat(reflist[i].alternate_path, alt_path[i]);
reflist[i].proximity = 0;
reflist[i].ttl = REFERRAL_TTL;
DEBUG(10, ("parse_symlink: Created alt path: %s\n", reflist[i].alternate_path));
}
if(refcount)
*refcount = count;
return True;
}
BOOL is_msdfs_link(connection_struct* conn, char* path,
struct referral** reflistp, int* refcnt)
{
SMB_STRUCT_STAT st;
pstring referral;
int referral_len = 0;
if (!path || !conn)
return False;
strlower(path);
if (conn->vfs_ops.lstat(conn,dos_to_unix(path,False),&st) != 0) {
DEBUG(5,("is_msdfs_link: %s does not exist.\n",path));
return False;
}
if (S_ISLNK(st.st_mode)) {
referral_len = conn->vfs_ops.readlink(conn, path, referral,
sizeof(pstring));
if (referral_len == -1) {
DEBUG(0,("is_msdfs_link: Error reading msdfs link %s: %s\n", path, strerror(errno)));
return False;
}
referral[referral_len] = '\0';
DEBUG(5,("is_msdfs_link: %s -> %s\n",path,referral));
if (parse_symlink(referral, reflistp, refcnt))
return True;
}
return False;
}
static BOOL resolve_dfs_path(char* dfspath, struct dfs_path* dp,
connection_struct* conn,
BOOL findfirst_flag,
struct referral** reflistpp, int* refcntp,
BOOL* self_referralp, int* consumedcntp)
{
fstring localpath;
char *p;
fstring reqpath;
if (!dp || !conn) {
DEBUG(1,("resolve_dfs_path: NULL dfs_path* or NULL connection_struct*!\n"));
return False;
}
if (dp->reqpath[0] == '\0') {
if (self_referralp) {
DEBUG(6,("resolve_dfs_path: self-referral. returning False\n"));
*self_referralp = True;
}
return False;
}
fstrcpy(localpath, conn->connectpath);
fstrcat(localpath, "/");
fstrcat(localpath, dp->reqpath);
if (is_msdfs_link(conn, localpath, reflistpp, refcntp)) {
if (findfirst_flag) {
DEBUG(6,("resolve_dfs_path (FindFirst) No redirection "
"for dfs link %s.\n", dfspath));
return False;
} else {
DEBUG(6,("resolve_dfs_path: %s resolves to a valid Dfs link.\n",
dfspath));
if (consumedcntp)
*consumedcntp = strlen(dfspath);
return True;
}
}
fstrcpy(reqpath, dp->reqpath);
p = strrchr(reqpath, '/');
if (p) {
*p = '\0';
fstrcpy(localpath, conn->connectpath);
fstrcat(localpath, "/");
fstrcat(localpath, reqpath);
if (is_msdfs_link(conn, localpath, reflistpp, refcntp)) {
DEBUG(4, ("resolve_dfs_path: Redirecting %s because parent %s is dfs link\n", dfspath, localpath));
if (consumedcntp) {
char *q;
pstring buf;
safe_strcpy(buf, dfspath, sizeof(buf));
trim_string(buf, NULL, "\\");
q = strrchr(buf, '\\');
if (q)
*q = '\0';
*consumedcntp = strlen(buf);
DEBUG(10, ("resolve_dfs_path: Path consumed: %d\n", *consumedcntp));
}
return True;
}
}
return False;
}
BOOL dfs_redirect(char* pathname, connection_struct* conn,
BOOL findfirst_flag)
{
struct dfs_path dp;
if (!conn || !pathname)
return False;
parse_dfs_path(pathname, &dp);
if (!lp_msdfs_root(SNUM(conn))) {
fstrcpy(pathname, dp.reqpath);
return False;
}
if (strcasecmp(dp.servicename, lp_servicename(SNUM(conn)) ) != 0)
return False;
if (resolve_dfs_path(pathname, &dp, conn, findfirst_flag,
NULL, NULL, NULL, NULL)) {
DEBUG(3,("dfs_redirect: Redirecting %s\n", pathname));
return True;
} else {
DEBUG(3,("dfs_redirect: Not redirecting %s.\n", pathname));
fstrcpy(pathname, dp.reqpath);
DEBUG(3,("dfs_redirect: Path converted to non-dfs path %s\n",
pathname));
return False;
}
return False;
}
BOOL get_referred_path(char *pathname, struct junction_map* jn,
int* consumedcntp, BOOL* self_referralp)
{
struct dfs_path dp;
struct connection_struct conns;
struct connection_struct* conn = &conns;
int snum;
BOOL self_referral = False;
if (!pathname || !jn)
return False;
if (self_referralp)
*self_referralp = False;
else
self_referralp = &self_referral;
parse_dfs_path(pathname, &dp);
if (global_myname && (strcasecmp(global_myname, dp.hostname) != 0)) {
DEBUG(3, ("get_referred_path: Invalid hostname %s in path %s\n",
dp.hostname, pathname));
return False;
}
pstrcpy(jn->service_name, dp.servicename);
pstrcpy(jn->volume_name, dp.reqpath);
snum = lp_servicenumber(jn->service_name);
if(snum < 0)
return False;
if (!create_conn_struct(conn, snum))
return False;
if (!lp_msdfs_root(SNUM(conn))) {
DEBUG(3,("get_referred_path: .%s. in dfs path %s is not a dfs root.\n",
dp.servicename, pathname));
return False;
}
if (!resolve_dfs_path(pathname, &dp, conn, False,
&jn->referral_list, &jn->referral_count,
self_referralp, consumedcntp)) {
if (!*self_referralp) {
DEBUG(3,("get_referred_path: No valid referrals for path %s\n", pathname));
return False;
}
}
if (*self_referralp) {
struct referral* ref;
jn->referral_count = 1;
if((ref = (struct referral*) malloc(sizeof(struct referral)))
== NULL) {
DEBUG(0,("malloc failed for referral\n"));
return False;
}
pstrcpy(ref->alternate_path,pathname);
ref->proximity = 0;
ref->ttl = REFERRAL_TTL;
jn->referral_list = ref;
if (consumedcntp)
*consumedcntp = strlen(pathname);
}
return True;
}
static int setup_ver2_dfs_referral(char* pathname, char** ppdata,
struct junction_map* junction,
int consumedcnt,
BOOL self_referral)
{
char* pdata = *ppdata;
unsigned char uni_requestedpath[1024];
int uni_reqpathoffset1,uni_reqpathoffset2;
int uni_curroffset;
int requestedpathlen=0;
int offset;
int reply_size = 0;
int i=0;
DEBUG(10,("setting up version2 referral\nRequested path:\n"));
requestedpathlen = (dos_struni2(uni_requestedpath,pathname,512)+1)*2;
dump_data(10,uni_requestedpath,requestedpathlen);
DEBUG(10,("ref count = %u\n",junction->referral_count));
uni_reqpathoffset1 = REFERRAL_HEADER_SIZE +
VERSION2_REFERRAL_SIZE * junction->referral_count;
uni_reqpathoffset2 = uni_reqpathoffset1 + requestedpathlen;
uni_curroffset = uni_reqpathoffset2 + requestedpathlen;
reply_size = REFERRAL_HEADER_SIZE + VERSION2_REFERRAL_SIZE*junction->referral_count +
2 * requestedpathlen;
DEBUG(10,("reply_size: %u\n",reply_size));
for(i=0;i<junction->referral_count;i++) {
DEBUG(10,("referral %u : %s\n",i,junction->referral_list[i].alternate_path));
reply_size += (strlen(junction->referral_list[i].alternate_path)+1)*2;
}
DEBUG(10,("reply_size = %u\n",reply_size));
reply_size += 0x16;
pdata = Realloc(pdata,reply_size);
if(pdata == NULL) {
DEBUG(0,("malloc failed for Realloc!\n"));
return -1;
} else
*ppdata = pdata;
memcpy(pdata+uni_reqpathoffset1,uni_requestedpath,requestedpathlen);
memcpy(pdata+uni_reqpathoffset2,uni_requestedpath,requestedpathlen);
SSVAL(pdata,0,consumedcnt * 2);
SSVAL(pdata,2,junction->referral_count);
if(self_referral)
SIVAL(pdata,4,DFSREF_REFERRAL_SERVER | DFSREF_STORAGE_SERVER);
else
SIVAL(pdata,4,DFSREF_STORAGE_SERVER);
offset = 8;
for(i=0;i<junction->referral_count;i++) {
struct referral* ref = &(junction->referral_list[i]);
int unilen;
SSVAL(pdata,offset,2);
SSVAL(pdata,offset+2,VERSION2_REFERRAL_SIZE);
if(self_referral)
SSVAL(pdata,offset+4,1);
else
SSVAL(pdata,offset+4,0);
SSVAL(pdata,offset+6,0);
SIVAL(pdata,offset+8,ref->proximity);
SIVAL(pdata,offset+12,ref->ttl);
SSVAL(pdata,offset+16,uni_reqpathoffset1-offset);
SSVAL(pdata,offset+18,uni_reqpathoffset2-offset);
unilen = (dos_struni2(pdata+uni_curroffset,ref->alternate_path,512) +1)*2;
SSVAL(pdata,offset+20,uni_curroffset-offset);
uni_curroffset += unilen;
offset += VERSION2_REFERRAL_SIZE;
}
memset(pdata+uni_curroffset,'\0',0x16);
return reply_size;
}
static int setup_ver3_dfs_referral(char* pathname, char** ppdata,
struct junction_map* junction,
int consumedcnt,
BOOL self_referral)
{
char* pdata = *ppdata;
unsigned char uni_reqpath[1024];
int uni_reqpathoffset1, uni_reqpathoffset2;
int uni_curroffset;
int reply_size = 0;
int reqpathlen = 0;
int offset,i=0;
DEBUG(10,("setting up version3 referral\n"));
reqpathlen = (dos_struni2(uni_reqpath,pathname,512)+1)*2;
dump_data(10,uni_reqpath,reqpathlen);
uni_reqpathoffset1 = REFERRAL_HEADER_SIZE + VERSION3_REFERRAL_SIZE * junction->referral_count;
uni_reqpathoffset2 = uni_reqpathoffset1 + reqpathlen;
reply_size = uni_curroffset = uni_reqpathoffset2 + reqpathlen;
for(i=0;i<junction->referral_count;i++) {
DEBUG(10,("referral %u : %s\n",i,junction->referral_list[i].alternate_path));
reply_size += (strlen(junction->referral_list[i].alternate_path)+1)*2;
}
pdata = Realloc(pdata,reply_size);
if(pdata == NULL) {
DEBUG(0,("version3 referral setup: malloc failed for Realloc!\n"));
return -1;
} else
*ppdata = pdata;
SSVAL(pdata,0,consumedcnt * 2);
SSVAL(pdata,2,junction->referral_count);
if(self_referral)
SIVAL(pdata,4,DFSREF_REFERRAL_SERVER | DFSREF_STORAGE_SERVER);
else
SIVAL(pdata,4,DFSREF_STORAGE_SERVER);
memcpy(pdata+uni_reqpathoffset1,uni_reqpath,reqpathlen);
memcpy(pdata+uni_reqpathoffset2,uni_reqpath,reqpathlen);
offset = 8;
for(i=0;i<junction->referral_count;i++) {
struct referral* ref = &(junction->referral_list[i]);
int unilen;
SSVAL(pdata,offset,3);
SSVAL(pdata,offset+2,VERSION3_REFERRAL_SIZE);
if(self_referral)
SSVAL(pdata,offset+4,1);
else
SSVAL(pdata,offset+4,0);
SSVAL(pdata,offset+6,0);
SIVAL(pdata,offset+8,ref->ttl);
SSVAL(pdata,offset+12,uni_reqpathoffset1-offset);
SSVAL(pdata,offset+14,uni_reqpathoffset2-offset);
unilen = (dos_struni2(pdata+uni_curroffset,ref->alternate_path,512) +1)*2;
SSVAL(pdata,offset+16,uni_curroffset-offset);
memset(pdata+offset+18,'\0',16);
uni_curroffset += unilen;
offset += VERSION3_REFERRAL_SIZE;
}
return reply_size;
}
int setup_dfs_referral(char* pathname, int max_referral_level, char** ppdata)
{
struct junction_map junction;
int consumedcnt;
BOOL self_referral = False;
pstring buf;
int reply_size = 0;
ZERO_STRUCT(junction);
if (!pathname)
return -1;
safe_strcpy(buf, pathname, sizeof(buf));
if (!get_referred_path(buf, &junction, &consumedcnt,
&self_referral))
return -1;
if (!self_referral)
{
if( DEBUGLVL( 3 ) ) {
int i=0;
dbgtext("setup_dfs_referral: Path %s to alternate path(s):",pathname);
for(i=0;i<junction.referral_count;i++)
dbgtext(" %s",junction.referral_list[i].alternate_path);
dbgtext(".\n");
}
}
DEBUG(10,("max_referral_level :%d\n",max_referral_level));
if(max_referral_level<2 || max_referral_level>3)
max_referral_level = 2;
switch(max_referral_level) {
case 2:
{
reply_size = setup_ver2_dfs_referral(pathname, ppdata, &junction,
consumedcnt, self_referral);
SAFE_FREE(junction.referral_list);
break;
}
case 3:
{
reply_size = setup_ver3_dfs_referral(pathname, ppdata, &junction,
consumedcnt, self_referral);
SAFE_FREE(junction.referral_list);
break;
}
default:
{
DEBUG(0,("setup_dfs_referral: Invalid dfs referral version: %d\n", max_referral_level));
return -1;
}
}
DEBUG(10,("DFS Referral pdata:\n"));
dump_data(10,*ppdata,reply_size);
return reply_size;
}
int dfs_path_error(char* inbuf, char* outbuf)
{
return ERROR_BOTH(NT_STATUS_PATH_NOT_COVERED, ERRSRV, ERRbadpath);
}
BOOL create_junction(char* pathname, struct junction_map* jn)
{
struct dfs_path dp;
parse_dfs_path(pathname,&dp);
if(global_myname && (strcasecmp(global_myname,dp.hostname)!=0)) {
DEBUG(4,("create_junction: Invalid hostname %s in dfs path %s\n",
dp.hostname, pathname));
return False;
}
if(!lp_msdfs_root(lp_servicenumber(dp.servicename))) {
DEBUG(4,("create_junction: %s is not an msdfs root.\n",
dp.servicename));
return False;
}
pstrcpy(jn->service_name,dp.servicename);
pstrcpy(jn->volume_name,dp.reqpath);
return True;
}
static BOOL junction_to_local_path(struct junction_map* jn, char* path,
int max_pathlen, connection_struct *conn)
{
int snum;
if(!path || !jn)
return False;
snum = lp_servicenumber(jn->service_name);
if(snum < 0)
return False;
safe_strcpy(path, lp_pathname(snum), max_pathlen-1);
safe_strcat(path, "/", max_pathlen-1);
strlower(jn->volume_name);
safe_strcat(path, jn->volume_name, max_pathlen-1);
if (!create_conn_struct(conn, snum))
return False;
return True;
}
BOOL create_msdfs_link(struct junction_map* jn, BOOL exists)
{
pstring path;
pstring msdfs_link;
connection_struct conns;
connection_struct *conn = &conns;
int i=0;
BOOL insert_comma;
if(!junction_to_local_path(jn, path, sizeof(path), conn))
return False;
pstrcpy(msdfs_link, "msdfs:");
for(i=0; i<jn->referral_count; i++) {
char* refpath = jn->referral_list[i].alternate_path;
trim_string(refpath, "\\", "\\");
if(*refpath == '\0') {
if (i == 0)
insert_comma = False;
continue;
}
if (i > 0 && insert_comma)
pstrcat(msdfs_link, ",");
pstrcat(msdfs_link, refpath);
if (!insert_comma)
insert_comma = True;
}
DEBUG(5,("create_msdfs_link: Creating new msdfs link: %s -> %s\n", path, msdfs_link));
if(exists)
if(conn->vfs_ops.unlink(conn,path)!=0)
return False;
if(conn->vfs_ops.symlink(conn, msdfs_link, path) < 0) {
DEBUG(1,("create_msdfs_link: symlink failed %s -> %s\nError: %s\n",
path, msdfs_link, strerror(errno)));
return False;
}
return True;
}
BOOL remove_msdfs_link(struct junction_map* jn)
{
pstring path;
connection_struct conns;
connection_struct *conn = &conns;
if(!junction_to_local_path(jn, path, sizeof(path), conn))
return False;
if(conn->vfs_ops.unlink(conn, path)!=0)
return False;
return True;
}
static BOOL form_junctions(int snum, struct junction_map* jn, int* jn_count)
{
int cnt = *jn_count;
DIR *dirp;
char* dname;
pstring connect_path;
char* service_name = lp_servicename(snum);
connection_struct conns;
connection_struct *conn = &conns;
pstrcpy(connect_path,lp_pathname(snum));
if(*connect_path == '\0')
return False;
if (!create_conn_struct(conn, snum))
return False;
{
struct referral *ref = NULL;
pstring alt_path;
pstrcpy(jn[cnt].service_name, service_name);
jn[cnt].volume_name[0] = '\0';
jn[cnt].referral_count = 1;
slprintf(alt_path,sizeof(alt_path)-1,"\\\\%s\\%s",
global_myname, service_name);
ref = jn[cnt].referral_list = (struct referral*) malloc(sizeof(struct referral));
if (jn[cnt].referral_list == NULL) {
DEBUG(0, ("Malloc failed!\n"));
return False;
}
safe_strcpy(ref->alternate_path, alt_path, sizeof(pstring));
ref->proximity = 0;
ref->ttl = REFERRAL_TTL;
cnt++;
}
dirp = conn->vfs_ops.opendir(conn, dos_to_unix(connect_path,False));
if(!dirp)
return False;
while((dname = vfs_readdirname(conn, dirp)) != NULL) {
pstring pathreal;
pstrcpy(pathreal, connect_path);
pstrcat(pathreal, "/");
pstrcat(pathreal, dname);
if (is_msdfs_link(conn, pathreal, &(jn[cnt].referral_list),
&(jn[cnt].referral_count))) {
pstrcpy(jn[cnt].service_name, service_name);
pstrcpy(jn[cnt].volume_name, dname);
cnt++;
}
}
conn->vfs_ops.closedir(conn,dirp);
*jn_count = cnt;
return True;
}
int enum_msdfs_links(struct junction_map* jn)
{
int i=0;
int jn_count = 0;
if(!lp_host_msdfs())
return -1;
for(i=0;*lp_servicename(i);i++) {
if(lp_msdfs_root(i))
form_junctions(i,jn,&jn_count);
}
return jn_count;
}
#else
int setup_dfs_referral(char* pathname, int max_referral_level, char** ppdata)
{
return -1;
}
BOOL is_msdfs_link(connection_struct* conn, char* path,
struct referral** reflistpp, int* refcntp)
{
return False;
}
#endif