IMS PCRF开源代码

/* The UCT Policy Control Framework Copyright (C) 2006 - University of Cape Town Richard Good <rgood@crg.ee.uct.ac.za> This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <http://www.gnu.org/licenses/>. */ /** March 2008*/ package pcrf; import java.io.*; import java.net.*; import java.util.*; import org.apache.commons.httpclient.HttpClient; import org.apache.commons.httpclient.UsernamePasswordCredentials; import org.apache.commons.httpclient.auth.AuthScope; import org.apache.commons.httpclient.methods.GetMethod; import org.apache.commons.httpclient.methods.PutMethod; import org.apache.commons.httpclient.methods.StringRequestEntity; import javax.xml.parsers.DocumentBuilder; import javax.xml.parsers.DocumentBuilderFactory; import javax.xml.parsers.ParserConfigurationException; import org.w3c.dom.Document; import org.w3c.dom.Element; import org.w3c.dom.NodeList; import org.xml.sax.SAXException; import org.apache.xml.serialize.XMLSerializer; import org.apache.xml.serialize.OutputFormat; /** * xcap: * * This class defines the XCAP methods for retrieving and modifying policies stored on an XDMS server * It also defines methods for parsing and reading the XML document * @author Richard Good */public class xcap { String policy_db = ""; static private Document dom = null; String username = ""; String password = ""; int remote_xcap_server; //zero if local one if remote /** * Constructor for xcap * -initialises variables * @param args */ public xcap(String policy_db, String policy, String username, String password, int remote_xcap_server) throws IOException { this.username = username; this.password = password; this.remote_xcap_server = remote_xcap_server; /**TO DO fix OpenXCAP to allow different Applications*/ if(remote_xcap_server == 1) { this.policy_db = policy_db + "resource-lists/users/" + username + "/resource-lists.xml"; } else { this.policy_db = policy_db + policy + ".xml"; } parseXML(getXML()); } /** * This converts an XML Document into a String used to convey the doc via HTTP * @param dom The XML document to be converted to a string * @return xmlStr the String representation of the xml doc **/ private String XMLDocToString(Document dom) { String xmlStr = null; StringWriter strWriter = null; XMLSerializer probeMsgSerializer = null; OutputFormat outFormat = new OutputFormat(); try { probeMsgSerializer = new XMLSerializer(); strWriter = new StringWriter(); outFormat = new OutputFormat(); // Setup format settings outFormat.setVersion("1.0"); outFormat.setEncoding("UTF-8"); outFormat.setIndenting(true); outFormat.setIndent(4); // Define a Writer probeMsgSerializer.setOutputCharStream(strWriter); // Apply the format settings probeMsgSerializer.setOutputFormat(outFormat); // Serialize XML Document probeMsgSerializer.serialize(dom); xmlStr = strWriter.toString(); strWriter.close(); } catch (IOException ioEx) { System.out.println("Error " + ioEx); } return xmlStr; } /** * This creates the XML parser **/ private void parseXML(InputStream xmlDoc) { //get the factory DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance(); try { //Using factory get an instance of document builder DocumentBuilder db = dbf.newDocumentBuilder(); //parse using builder to get DOM representation of the XML file dom = db.parse(xmlDoc); }catch(ParserConfigurationException pce) { pce.printStackTrace(); }catch(SAXException se) { se.printStackTrace(); }catch(IOException ioe) { ioe.printStackTrace(); } } /** * This receives a list of codecs - checks the domain policies and returns whether they are acceptable or not * @param codecs vector of requested requested codecs * @return 1 if authorised 0 if not */ public int checkCodecs(Vector codecs) throws IOException { //fetch and parse the policy XML document parseXML(getXML()); //get the root elememt Element docEle = dom.getDocumentElement(); //get all authorised codec id's in a vector Vector authCodecs = new Vector(); NodeList nl = docEle.getElementsByTagName("rule"); if(nl != null && nl.getLength() > 0) { for(int i = 0 ; i < nl.getLength();i++) { //get the authorised codec rule Element rule = (Element)nl.item(i); String id = rule.getAttribute("id"); if(id.equals("codec authorisation")) { NodeList nl3 = rule.getElementsByTagName("conditions"); if(nl3 != null && nl3.getLength() > 0) { Element conditions = (Element)nl3.item(0); NodeList nl1 = conditions.getElementsByTagName("codec_id"); if(nl1 != null && nl1.getLength() > 0) { Element codec_id = (Element)nl1.item(0); NodeList nl2 = codec_id.getElementsByTagName("id"); if(nl2 != null && nl2.getLength() > 0) { for(int a = 0 ; a < nl2.getLength();a++) { Element ID = (Element)nl2.item(a); authCodecs.add(ID.getFirstChild().getNodeValue()); } } } } } } } //check that all requested codecs are authorised int authorised = 1; int supported = 0; for ( int b = 0; b < codecs.size(); b++) { supported = 0; for (int c = 0; c < authCodecs.size(); c++) { if(Integer.valueOf(codecs.get(b).toString()).intValue() == Integer.valueOf(authCodecs.get(c).toString()).intValue()) { supported = 1; } } //Note this is to support dynamic codecs, see RFC 3551 if (supported == 0 && (Integer.valueOf(codecs.get(b).toString()).intValue() < 96 || Integer.valueOf(codecs.get(b).toString()).intValue() > 127)) { authorised = 0; } } //1 if authorised 0 if not return authorised; } /** * This receives the orig and term domains * It checks that both domains are allowed * @param String orig originating domain * @param String term terminating domain * @return 1 if authorised 0 if not */ public int checkDomain(String orig, String term) throws IOException { //fetch and parse the policy XML document parseXML(getXML()); //get the root elememt Element docEle = dom.getDocumentElement(); int orig_authorised = 0; int term_authorised = 0; Vector domains = new Vector(); NodeList nl = docEle.getElementsByTagName("rule"); if(nl != null && nl.getLength() > 0) { for(int i = 0 ; i < nl.getLength();i++) { //get the authorised codec rule Element rule = (Element)nl.item(i); String id = rule.getAttribute("id"); if(id.equals("domain authorisation")) { NodeList nl3 = rule.getElementsByTagName("conditions"); if(nl3 != null && nl3.getLength() > 0) { Element conditions = (Element)nl3.item(0); NodeList nl1 = conditions.getElementsByTagName("domain"); if(nl1 != null && nl1.getLength() > 0) { Element domain = (Element)nl1.item(0); NodeList nl2 = domain.getElementsByTagName("address"); if(nl2 != null && nl2.getLength() > 0) { for(int a = 0 ; a < nl2.getLength();a++) { Element address = (Element)nl