def colocation_rm(argv): elementFound = False if len(argv) < 2: usage.constraint() sys.exit(1) (dom, constraintsElement) = getCurrentConstraints() resource1 = argv[0] resource2 = argv[1] for co_loc in constraintsElement.getElementsByTagName('rsc_colocation')[:]: if co_loc.getAttribute("rsc") == resource1 and co_loc.getAttribute( "with-rsc") == resource2: constraintsElement.removeChild(co_loc) elementFound = True if co_loc.getAttribute("rsc") == resource2 and co_loc.getAttribute( "with-rsc") == resource1: constraintsElement.removeChild(co_loc) elementFound = True if elementFound == True: utils.replace_cib_configuration(dom) else: print("No matching resources found in ordering list")
def resource_master_remove(argv): if len(argv) < 1: usage.resource() sys.exit(1) dom = utils.get_cib_dom() master_id = argv.pop(0) master_found = False # Check to see if there's a resource/group with the master_id if so, we remove the parent for rg in (dom.getElementsByTagName("primitive") + dom.getElementsByTagName("group")): if rg.getAttribute("id") == master_id and rg.parentNode.tagName == "master": master_id = rg.parentNode.getAttribute("id") resources_to_cleanup = [] for master in dom.getElementsByTagName("master"): if master.getAttribute("id") == master_id: childNodes = master.getElementsByTagName("primitive") for child in childNodes: resources_to_cleanup.append(child.getAttribute("id")) master_found = True break if not master_found: print "Error: Unable to find multi-state resource with id %s" % master_id sys.exit(1) master.parentNode.removeChild(master) utils.replace_cib_configuration(dom) if (not utils.usefile): for r in resources_to_cleanup: args = ["crm_resource","-C","-r",r] cmdoutput, retVal = utils.run(args)
def acl_permission(argv): if len(argv) < 1: usage.acl("permission") sys.exit(1) dom = utils.get_cib_dom() dom, acls = get_acls(dom) command = argv.pop(0) if command == "add": if len(argv) < 4: usage.acl("permission add") sys.exit(1) role_id = argv.pop(0) found = False for role in dom.getElementsByTagName("acl_role"): if role.getAttribute("id") == role_id: found = True break if found == False: acl_role(["create", role_id] + argv) return while len(argv) >= 3: kind = argv.pop(0) se = dom.createElement("acl_permission") se.setAttribute("id", utils.find_unique_id(dom, role_id + "-" + kind)) se.setAttribute("kind", kind) xp_id = argv.pop(0) if xp_id == "xpath": xpath_query = argv.pop(0) se.setAttribute("xpath",xpath_query) elif xp_id == "id": acl_ref = argv.pop(0) se.setAttribute("reference",acl_ref) else: usage.acl("permission add") role.appendChild(se) utils.replace_cib_configuration(dom) elif command == "delete": if len(argv) < 1: usage.acl("permission delete") sys.exit(1) perm_id = argv.pop(0) found = False for elem in dom.getElementsByTagName("acl_permission"): if elem.getAttribute("id") == perm_id: elem.parentNode.removeChild(elem) found = True if not found: utils.err("Unable to find permission with id: %s" % perm_id) utils.replace_cib_configuration(dom) else: usage.acl("permission") sys.exit(1)
def cluster_remote_node(argv): if len(argv) < 1: usage.cluster(["remote-node"]) sys.exit(1) command = argv.pop(0) if command == "add": if len(argv) < 2: usage.cluster(["remote-node"]) sys.exit(1) hostname = argv.pop(0) rsc = argv.pop(0) if not utils.is_resource(rsc): utils.err("unable to find resource '%s'", rsc) resource.resource_update(rsc, ["meta", "remote-node="+hostname] + argv) elif command in ["remove","delete"]: if len(argv) < 1: usage.cluster(["remote-node"]) sys.exit(1) hostname = argv.pop(0) dom = utils.get_cib_dom() nvpairs = dom.getElementsByTagName("nvpair") nvpairs_to_remove = [] for nvpair in nvpairs: if nvpair.getAttribute("name") == "remote-node" and nvpair.getAttribute("value") == hostname: for np in nvpair.parentNode.getElementsByTagName("nvpair"): if np.getAttribute("name").startswith("remote-"): nvpairs_to_remove.append(np) for nvpair in nvpairs_to_remove[:]: nvpair.parentNode.removeChild(nvpair) utils.replace_cib_configuration(dom) else: usage.cluster(["remote-node"]) sys.exit(1)
def order_set(argv): setoptions = [] for i in range(len(argv)): if argv[i] == "setoptions": setoptions = argv[i + 1:] argv[i:] = [] break argv.insert(0, "set") resource_sets = set_args_into_array(argv) if not check_empty_resource_sets(resource_sets): usage.constraint(["order set"]) sys.exit(1) cib, constraints = getCurrentConstraints(utils.get_cib_dom()) attributes = [] id_specified = False for opt in setoptions: if "=" not in opt: utils.err("missing value of '%s' option" % opt) name, value = opt.split("=", 1) if name == "id": id_valid, id_error = utils.validate_xml_id(value, 'constraint id') if not id_valid: utils.err(id_error) if utils.does_id_exist(cib, value): utils.err( "id '%s' is already in use, please specify another one" % value) id_specified = True attributes.append((name, value)) elif name == "kind": normalized_value = value.lower().capitalize() if normalized_value not in OPTIONS_KIND: utils.err("invalid kind value '%s', allowed values are: %s" % (value, ", ".join(OPTIONS_KIND))) attributes.append((name, normalized_value)) elif name == "symmetrical": if value.lower() not in OPTIONS_SYMMETRICAL: utils.err( "invalid symmetrical value '%s', allowed values are: %s" % (value, ", ".join(OPTIONS_SYMMETRICAL))) attributes.append((name, value.lower())) else: utils.err("invalid option '%s', allowed options are: %s" % (name, "kind, symmetrical, id")) if not id_specified: order_id = "pcs_rsc_order" for a in argv: if "=" not in a: order_id += "_" + a attributes.append(("id", utils.find_unique_id(cib, order_id))) rsc_order = cib.createElement("rsc_order") for name, value in attributes: rsc_order.setAttribute(name, value) set_add_resource_sets(rsc_order, resource_sets, cib) constraints.appendChild(rsc_order) utils.replace_cib_configuration(cib)
def cluster_remote_node(argv): if len(argv) < 1: usage.cluster(["remote-node"]) sys.exit(1) command = argv.pop(0) if command == "add": if len(argv) < 2: usage.cluster(["remote-node"]) sys.exit(1) hostname = argv.pop(0) rsc = argv.pop(0) if not utils.is_resource(rsc): utils.err("unable to find resource '%s'", rsc) resource.resource_update(rsc, ["meta", "remote-node="+hostname] + argv) elif command == "remove": if len(argv) < 1: usage.cluster(["remote-node"]) sys.exit(1) hostname = argv.pop(0) dom = utils.get_cib_dom() nvpairs = dom.getElementsByTagName("nvpair") nvpairs_to_remove = [] for nvpair in nvpairs: if nvpair.getAttribute("name") == "remote-node" and nvpair.getAttribute("value") == hostname: for np in nvpair.parentNode.getElementsByTagName("nvpair"): if np.getAttribute("name").startswith("remote-"): nvpairs_to_remove.append(np) for nvpair in nvpairs_to_remove[:]: nvpair.parentNode.removeChild(nvpair) utils.replace_cib_configuration(dom) else: usage.cluster(["remote-node"]) sys.exit(1)
def stonith_level_add(level, node, devices): dom = utils.get_cib_dom() if not "--force" in utils.pcs_options: for dev in devices.split(","): if not utils.is_stonith_resource(dev): utils.err("%s is not a stonith id (use --force to override)" % dev) if not utils.is_corosync_node(node): utils.err("%s is not currently a node (use --force to override)" % node) ft = dom.getElementsByTagName("fencing-topology") if len(ft) == 0: conf = dom.getElementsByTagName("configuration")[0] ft = dom.createElement("fencing-topology") conf.appendChild(ft) else: ft = ft[0] fls = ft.getElementsByTagName("fencing-level") for fl in fls: if fl.getAttribute("target") == node and fl.getAttribute("index") == level and fl.getAttribute("devices") == devices: utils.err("unable to add fencing level, fencing level for node: %s, at level: %s, with device: %s already exists" % (node,level,devices)) new_fl = dom.createElement("fencing-level") ft.appendChild(new_fl) new_fl.setAttribute("target", node) new_fl.setAttribute("index", level) new_fl.setAttribute("devices", devices) new_fl.setAttribute("id", utils.find_unique_id(dom, "fl-" + node +"-" + level)) utils.replace_cib_configuration(dom)
def stonith_level_add(level, node, devices): dom = utils.get_cib_dom() if not "--force" in utils.pcs_options: for dev in devices.split(","): if not utils.is_stonith_resource(dev): utils.err("%s is not a stonith id (use --force to override)" % dev) if not utils.is_pacemaker_node(node) and not utils.is_corosync_node(node): utils.err("%s is not currently a node (use --force to override)" % node) ft = dom.getElementsByTagName("fencing-topology") if len(ft) == 0: conf = dom.getElementsByTagName("configuration")[0] ft = dom.createElement("fencing-topology") conf.appendChild(ft) else: ft = ft[0] fls = ft.getElementsByTagName("fencing-level") for fl in fls: if fl.getAttribute("target") == node and fl.getAttribute("index") == level and fl.getAttribute("devices") == devices: utils.err("unable to add fencing level, fencing level for node: %s, at level: %s, with device: %s already exists" % (node,level,devices)) new_fl = dom.createElement("fencing-level") ft.appendChild(new_fl) new_fl.setAttribute("target", node) new_fl.setAttribute("index", level) new_fl.setAttribute("devices", devices) new_fl.setAttribute("id", utils.find_unique_id(dom, "fl-" + node +"-" + level)) utils.replace_cib_configuration(dom)
def location_rule(argv): if len(argv) < 3: usage.constraint("location rule") sys.exit(1) res_name = argv.pop(0) resource_valid, resource_error = utils.validate_constraint_resource( utils.get_cib_dom(), res_name ) if not resource_valid: utils.err(resource_error) argv.pop(0) cib = utils.get_cib_dom() constraints = cib.getElementsByTagName("constraints")[0] lc = cib.createElement("rsc_location") constraints.appendChild(lc) lc_id = utils.find_unique_id(cib, "location-" + res_name) lc.setAttribute("id", lc_id) lc.setAttribute("rsc", res_name) rule_utils.dom_rule_add(lc, argv) utils.replace_cib_configuration(cib)
def constraint_resource_update(old_id, passed_dom=None): dom = utils.get_cib_dom() if passed_dom is None else passed_dom resources = dom.getElementsByTagName("primitive") found_resource = None for res in resources: if res.getAttribute("id") == old_id: found_resource = res break new_id = None if found_resource: if found_resource.parentNode.tagName == "master" or found_resource.parentNode.tagName == "clone": new_id = found_resource.parentNode.getAttribute("id") if new_id: constraints = dom.getElementsByTagName("rsc_location") constraints += dom.getElementsByTagName("rsc_order") constraints += dom.getElementsByTagName("rsc_colocation") attrs_to_update=["rsc","first","then", "with-rsc"] for constraint in constraints: for attr in attrs_to_update: if constraint.getAttribute(attr) == old_id: constraint.setAttribute(attr, new_id) if passed_dom is None: utils.replace_cib_configuration(dom) if passed_dom: return dom
def constraint_rule(argv): if len(argv) < 2: usage.constraint("rule") sys.exit(1) found = False command = argv.pop(0) constraint_id = None rule_id = None if command == "add": constraint_id = argv.pop(0) cib = utils.get_cib_dom() constraint = utils.dom_get_element_with_id( cib.getElementsByTagName("constraints")[0], "rsc_location", constraint_id ) if not constraint: utils.err("Unable to find constraint: " + constraint_id) options, rule_argv = rule_utils.parse_argv(argv) rule_utils.dom_rule_add(constraint, options, rule_argv) location_rule_check_duplicates(cib, constraint) utils.replace_cib_configuration(cib) elif command in ["remove","delete"]: cib = utils.get_cib_etree() temp_id = argv.pop(0) constraints = cib.find('.//constraints') loc_cons = cib.findall(str('.//rsc_location')) rules = cib.findall(str('.//rule')) for loc_con in loc_cons: for rule in loc_con: if rule.get("id") == temp_id: if len(loc_con) > 1: print("Removing Rule: {0}".format(rule.get("id"))) loc_con.remove(rule) found = True break else: print( "Removing Constraint: {0}".format(loc_con.get("id")) ) constraints.remove(loc_con) found = True break if found == True: break if found: utils.replace_cib_configuration(cib) else: utils.err("unable to find rule with id: %s" % temp_id) else: usage.constraint("rule") sys.exit(1)
def colocation_set(argv): setoptions = [] for i in range(len(argv)): if argv[i] == "setoptions": setoptions = argv[i+1:] argv[i:] = [] break current_set = set_args_into_array(argv) colocation_id = "pcs_rsc_colocation" for a in argv: if a.find('=') == -1: colocation_id = colocation_id + "_" + a cib = utils.get_cib_etree() constraints = cib.find(".//constraints") if constraints == None: constraints = ET.SubElement(cib, "constraints") rsc_colocation = ET.SubElement(constraints,"rsc_colocation") rsc_colocation.set("id", utils.find_unique_id(cib,colocation_id)) rsc_colocation.set("score","INFINITY") for opt in setoptions: if opt.find("=") != -1: name,value = opt.split("=") rsc_colocation.set(name,value) set_add_resource_sets(rsc_colocation, current_set, cib) utils.replace_cib_configuration(cib)
def order_rm(argv): if len(argv) == 0: usage.constraint() sys.exit(1) elementFound = False (dom,constraintsElement) = getCurrentConstraints() for resource in argv: for ord_loc in constraintsElement.getElementsByTagName('rsc_order')[:]: if ord_loc.getAttribute("first") == resource or ord_loc.getAttribute("then") == resource: constraintsElement.removeChild(ord_loc) elementFound = True resource_refs_to_remove = [] for ord_set in constraintsElement.getElementsByTagName('resource_ref'): if ord_set.getAttribute("id") == resource: resource_refs_to_remove.append(ord_set) elementFound = True for res_ref in resource_refs_to_remove: res_set = res_ref.parentNode res_order = res_set.parentNode res_ref.parentNode.removeChild(res_ref) if len(res_set.getElementsByTagName('resource_ref')) <= 0: res_set.parentNode.removeChild(res_set) if len(res_order.getElementsByTagName('resource_set')) <= 0: res_order.parentNode.removeChild(res_order) if elementFound == True: utils.replace_cib_configuration(dom) else: utils.err("No matching resources found in ordering list")
def remove_constraints_containing(resource_id,output=False,constraints_element = None, passed_dom=None): constraints,set_constraints = find_constraints_containing(resource_id, passed_dom) for c in constraints: if output == True: print "Removing Constraint - " + c if constraints_element != None: constraint_rm([c], True, constraints_element, passed_dom=passed_dom) else: constraint_rm([c], passed_dom=passed_dom) if len(set_constraints) != 0: (dom, constraintsElement) = getCurrentConstraints(passed_dom) for c in constraintsElement.getElementsByTagName("resource_ref")[:]: # If resource id is in a set, remove it from the set, if the set # is empty, then we remove the set, if the parent of the set # is empty then we remove it if c.getAttribute("id") == resource_id: pn = c.parentNode pn.removeChild(c) if output == True: print "Removing %s from set %s" % (resource_id,pn.getAttribute("id")) if pn.getElementsByTagName("resource_ref").length == 0: print "Removing set %s" % pn.getAttribute("id") pn2 = pn.parentNode pn2.removeChild(pn) if pn2.getElementsByTagName("resource_set").length == 0: pn2.parentNode.removeChild(pn2) print "Removing constraint %s" % pn2.getAttribute("id") if passed_dom: return dom utils.replace_cib_configuration(dom)
def constraint_rule(argv): if len(argv) < 2: usage.constraint("rule") sys.exit(1) found = False command = argv.pop(0) constraint_id = None rule_id = None cib = utils.get_cib_etree() if command == "add": constraint_id = argv.pop(0) constraint = None for a in cib.findall(".//configuration//*"): if a.get("id") == constraint_id and a.tag == "rsc_location": found = True constraint = a if not found: utils.err("Unable to find constraint: " + constraint_id) utils.rule_add(constraint, argv) utils.replace_cib_configuration(cib) elif command in ["remove","delete"]: temp_id = argv.pop(0) constraints = cib.find('.//constraints') loc_cons = cib.findall('.//rsc_location') rules = cib.findall('.//rule') for loc_con in loc_cons: for rule in loc_con: if rule.get("id") == temp_id: if len(loc_con) > 1: print "Removing Rule:",rule.get("id") loc_con.remove(rule) found = True break else: print "Removing Constraint:",loc_con.get("id") constraints.remove(loc_con) found = True break if found == True: break if found: utils.replace_cib_configuration(cib) else: utils.err("unable to find rule with id: %s" % temp_id) else: usage.constraint("rule") sys.exit(1)
def stonith_cmd(argv): if len(argv) == 0: argv = ["show"] sub_cmd = argv.pop(0) if (sub_cmd == "help"): usage.stonith(argv) elif (sub_cmd == "list"): stonith_list_available(argv) elif (sub_cmd == "describe"): if len(argv) == 1: stonith_list_options(argv[0]) else: usage.stonith() sys.exit(1) elif (sub_cmd == "create"): if len(argv) < 2: usage.stonith() sys.exit(1) stn_id = argv.pop(0) stn_type = "stonith:"+argv.pop(0) st_values, op_values, meta_values = resource.parse_resource_options( argv, with_clone=False ) resource.resource_create(stn_id, stn_type, st_values, op_values, meta_values) elif (sub_cmd == "update"): stn_id = argv.pop(0) resource.resource_update(stn_id,argv) elif (sub_cmd == "delete"): if len(argv) == 1: stn_id = argv.pop(0) utils.replace_cib_configuration( stonith_level_rm_device(utils.get_cib_dom(), stn_id) ) resource.resource_remove(stn_id) else: usage.stonith(["delete"]) sys.exit(1) elif (sub_cmd == "show"): resource.resource_show(argv, True) stonith_level([]) elif (sub_cmd == "level"): stonith_level(argv) elif (sub_cmd == "fence"): stonith_fence(argv) elif (sub_cmd == "cleanup"): if len(argv) == 0: resource.resource_cleanup_all() else: res_id = argv.pop(0) resource.resource_cleanup(res_id) elif (sub_cmd == "confirm"): stonith_confirm(argv) else: usage.stonith() sys.exit(1)
def constraint_rm(argv, returnStatus=False, constraintsElement=None, passed_dom=None): if len(argv) < 1: usage.constraint() sys.exit(1) bad_constraint = False if len(argv) != 1: for arg in argv: if not constraint_rm([arg], True, passed_dom=passed_dom): bad_constraint = True if bad_constraint: sys.exit(1) return else: c_id = argv.pop(0) elementFound = False if not constraintsElement: (dom, constraintsElement) = getCurrentConstraints(passed_dom) use_cibadmin = True else: use_cibadmin = False for co in constraintsElement.childNodes[:]: if co.nodeType != xml.dom.Node.ELEMENT_NODE: continue if co.getAttribute("id") == c_id: constraintsElement.removeChild(co) elementFound = True if not elementFound: for rule in constraintsElement.getElementsByTagName("rule")[:]: if rule.getAttribute("id") == c_id: elementFound = True parent = rule.parentNode parent.removeChild(rule) if len(parent.getElementsByTagName("rule")) == 0: parent.parentNode.removeChild(parent) if elementFound == True: if passed_dom: return dom if use_cibadmin: utils.replace_cib_configuration(dom) if returnStatus: return True else: print("Error: Unable to find constraint - '%s'" % c_id, file=sys.stderr) if returnStatus: return False sys.exit(1)
def location_add(argv, rm=False): if len(argv) != 4 and (rm == False or len(argv) < 1): usage.constraint() sys.exit(1) constraint_id = argv.pop(0) # If we're removing, we only care about the id if (rm == True): resource_name = "" node = "" score = "" else: id_valid, id_error = utils.validate_xml_id(constraint_id, 'constraint id') if not id_valid: utils.err(id_error) resource_name = argv.pop(0) node = argv.pop(0) score = argv.pop(0) resource_valid, resource_error = utils.validate_constraint_resource( utils.get_cib_dom(), resource_name) if not resource_valid: utils.err(resource_error) if not utils.is_score(score): utils.err( "invalid score '%s', use integer or INFINITY or -INFINITY" % score) # Verify current constraint doesn't already exist # If it does we replace it with the new constraint (dom, constraintsElement) = getCurrentConstraints() elementsToRemove = [] # If the id matches, or the rsc & node match, then we replace/remove for rsc_loc in constraintsElement.getElementsByTagName('rsc_location'): if (constraint_id == rsc_loc.getAttribute("id")) or \ (rsc_loc.getAttribute("rsc") == resource_name and \ rsc_loc.getAttribute("node") == node and not rm): elementsToRemove.append(rsc_loc) for etr in elementsToRemove: constraintsElement.removeChild(etr) if (rm == True and len(elementsToRemove) == 0): utils.err("resource location id: " + constraint_id + " not found.") if (not rm): element = dom.createElement("rsc_location") element.setAttribute("id", constraint_id) element.setAttribute("rsc", resource_name) element.setAttribute("node", node) element.setAttribute("score", score) constraintsElement.appendChild(element) utils.replace_cib_configuration(dom)
def stonith_level_rm(level, node, devices): dom = utils.get_cib_dom() if devices != "": node_devices_combo = node + "," + devices else: node_devices_combo = node ft = dom.getElementsByTagName("fencing-topology") if len(ft) == 0: utils.err( "unable to remove fencing level, fencing level for node: %s, at level: %s, with device: %s doesn't exist" % (node, level, devices)) else: ft = ft[0] fls = ft.getElementsByTagName("fencing-level") fls_to_remove = [] if node != "": if devices != "": found = False for fl in fls: if fl.getAttribute("target") == node and fl.getAttribute( "index") == level and fl.getAttribute( "devices") == devices: found = True break if fl.getAttribute("index") == level and fl.getAttribute( "devices") == node_devices_combo: found = True break if found == False: utils.err( "unable to remove fencing level, fencing level for node: %s, at level: %s, with device: %s doesn't exist" % (node, level, devices)) fl.parentNode.removeChild(fl) else: for fl in fls: if fl.getAttribute("index") == level and ( fl.getAttribute("target") == node or fl.getAttribute("devices") == node): fl.parentNode.removeChild(fl) else: for fl in fls: if fl.getAttribute("index") == level: parent = fl.parentNode parent.removeChild(fl) if len(parent.getElementsByTagName("fencing-level")) == 0: parent.parentNode.removeChild(parent) break utils.replace_cib_configuration(dom)
def order_set(argv): current_set = set_args_into_array(argv) cib = utils.get_cib_etree() constraints = cib.find(".//constraints") if constraints == None: constraints = ET.SubElement(cib, "constraints") rsc_order = ET.SubElement(constraints,"rsc_order") rsc_order.set("id", utils.find_unique_id(cib,"pcs_rsc_order")) set_add_resource_sets(rsc_order, current_set, cib) utils.replace_cib_configuration(cib)
def set_node_utilization(node, argv): cib = utils.get_cib_dom() node_el = utils.dom_get_node(cib, node) if node_el is None: utils.err("Unable to find a node: {0}".format(node)) utils.dom_update_utilization( node_el, utils.convert_args_to_tuples(argv), "nodes-" ) utils.replace_cib_configuration(cib)
def order_set(argv): current_set = set_args_into_array(argv) cib = utils.get_cib_etree() constraints = cib.find(".//constraints") if constraints == None: constraints = ET.SubElement(cib, "constraints") rsc_order = ET.SubElement(constraints, "rsc_order") rsc_order.set("id", utils.find_unique_id(cib, "pcs_rsc_order")) set_add_resource_sets(rsc_order, current_set, cib) utils.replace_cib_configuration(cib)
def stonith_cmd(argv): if len(argv) == 0: argv = ["show"] sub_cmd = argv.pop(0) if (sub_cmd == "help"): usage.stonith(argv) elif (sub_cmd == "list"): stonith_list_available(argv) elif (sub_cmd == "describe"): if len(argv) == 1: stonith_list_options(argv[0]) else: usage.stonith() sys.exit(1) elif (sub_cmd == "create"): if len(argv) < 2: usage.stonith() sys.exit(1) stn_id = argv.pop(0) stn_type = "stonith:" + argv.pop(0) st_values, op_values, meta_values = resource.parse_resource_options( argv, with_clone=False) resource.resource_create(stn_id, stn_type, st_values, op_values, meta_values) elif (sub_cmd == "update"): stn_id = argv.pop(0) resource.resource_update(stn_id, argv) elif (sub_cmd == "delete"): if len(argv) == 1: stn_id = argv.pop(0) utils.replace_cib_configuration( stonith_level_rm_device(utils.get_cib_dom(), stn_id)) resource.resource_remove(stn_id) else: usage.stonith(["delete"]) sys.exit(1) elif (sub_cmd == "show"): resource.resource_show(argv, True) stonith_level([]) elif (sub_cmd == "level"): stonith_level(argv) elif (sub_cmd == "fence"): stonith_fence(argv) elif (sub_cmd == "cleanup"): if len(argv) == 0: resource.resource_cleanup_all() else: res_id = argv.pop(0) resource.resource_cleanup(res_id) elif (sub_cmd == "confirm"): stonith_confirm(argv) else: usage.stonith() sys.exit(1)
def acl_target(argv, group=False): if len(argv) < 2: if group: usage.acl("group") sys.exit(1) else: usage.acl("target") sys.exit(1) dom = utils.get_cib_dom() dom, acls = get_acls(dom) command = argv.pop(0) tug_id = argv.pop(0) if command == "create": if utils.does_id_exist(dom, tug_id): utils.err(tug_id + " already exists in cib") if group: element = dom.createElement("acl_group") else: element = dom.createElement("acl_target") element.setAttribute("id", tug_id) acls.appendChild(element) for role in argv: r = dom.createElement("role") r.setAttribute("id", role) element.appendChild(r) utils.replace_cib_configuration(dom) elif command == "delete": found = False if group: elist = dom.getElementsByTagName("acl_group") else: elist = dom.getElementsByTagName("acl_target") for elem in elist: if elem.getAttribute("id") == tug_id: found = True elem.parentNode.removeChild(elem) break if not found: if group: utils.err("unable to find acl group: %s" % tug_id) else: utils.err("unable to find acl target/user: %s" % tug_id) utils.replace_cib_configuration(dom) else: if group: usage.acl("group") else: usage.acl("target") sys.exit(1)
def acl_target(argv,group=False): if len(argv) < 2: if group: usage.acl("group") sys.exit(1) else: usage.acl("target") sys.exit(1) dom = utils.get_cib_dom() dom, acls = get_acls(dom) command = argv.pop(0) tug_id = argv.pop(0) if command == "create": if utils.does_id_exist(dom,tug_id): utils.err(tug_id + " already exists in cib") if group: element = dom.createElement("acl_group") else: element = dom.createElement("acl_target") element.setAttribute("id", tug_id) acls.appendChild(element) for role in argv: r = dom.createElement("role") r.setAttribute("id", role) element.appendChild(r) utils.replace_cib_configuration(dom) elif command == "delete": found = False if group: elist = dom.getElementsByTagName("acl_group") else: elist = dom.getElementsByTagName("acl_target") for elem in elist: if elem.getAttribute("id") == tug_id: found = True elem.parentNode.removeChild(elem) break if not found: if group: utils.err("unable to find acl group: %s" % tug_id) else: utils.err("unable to find acl target/user: %s" % tug_id) utils.replace_cib_configuration(dom) else: if group: usage.acl("group") else: usage.acl("target") sys.exit(1)
def constraint_rule(argv): if len(argv) < 2: usage.constraint("rule") sys.exit(1) found = False command = argv.pop(0) constraint_id = None rule_id = None if command == "add": constraint_id = argv.pop(0) cib = utils.get_cib_dom() constraint = utils.dom_get_element_with_id( cib.getElementsByTagName("constraints")[0], "rsc_location", constraint_id) if not constraint: utils.err("Unable to find constraint: " + constraint_id) options, rule_argv = rule_utils.parse_argv(argv) rule_utils.dom_rule_add(constraint, options, rule_argv) location_rule_check_duplicates(cib, constraint) utils.replace_cib_configuration(cib) elif command in ["remove", "delete"]: cib = utils.get_cib_etree() temp_id = argv.pop(0) constraints = cib.find('.//constraints') loc_cons = cib.findall('.//rsc_location') rules = cib.findall('.//rule') for loc_con in loc_cons: for rule in loc_con: if rule.get("id") == temp_id: if len(loc_con) > 1: print("Removing Rule:", rule.get("id")) loc_con.remove(rule) found = True break else: print("Removing Constraint:", loc_con.get("id")) constraints.remove(loc_con) found = True break if found == True: break if found: utils.replace_cib_configuration(cib) else: utils.err("unable to find rule with id: %s" % temp_id) else: usage.constraint("rule") sys.exit(1)
def cluster_cib_rollback(argv): if len(argv) != 1: usage.cluster(["cib-rollback"]) sys.exit(1) cib_path = os.path.join(settings.cib_dir, argv[0]) try: snapshot_dom = parse(cib_path) except Exception as e: utils.err("unable to read CIB from '%s': %s" % (cib_path, e)) utils.replace_cib_configuration(snapshot_dom)
def config_checkpoint_restore(argv): if len(argv) != 1: usage.config(["checkpoint", "restore"]) sys.exit(1) cib_path = os.path.join(settings.cib_dir, "cib-%s.raw" % argv[0]) try: snapshot_dom = parse(cib_path) except Exception as e: utils.err("unable to read the checkpoint: %s" % e) utils.replace_cib_configuration(snapshot_dom)
def location_add(argv,rm=False): if len(argv) != 4 and (rm == False or len(argv) < 1): usage.constraint() sys.exit(1) constraint_id = argv.pop(0) # If we're removing, we only care about the id if (rm == True): resource_name = "" node = "" score = "" else: id_valid, id_error = utils.validate_xml_id(constraint_id, 'constraint id') if not id_valid: utils.err(id_error) resource_name = argv.pop(0) node = argv.pop(0) score = argv.pop(0) resource_valid, resource_error = utils.validate_constraint_resource( utils.get_cib_dom(), resource_name ) if not resource_valid: utils.err(resource_error) if not utils.is_score(score): utils.err("invalid score '%s', use integer or INFINITY or -INFINITY" % score) # Verify current constraint doesn't already exist # If it does we replace it with the new constraint (dom,constraintsElement) = getCurrentConstraints() elementsToRemove = [] # If the id matches, or the rsc & node match, then we replace/remove for rsc_loc in constraintsElement.getElementsByTagName('rsc_location'): if (constraint_id == rsc_loc.getAttribute("id")) or \ (rsc_loc.getAttribute("rsc") == resource_name and \ rsc_loc.getAttribute("node") == node and not rm): elementsToRemove.append(rsc_loc) for etr in elementsToRemove: constraintsElement.removeChild(etr) if (rm == True and len(elementsToRemove) == 0): utils.err("resource location id: " + constraint_id + " not found.") if (not rm): element = dom.createElement("rsc_location") element.setAttribute("id",constraint_id) element.setAttribute("rsc",resource_name) element.setAttribute("node",node) element.setAttribute("score",score) constraintsElement.appendChild(element) utils.replace_cib_configuration(dom)
def constraint_rm(argv,returnStatus=False, constraintsElement=None, passed_dom=None): if len(argv) < 1: usage.constraint() sys.exit(1) bad_constraint = False if len(argv) != 1: for arg in argv: if not constraint_rm([arg],True, passed_dom=passed_dom): bad_constraint = True if bad_constraint: sys.exit(1) return else: c_id = argv.pop(0) elementFound = False if not constraintsElement: (dom, constraintsElement) = getCurrentConstraints(passed_dom) use_cibadmin = True else: use_cibadmin = False for co in constraintsElement.childNodes[:]: if co.nodeType != xml.dom.Node.ELEMENT_NODE: continue if co.getAttribute("id") == c_id: constraintsElement.removeChild(co) elementFound = True if not elementFound: for rule in constraintsElement.getElementsByTagName("rule")[:]: if rule.getAttribute("id") == c_id: elementFound = True parent = rule.parentNode parent.removeChild(rule) if len(parent.getElementsByTagName("rule")) == 0: parent.parentNode.removeChild(parent) if elementFound == True: if passed_dom: return dom if use_cibadmin: utils.replace_cib_configuration(dom) if returnStatus: return True else: print >> sys.stderr, "Error: Unable to find constraint - '%s'" % c_id if returnStatus: return False sys.exit(1)
def unset_property(argv): if len(argv) < 1: usage.property() sys.exit(1) if "--node" in utils.pcs_options: for arg in argv: utils.set_node_attribute(arg, "",utils.pcs_options["--node"]) else: cib_dom = utils.get_cib_dom() for arg in argv: utils.set_cib_property(arg, "", cib_dom) utils.replace_cib_configuration(cib_dom)
def location_rule(argv): if len(argv) < 3: usage.constraint(["location", "rule"]) sys.exit(1) res_name = argv.pop(0) resource_valid, resource_error, correct_id \ = utils.validate_constraint_resource(utils.get_cib_dom(), res_name) if "--autocorrect" in utils.pcs_options and correct_id: res_name = correct_id elif not resource_valid: utils.err(resource_error) argv.pop(0) # pop "rule" options, rule_argv = rule_utils.parse_argv(argv, { "constraint-id": None, "resource-discovery": None, }) # If resource-discovery is specified, we use it with the rsc_location # element not the rule if "resource-discovery" in options and options["resource-discovery"]: utils.checkAndUpgradeCIB(2, 2, 0) cib, constraints = getCurrentConstraints(utils.get_cib_dom()) lc = cib.createElement("rsc_location") lc.setAttribute("resource-discovery", options.pop("resource-discovery")) else: cib, constraints = getCurrentConstraints(utils.get_cib_dom()) lc = cib.createElement("rsc_location") constraints.appendChild(lc) if options.get("constraint-id"): id_valid, id_error = utils.validate_xml_id(options["constraint-id"], 'constraint id') if not id_valid: utils.err(id_error) if utils.does_id_exist(cib, options["constraint-id"]): utils.err("id '%s' is already in use, please specify another one" % options["constraint-id"]) lc.setAttribute("id", options["constraint-id"]) del options["constraint-id"] else: lc.setAttribute("id", utils.find_unique_id(cib, "location-" + res_name)) lc.setAttribute("rsc", res_name) rule_utils.dom_rule_add(lc, options, rule_argv) location_rule_check_duplicates(constraints, lc) utils.replace_cib_configuration(cib)
def order_set(argv): current_set = set_args_into_array(argv) order_id = "pcs_rsc_order" for a in argv: if a.find('=') == -1: order_id = order_id + "_" + a cib, constraints = getCurrentConstraints(utils.get_cib_dom()) rsc_order = cib.createElement("rsc_order") constraints.appendChild(rsc_order) rsc_order.setAttribute("id", utils.find_unique_id(cib, order_id)) set_add_resource_sets(rsc_order, current_set, cib) utils.replace_cib_configuration(cib)
def acl_permission(argv): if len(argv) < 1: usage.acl(["permission"]) sys.exit(1) dom = utils.get_cib_dom() dom, acls = get_acls(dom) command = argv.pop(0) if command == "add": if len(argv) < 4: usage.acl(["permission add"]) sys.exit(1) role_id = argv.pop(0) found = False for role in dom.getElementsByTagName("acl_role"): if role.getAttribute("id") == role_id: found = True break if found == False: acl_role(["create", role_id] + argv) return if not argv: usage.acl(["permission add"]) sys.exit(1) if not add_permissions_to_role(role, argv): usage.acl(["permission add"]) sys.exit(1) utils.replace_cib_configuration(dom) elif command == "delete": if len(argv) < 1: usage.acl(["permission delete"]) sys.exit(1) perm_id = argv.pop(0) found = False for elem in dom.getElementsByTagName("acl_permission"): if elem.getAttribute("id") == perm_id: elem.parentNode.removeChild(elem) found = True if not found: utils.err("Unable to find permission with id: %s" % perm_id) utils.replace_cib_configuration(dom) else: usage.acl(["permission"]) sys.exit(1)
def location_rule(argv): if len(argv) < 3: usage.constraint(["location", "rule"]) sys.exit(1) res_name = argv.pop(0) resource_valid, resource_error, correct_id \ = utils.validate_constraint_resource(utils.get_cib_dom(), res_name) if "--autocorrect" in utils.pcs_options and correct_id: res_name = correct_id elif not resource_valid: utils.err(resource_error) argv.pop(0) # pop "rule" options, rule_argv = rule_utils.parse_argv(argv, {"constraint-id": None, "resource-discovery": None,}) # If resource-discovery is specified, we use it with the rsc_location # element not the rule if "resource-discovery" in options and options["resource-discovery"]: utils.checkAndUpgradeCIB(2,2,0) cib, constraints = getCurrentConstraints(utils.get_cib_dom()) lc = cib.createElement("rsc_location") lc.setAttribute("resource-discovery", options.pop("resource-discovery")) else: cib, constraints = getCurrentConstraints(utils.get_cib_dom()) lc = cib.createElement("rsc_location") constraints.appendChild(lc) if options.get("constraint-id"): id_valid, id_error = utils.validate_xml_id( options["constraint-id"], 'constraint id' ) if not id_valid: utils.err(id_error) if utils.does_id_exist(cib, options["constraint-id"]): utils.err( "id '%s' is already in use, please specify another one" % options["constraint-id"] ) lc.setAttribute("id", options["constraint-id"]) del options["constraint-id"] else: lc.setAttribute("id", utils.find_unique_id(cib, "location-" + res_name)) lc.setAttribute("rsc", res_name) rule_utils.dom_rule_add(lc, options, rule_argv) location_rule_check_duplicates(constraints, lc) utils.replace_cib_configuration(cib)
def set_property(argv): prop_def_dict = utils.get_cluster_properties_definition() nodes_attr = "--node" in utils.pcs_options failed = False forced = "--force" in utils.pcs_options properties = {} for arg in argv: args = arg.split('=') if len(args) != 2: utils.err("invalid property format: '{0}'".format(arg), False) failed = True elif not args[0]: utils.err("empty property name: '{0}'".format(arg), False) failed = True elif nodes_attr or forced or args[1].strip() == "": properties[args[0]] = args[1] else: try: if utils.is_valid_cluster_property( prop_def_dict, args[0], args[1] ): properties[args[0]] = args[1] else: utils.err( "invalid value of property: '{0}', (use --force to " "override)".format(arg), False ) failed = True except utils.UnknownPropertyException: utils.err( "unknown cluster property: '{0}', (use --force to " "override)".format(args[0]), False ) failed = True if failed: sys.exit(1) if nodes_attr: for prop, value in properties.items(): utils.set_node_attribute(prop, value, utils.pcs_options["--node"]) else: cib_dom = utils.get_cib_dom() for prop, value in properties.items(): utils.set_cib_property(prop, value, cib_dom) utils.replace_cib_configuration(cib_dom)
def stonith_level_rm(level, node, devices): dom = utils.get_cib_dom() if devices != "": node_devices_combo = node + "," + devices else: node_devices_combo = node ft = dom.getElementsByTagName("fencing-topology") if len(ft) == 0: utils.err("unable to remove fencing level, fencing level for node: %s, at level: %s, with device: %s doesn't exist" % (node,level,devices)) else: ft = ft[0] fls = ft.getElementsByTagName("fencing-level") fls_to_remove = [] if node != "": if devices != "": found = False for fl in fls: if fl.getAttribute("target") == node and fl.getAttribute("index") == level and fl.getAttribute("devices") == devices: found = True break if fl.getAttribute("index") == level and fl.getAttribute("devices") == node_devices_combo: found = True break if found == False: utils.err("unable to remove fencing level, fencing level for node: %s, at level: %s, with device: %s doesn't exist" % (node,level,devices)) fl.parentNode.removeChild(fl) else: for fl in fls: if fl.getAttribute("index") == level and (fl.getAttribute("target") == node or fl.getAttribute("devices") == node): fl.parentNode.removeChild(fl) else: for fl in fls: if fl.getAttribute("index") == level: parent = fl.parentNode parent.removeChild(fl) if len(parent.getElementsByTagName("fencing-level")) == 0: parent.parentNode.removeChild(parent) break utils.replace_cib_configuration(dom)
def resource_operation_add(res_id, argv): if len(argv) < 1: usage.resource() sys.exit(1) op_name = argv.pop(0) dom = utils.get_cib_dom() resource_found = False for resource in dom.getElementsByTagName("primitive"): if resource.getAttribute("id") == res_id: resource_found = True break if not resource_found: print "Unable to find resource: %s" % res_id sys.exit(1) op_properties = convert_args_to_tuples(argv) op_properties.sort(key=lambda a:a[0]) op_properties.insert(0,('name', op_name)) found_match = False op = dom.createElement("op") op_id = res_id + "-" for prop in op_properties: op.setAttribute(prop[0], prop[1]) op_id += prop[0] + "-" + prop[1] + "-" op_id = op_id[:-1] op_id = utils.find_unique_id(dom, op_id) op.setAttribute("id", op_id) operations = resource.getElementsByTagName("operations") if len(operations) == 0: operations = dom.createElement("operations") resource.appendChild(operations) else: operations = operations[0] if utils.operation_exists(operations,op): print "Error: identical operation already exists for %s" % res_id sys.exit(1) operations.appendChild(op) utils.replace_cib_configuration(dom)
def location_rule(argv): if len(argv) < 3: usage.constraint("location rule") sys.exit(1) res_name = argv.pop(0) argv.pop(0) cib = utils.get_cib_etree() constraints = cib.find(".//constraints") lc = ET.SubElement(constraints,"rsc_location") lc_id = utils.find_unique_id(cib, "location-" + res_name) lc.set("id", lc_id) lc.set("rsc", res_name) utils.rule_add(lc, argv) utils.replace_cib_configuration(cib)
def location_rule(argv): if len(argv) < 3: usage.constraint("location rule") sys.exit(1) res_name = argv.pop(0) argv.pop(0) cib = utils.get_cib_etree() constraints = cib.find(".//constraints") lc = ET.SubElement(constraints, "rsc_location") lc_id = utils.find_unique_id(cib, "location-" + res_name) lc.set("id", lc_id) lc.set("rsc", res_name) utils.rule_add(lc, argv) utils.replace_cib_configuration(cib)
def location_rule(argv): if len(argv) < 3: usage.constraint("location rule") sys.exit(1) res_name = argv.pop(0) (dom, constraintsElement) = getCurrentConstraints() lc = dom.createElement("rsc_location") lc_id = utils.find_unique_id(dom, "location-" + res_name) lc.setAttribute("id", lc_id) lc.setAttribute("rsc", res_name) rule = utils.getRule(dom, lc, argv) lc.appendChild(rule) constraintsElement.appendChild(lc) utils.replace_cib_configuration(dom)
def resource_operation_remove(res_id, argv): if len(argv) < 1: usage.resource() sys.exit(1) original_argv = " ".join(argv) op_name = argv.pop(0) dom = utils.get_cib_dom() resource_found = False for resource in dom.getElementsByTagName("primitive"): if resource.getAttribute("id") == res_id: resource_found = True break if not resource_found: print "Unable to find resource: %s" % res_id sys.exit(1) op_properties = convert_args_to_tuples(argv) op_properties.append(('name', op_name)) found_match = False for op in resource.getElementsByTagName("op"): temp_properties = [] for attrName in op.attributes.keys(): if attrName == "id": continue temp_properties.append((attrName,op.attributes.get(attrName).nodeValue)) if len(set(op_properties) ^ set(temp_properties)) == 0: found_match = True parent = op.parentNode parent.removeChild(op) if len(parent.getElementsByTagName("op")) == 0: parent.parentNode.removeChild(parent) break if not found_match: print "Unable to find operation matching: %s" % original_argv sys.exit(1) utils.replace_cib_configuration(dom)
def colocation_set(argv): setoptions = [] for i in range(len(argv)): if argv[i] == "setoptions": setoptions = argv[i+1:] argv[i:] = [] break current_set = set_args_into_array(argv) colocation_id = "pcs_rsc_colocation" for a in argv: if a.find('=') == -1: colocation_id = colocation_id + "_" + a cib = utils.get_cib_etree() constraints = cib.find(".//constraints") if constraints == None: constraints = ET.SubElement(cib, "constraints") rsc_colocation = ET.SubElement(constraints,"rsc_colocation") rsc_colocation.set("id", utils.find_unique_id(cib,colocation_id)) rsc_colocation.set("score","INFINITY") score_options = ("score", "score-attribute", "score-attribute-mangle") score_specified = False for opt in setoptions: if opt.find("=") != -1: name,value = opt.split("=") if name not in score_options: utils.err( "invalid option '%s', allowed options are: %s" % (name, ", ".join(score_options)) ) if score_specified: utils.err("you cannot specify multiple score options") score_specified = True if name == "score" and not utils.is_score(value): utils.err( "invalid score '%s', use integer or INFINITY or -INFINITY" % value ) rsc_colocation.set(name,value) set_add_resource_sets(rsc_colocation, current_set, cib) utils.replace_cib_configuration(cib)
def resource_operation_add(res_id, argv): if len(argv) < 1: usage.resource() sys.exit(1) op_name = argv.pop(0) dom = utils.get_cib_dom() resource_found = False for resource in dom.getElementsByTagName("primitive"): if resource.getAttribute("id") == res_id: resource_found = True break if not resource_found: print "Unable to find resource: %s" % res_id sys.exit(1) op_properties = convert_args_to_tuples(argv) op_properties.sort(key=lambda a:a[0]) op_properties.insert(0,('name', op_name)) found_match = False op = dom.createElement("op") op_id = res_id + "-" for prop in op_properties: op.setAttribute(prop[0], prop[1]) op_id += prop[0] + "-" + prop[1] + "-" op_id = op_id[:-1] op.setAttribute("id", op_id) operations = resource.getElementsByTagName("operations") if len(operations) == 0: operations = dom.createElement("operations") resource.appendChild(operations) else: operations = operations[0] operations.appendChild(op) utils.replace_cib_configuration(dom)
def stonith_level_clear(node = None): dom = utils.get_cib_dom() ft = dom.getElementsByTagName("fencing-topology") if len(ft) == 0: return if node == None: ft = ft[0] childNodes = ft.childNodes[:] for node in childNodes: node.parentNode.removeChild(node) else: fls = dom.getElementsByTagName("fencing-level") if len(fls) == 0: return for fl in fls: if fl.getAttribute("target") == node or fl.getAttribute("devices") == node: fl.parentNode.removeChild(fl) utils.replace_cib_configuration(dom)
def colocation_set(argv): setoptions = [] for i in range(len(argv)): if argv[i] == "setoptions": setoptions = argv[i + 1:] argv[i:] = [] break current_set = set_args_into_array(argv) cib = utils.get_cib_etree() constraints = cib.find(".//constraints") if constraints == None: constraints = ET.SubElement(cib, "constraints") rsc_colocation = ET.SubElement(constraints, "rsc_colocation") rsc_colocation.set("id", utils.find_unique_id(cib, "pcs_rsc_colocation")) for opt in setoptions: if opt.find("=") != -1: name, value = opt.split("=") rsc_colocation.set(name, value) set_add_resource_sets(rsc_colocation, current_set, cib) utils.replace_cib_configuration(cib)
def resource_group_rm(group_name, resource_ids): resource_id = resource_ids[0] dom = utils.get_cib_dom() dom = dom.getElementsByTagName("configuration")[0] group_match = None for group in dom.getElementsByTagName("group"): if group.getAttribute("id") == group_name: group_match = group break if not group_match: print "ERROR: Group '%s' does not exist" % group_name sys.exit(1) resources_to_move = [] for resource_id in resource_ids: found_resource = False for resource in group_match.getElementsByTagName("primitive"): if resource.getAttribute("id") == resource_id: found_resource = True resources_to_move.append(resource) break if not found_resource: print "ERROR Resource '%s' does not exist in group '%s'" % (resource_id, group_name) sys.exit(1) for resource in resources_to_move: parent = resource.parentNode resource.parentNode.removeChild(resource) parent.parentNode.appendChild(resource) if len(group_match.getElementsByTagName("primitive")) == 0: group_match.parentNode.removeChild(group_match) utils.replace_cib_configuration(dom) return True
def constraint_resource_update(old_id, passed_dom=None): dom = utils.get_cib_dom() if passed_dom is None else passed_dom new_id = None clone_ms_parent = utils.dom_get_resource_clone_ms_parent(dom, old_id) if clone_ms_parent: new_id = clone_ms_parent.getAttribute("id") if new_id: constraints = dom.getElementsByTagName("rsc_location") constraints += dom.getElementsByTagName("rsc_order") constraints += dom.getElementsByTagName("rsc_colocation") attrs_to_update = ["rsc", "first", "then", "with-rsc"] for constraint in constraints: for attr in attrs_to_update: if constraint.getAttribute(attr) == old_id: constraint.setAttribute(attr, new_id) if passed_dom is None: utils.replace_cib_configuration(dom) if passed_dom: return dom