// SPDX-License-Identifier: GPL-2.0-or-later 
 | 
/* AFS cell alias detection 
 | 
 * 
 | 
 * Copyright (C) 2020 Red Hat, Inc. All Rights Reserved. 
 | 
 * Written by David Howells (dhowells@redhat.com) 
 | 
 */ 
 | 
  
 | 
#include <linux/slab.h> 
 | 
#include <linux/sched.h> 
 | 
#include <linux/namei.h> 
 | 
#include <keys/rxrpc-type.h> 
 | 
#include "internal.h" 
 | 
  
 | 
/* 
 | 
 * Sample a volume. 
 | 
 */ 
 | 
static struct afs_volume *afs_sample_volume(struct afs_cell *cell, struct key *key, 
 | 
                        const char *name, unsigned int namelen) 
 | 
{ 
 | 
    struct afs_volume *volume; 
 | 
    struct afs_fs_context fc = { 
 | 
        .type        = 0, /* Explicitly leave it to the VLDB */ 
 | 
        .volnamesz    = namelen, 
 | 
        .volname    = name, 
 | 
        .net        = cell->net, 
 | 
        .cell        = cell, 
 | 
        .key        = key, /* This might need to be something */ 
 | 
    }; 
 | 
  
 | 
    volume = afs_create_volume(&fc); 
 | 
    _leave(" = %p", volume); 
 | 
    return volume; 
 | 
} 
 | 
  
 | 
/* 
 | 
 * Compare two addresses. 
 | 
 */ 
 | 
static int afs_compare_addrs(const struct sockaddr_rxrpc *srx_a, 
 | 
                 const struct sockaddr_rxrpc *srx_b) 
 | 
{ 
 | 
    short port_a, port_b; 
 | 
    int addr_a, addr_b, diff; 
 | 
  
 | 
    diff = (short)srx_a->transport_type - (short)srx_b->transport_type; 
 | 
    if (diff) 
 | 
        goto out; 
 | 
  
 | 
    switch (srx_a->transport_type) { 
 | 
    case AF_INET: { 
 | 
        const struct sockaddr_in *a = &srx_a->transport.sin; 
 | 
        const struct sockaddr_in *b = &srx_b->transport.sin; 
 | 
        addr_a = ntohl(a->sin_addr.s_addr); 
 | 
        addr_b = ntohl(b->sin_addr.s_addr); 
 | 
        diff = addr_a - addr_b; 
 | 
        if (diff == 0) { 
 | 
            port_a = ntohs(a->sin_port); 
 | 
            port_b = ntohs(b->sin_port); 
 | 
            diff = port_a - port_b; 
 | 
        } 
 | 
        break; 
 | 
    } 
 | 
  
 | 
    case AF_INET6: { 
 | 
        const struct sockaddr_in6 *a = &srx_a->transport.sin6; 
 | 
        const struct sockaddr_in6 *b = &srx_b->transport.sin6; 
 | 
        diff = memcmp(&a->sin6_addr, &b->sin6_addr, 16); 
 | 
        if (diff == 0) { 
 | 
            port_a = ntohs(a->sin6_port); 
 | 
            port_b = ntohs(b->sin6_port); 
 | 
            diff = port_a - port_b; 
 | 
        } 
 | 
        break; 
 | 
    } 
 | 
  
 | 
    default: 
 | 
        WARN_ON(1); 
 | 
        diff = 1; 
 | 
    } 
 | 
  
 | 
out: 
 | 
    return diff; 
 | 
} 
 | 
  
 | 
/* 
 | 
 * Compare the address lists of a pair of fileservers. 
 | 
 */ 
 | 
static int afs_compare_fs_alists(const struct afs_server *server_a, 
 | 
                 const struct afs_server *server_b) 
 | 
{ 
 | 
    const struct afs_addr_list *la, *lb; 
 | 
    int a = 0, b = 0, addr_matches = 0; 
 | 
  
 | 
    la = rcu_dereference(server_a->addresses); 
 | 
    lb = rcu_dereference(server_b->addresses); 
 | 
  
 | 
    while (a < la->nr_addrs && b < lb->nr_addrs) { 
 | 
        const struct sockaddr_rxrpc *srx_a = &la->addrs[a]; 
 | 
        const struct sockaddr_rxrpc *srx_b = &lb->addrs[b]; 
 | 
        int diff = afs_compare_addrs(srx_a, srx_b); 
 | 
  
 | 
        if (diff < 0) { 
 | 
            a++; 
 | 
        } else if (diff > 0) { 
 | 
            b++; 
 | 
        } else { 
 | 
            addr_matches++; 
 | 
            a++; 
 | 
            b++; 
 | 
        } 
 | 
    } 
 | 
  
 | 
    return addr_matches; 
 | 
} 
 | 
  
 | 
/* 
 | 
 * Compare the fileserver lists of two volumes.  The server lists are sorted in 
 | 
 * order of ascending UUID. 
 | 
 */ 
 | 
static int afs_compare_volume_slists(const struct afs_volume *vol_a, 
 | 
                     const struct afs_volume *vol_b) 
 | 
{ 
 | 
    const struct afs_server_list *la, *lb; 
 | 
    int i, a = 0, b = 0, uuid_matches = 0, addr_matches = 0; 
 | 
  
 | 
    la = rcu_dereference(vol_a->servers); 
 | 
    lb = rcu_dereference(vol_b->servers); 
 | 
  
 | 
    for (i = 0; i < AFS_MAXTYPES; i++) 
 | 
        if (la->vids[i] != lb->vids[i]) 
 | 
            return 0; 
 | 
  
 | 
    while (a < la->nr_servers && b < lb->nr_servers) { 
 | 
        const struct afs_server *server_a = la->servers[a].server; 
 | 
        const struct afs_server *server_b = lb->servers[b].server; 
 | 
        int diff = memcmp(&server_a->uuid, &server_b->uuid, sizeof(uuid_t)); 
 | 
  
 | 
        if (diff < 0) { 
 | 
            a++; 
 | 
        } else if (diff > 0) { 
 | 
            b++; 
 | 
        } else { 
 | 
            uuid_matches++; 
 | 
            addr_matches += afs_compare_fs_alists(server_a, server_b); 
 | 
            a++; 
 | 
            b++; 
 | 
        } 
 | 
    } 
 | 
  
 | 
    _leave(" = %d [um %d]", addr_matches, uuid_matches); 
 | 
    return addr_matches; 
 | 
} 
 | 
  
 | 
/* 
 | 
 * Compare root.cell volumes. 
 | 
 */ 
 | 
static int afs_compare_cell_roots(struct afs_cell *cell) 
 | 
{ 
 | 
    struct afs_cell *p; 
 | 
  
 | 
    _enter(""); 
 | 
  
 | 
    rcu_read_lock(); 
 | 
  
 | 
    hlist_for_each_entry_rcu(p, &cell->net->proc_cells, proc_link) { 
 | 
        if (p == cell || p->alias_of) 
 | 
            continue; 
 | 
        if (!p->root_volume) 
 | 
            continue; /* Ignore cells that don't have a root.cell volume. */ 
 | 
  
 | 
        if (afs_compare_volume_slists(cell->root_volume, p->root_volume) != 0) 
 | 
            goto is_alias; 
 | 
    } 
 | 
  
 | 
    rcu_read_unlock(); 
 | 
    _leave(" = 0"); 
 | 
    return 0; 
 | 
  
 | 
is_alias: 
 | 
    rcu_read_unlock(); 
 | 
    cell->alias_of = afs_use_cell(p, afs_cell_trace_use_alias); 
 | 
    return 1; 
 | 
} 
 | 
  
 | 
/* 
 | 
 * Query the new cell for a volume from a cell we're already using. 
 | 
 */ 
 | 
static int afs_query_for_alias_one(struct afs_cell *cell, struct key *key, 
 | 
                   struct afs_cell *p) 
 | 
{ 
 | 
    struct afs_volume *volume, *pvol = NULL; 
 | 
    int ret; 
 | 
  
 | 
    /* Arbitrarily pick a volume from the list. */ 
 | 
    read_seqlock_excl(&p->volume_lock); 
 | 
    if (!RB_EMPTY_ROOT(&p->volumes)) 
 | 
        pvol = afs_get_volume(rb_entry(p->volumes.rb_node, 
 | 
                           struct afs_volume, cell_node), 
 | 
                      afs_volume_trace_get_query_alias); 
 | 
    read_sequnlock_excl(&p->volume_lock); 
 | 
    if (!pvol) 
 | 
        return 0; 
 | 
  
 | 
    _enter("%s:%s", cell->name, pvol->name); 
 | 
  
 | 
    /* And see if it's in the new cell. */ 
 | 
    volume = afs_sample_volume(cell, key, pvol->name, pvol->name_len); 
 | 
    if (IS_ERR(volume)) { 
 | 
        afs_put_volume(cell->net, pvol, afs_volume_trace_put_query_alias); 
 | 
        if (PTR_ERR(volume) != -ENOMEDIUM) 
 | 
            return PTR_ERR(volume); 
 | 
        /* That volume is not in the new cell, so not an alias */ 
 | 
        return 0; 
 | 
    } 
 | 
  
 | 
    /* The new cell has a like-named volume also - compare volume ID, 
 | 
     * server and address lists. 
 | 
     */ 
 | 
    ret = 0; 
 | 
    if (pvol->vid == volume->vid) { 
 | 
        rcu_read_lock(); 
 | 
        if (afs_compare_volume_slists(volume, pvol)) 
 | 
            ret = 1; 
 | 
        rcu_read_unlock(); 
 | 
    } 
 | 
  
 | 
    afs_put_volume(cell->net, volume, afs_volume_trace_put_query_alias); 
 | 
    afs_put_volume(cell->net, pvol, afs_volume_trace_put_query_alias); 
 | 
    return ret; 
 | 
} 
 | 
  
 | 
/* 
 | 
 * Query the new cell for volumes we know exist in cells we're already using. 
 | 
 */ 
 | 
static int afs_query_for_alias(struct afs_cell *cell, struct key *key) 
 | 
{ 
 | 
    struct afs_cell *p; 
 | 
  
 | 
    _enter("%s", cell->name); 
 | 
  
 | 
    if (mutex_lock_interruptible(&cell->net->proc_cells_lock) < 0) 
 | 
        return -ERESTARTSYS; 
 | 
  
 | 
    hlist_for_each_entry(p, &cell->net->proc_cells, proc_link) { 
 | 
        if (p == cell || p->alias_of) 
 | 
            continue; 
 | 
        if (RB_EMPTY_ROOT(&p->volumes)) 
 | 
            continue; 
 | 
        if (p->root_volume) 
 | 
            continue; /* Ignore cells that have a root.cell volume. */ 
 | 
        afs_use_cell(p, afs_cell_trace_use_check_alias); 
 | 
        mutex_unlock(&cell->net->proc_cells_lock); 
 | 
  
 | 
        if (afs_query_for_alias_one(cell, key, p) != 0) 
 | 
            goto is_alias; 
 | 
  
 | 
        if (mutex_lock_interruptible(&cell->net->proc_cells_lock) < 0) { 
 | 
            afs_unuse_cell(cell->net, p, afs_cell_trace_unuse_check_alias); 
 | 
            return -ERESTARTSYS; 
 | 
        } 
 | 
  
 | 
        afs_unuse_cell(cell->net, p, afs_cell_trace_unuse_check_alias); 
 | 
    } 
 | 
  
 | 
    mutex_unlock(&cell->net->proc_cells_lock); 
 | 
    _leave(" = 0"); 
 | 
    return 0; 
 | 
  
 | 
is_alias: 
 | 
    cell->alias_of = p; /* Transfer our ref */ 
 | 
    return 1; 
 | 
} 
 | 
  
 | 
/* 
 | 
 * Look up a VLDB record for a volume. 
 | 
 */ 
 | 
static char *afs_vl_get_cell_name(struct afs_cell *cell, struct key *key) 
 | 
{ 
 | 
    struct afs_vl_cursor vc; 
 | 
    char *cell_name = ERR_PTR(-EDESTADDRREQ); 
 | 
    bool skipped = false, not_skipped = false; 
 | 
    int ret; 
 | 
  
 | 
    if (!afs_begin_vlserver_operation(&vc, cell, key)) 
 | 
        return ERR_PTR(-ERESTARTSYS); 
 | 
  
 | 
    while (afs_select_vlserver(&vc)) { 
 | 
        if (!test_bit(AFS_VLSERVER_FL_IS_YFS, &vc.server->flags)) { 
 | 
            vc.ac.error = -EOPNOTSUPP; 
 | 
            skipped = true; 
 | 
            continue; 
 | 
        } 
 | 
        not_skipped = true; 
 | 
        cell_name = afs_yfsvl_get_cell_name(&vc); 
 | 
    } 
 | 
  
 | 
    ret = afs_end_vlserver_operation(&vc); 
 | 
    if (skipped && !not_skipped) 
 | 
        ret = -EOPNOTSUPP; 
 | 
    return ret < 0 ? ERR_PTR(ret) : cell_name; 
 | 
} 
 | 
  
 | 
static int yfs_check_canonical_cell_name(struct afs_cell *cell, struct key *key) 
 | 
{ 
 | 
    struct afs_cell *master; 
 | 
    char *cell_name; 
 | 
  
 | 
    cell_name = afs_vl_get_cell_name(cell, key); 
 | 
    if (IS_ERR(cell_name)) 
 | 
        return PTR_ERR(cell_name); 
 | 
  
 | 
    if (strcmp(cell_name, cell->name) == 0) { 
 | 
        kfree(cell_name); 
 | 
        return 0; 
 | 
    } 
 | 
  
 | 
    master = afs_lookup_cell(cell->net, cell_name, strlen(cell_name), 
 | 
                 NULL, false); 
 | 
    kfree(cell_name); 
 | 
    if (IS_ERR(master)) 
 | 
        return PTR_ERR(master); 
 | 
  
 | 
    cell->alias_of = master; /* Transfer our ref */ 
 | 
    return 1; 
 | 
} 
 | 
  
 | 
static int afs_do_cell_detect_alias(struct afs_cell *cell, struct key *key) 
 | 
{ 
 | 
    struct afs_volume *root_volume; 
 | 
    int ret; 
 | 
  
 | 
    _enter("%s", cell->name); 
 | 
  
 | 
    ret = yfs_check_canonical_cell_name(cell, key); 
 | 
    if (ret != -EOPNOTSUPP) 
 | 
        return ret; 
 | 
  
 | 
    /* Try and get the root.cell volume for comparison with other cells */ 
 | 
    root_volume = afs_sample_volume(cell, key, "root.cell", 9); 
 | 
    if (!IS_ERR(root_volume)) { 
 | 
        cell->root_volume = root_volume; 
 | 
        return afs_compare_cell_roots(cell); 
 | 
    } 
 | 
  
 | 
    if (PTR_ERR(root_volume) != -ENOMEDIUM) 
 | 
        return PTR_ERR(root_volume); 
 | 
  
 | 
    /* Okay, this cell doesn't have an root.cell volume.  We need to 
 | 
     * locate some other random volume and use that to check. 
 | 
     */ 
 | 
    return afs_query_for_alias(cell, key); 
 | 
} 
 | 
  
 | 
/* 
 | 
 * Check to see if a new cell is an alias of a cell we already have.  At this 
 | 
 * point we have the cell's volume server list. 
 | 
 * 
 | 
 * Returns 0 if we didn't detect an alias, 1 if we found an alias and an error 
 | 
 * if we had problems gathering the data required.  In the case the we did 
 | 
 * detect an alias, cell->alias_of is set to point to the assumed master. 
 | 
 */ 
 | 
int afs_cell_detect_alias(struct afs_cell *cell, struct key *key) 
 | 
{ 
 | 
    struct afs_net *net = cell->net; 
 | 
    int ret; 
 | 
  
 | 
    if (mutex_lock_interruptible(&net->cells_alias_lock) < 0) 
 | 
        return -ERESTARTSYS; 
 | 
  
 | 
    if (test_bit(AFS_CELL_FL_CHECK_ALIAS, &cell->flags)) { 
 | 
        ret = afs_do_cell_detect_alias(cell, key); 
 | 
        if (ret >= 0) 
 | 
            clear_bit_unlock(AFS_CELL_FL_CHECK_ALIAS, &cell->flags); 
 | 
    } else { 
 | 
        ret = cell->alias_of ? 1 : 0; 
 | 
    } 
 | 
  
 | 
    mutex_unlock(&net->cells_alias_lock); 
 | 
  
 | 
    if (ret == 1) 
 | 
        pr_notice("kAFS: Cell %s is an alias of %s\n", 
 | 
              cell->name, cell->alias_of->name); 
 | 
    return ret; 
 | 
} 
 |