public void updatePresence(Packet presence)
{
String type = presence.Type;
Session session = presence.Session;
Presence sessionPresence = session.getPresence();
String recipient = presence.To;
JabberID recipientID;
Boolean isUserSent;//it is not used yet, may be used in S2S comunication
//packet filling begins
if (recipient == null)
{ //directed to server
//server name
recipientID = new JabberID(Server.JabberServer.server_name);
isUserSent = true;
}
else
{
recipientID = new JabberID(recipient);
if (user.Equals(recipientID.User))
{
isUserSent = false;
}
else
{
isUserSent = true;
}
}
String sender = presence.From;
JabberID SenderID;
if (sender == null)
{
SenderID = session.JID;
}
else
{
SenderID = new JabberID(sender);
}
//added by marko
presence.From = SenderID.User + "@" + SenderID.Domain;
//original line:
//String subscriber = isUserSent ? recipientID.ToString() : SenderID.ToString();
String subscriber = recipientID.ToString();
if (type == null)
{
type = "available";
}
//packet filling ends
//packet routing begins
if (type.Equals("available") || type.Equals("unavailable"))
{
//user-managed presence updates are delivered untouched
if (!recipientID.ToString().Equals(Server.JabberServer.server_name))
{
//may need some work;
MessageHandler.deliverPacket(presence);
return;
}
//Server-managed presence updates are forwarded only to subscribers
//update the session's presence status
sessionPresence.Availible = type.Equals("available");
sessionPresence.Show = presence.getChildValue("show");
sessionPresence.Status = presence.getChildValue("status");
String priority = presence.getChildValue("priority");
sessionPresence.Priority = priority;
if (priority != null)
{
//some code to set the priority of the session
try
{
session.setPriority(int.Parse(priority));
}
catch (Exception ex)
{
JabberServer.output.WriteLine("Error in incoming priority, setting priority to default: 1");
session.setPriority(0);
}
}
//deliver to all users presence subscribers
updateSubscribers(presence);
informSubscriber();
return;
}
if (type.Equals("probe"))
{
//needed for server to server communication!!!!
JabberServer.output.WriteLine("Roster: We don't handle probes yet " + presence.ToString());
return;
}
//Marko believes it is checked for exceptions
//Subscriber sendersSubscriberItem = (Subscriber)subscribers[subscriber];
Subscriber sendersSubscriberItem = new Subscriber();
Hashtable Rosters = JabberServer.RosterHashtable;//get reference to rosters
//prepare sender's roster
UserRoster senderRoster = Rosters[this.user] as UserRoster;//getting senders roster
if (senderRoster == null)
{
senderRoster = new UserRoster();
senderRoster.user = this.user;
}
if (!Rosters.ContainsKey(this.user))
{
Rosters.Add(this.user, senderRoster);
}
//prepare recipients roster
UserRoster recipientRoster = Rosters[recipientID.User] as UserRoster;//getting recipients roster
if (recipientRoster == null)
{
recipientRoster = new UserRoster();
recipientRoster.user = recipientID.User;
}
if (!Rosters.ContainsKey(recipientID.User))
{
Rosters.Add(recipientID.User, recipientRoster);
}
sendersSubscriberItem = senderRoster.subscribers[subscriber] as Subscriber;//extra
if (sendersSubscriberItem == null)
{
//create subscription item
sendersSubscriberItem = new Subscriber();
sendersSubscriberItem.subscription = "none";
senderRoster.subscribers.Add(recipient, sendersSubscriberItem);
Packet itempacket = new Packet("item");
itempacket.setAttribute("jid", subscriber);
senderRoster.items.Add(subscriber, itempacket);
}
//begin subscription managment
if (type.Equals("subscribe"))
{
sendersSubscriberItem.ask = type;//set up subscription status
}
else if (type.Equals("subscribed"))//if subscription accepted
{
sendersSubscriberItem.ask = null;
if (sendersSubscriberItem.subscription.Equals("from"))
{
sendersSubscriberItem.subscription = "both";
}
else if (sendersSubscriberItem.subscription.Equals("none"))
{
sendersSubscriberItem.subscription = "to";
}
}
else if (type.Equals("unsubscribed"))
{//sender does not want to give presence updates to recipient
sendersSubscriberItem.ask = null;
if (sendersSubscriberItem.subscription.Equals("to"))
{
sendersSubscriberItem.subscription = "none";
}
else if (sendersSubscriberItem.subscription.Equals("both"))
{
sendersSubscriberItem.subscription = "from";
}
//notify recipient of sender's unavailble status:
Packet unavailiblePacket = new Packet("presence");
unavailiblePacket.setFrom(presence.From);
unavailiblePacket.To = presence.To;
unavailiblePacket.setAttribute("type", "unavailable");
MessageHandler.deliverPacket(unavailiblePacket);
}
else if (type.Equals("unsubscribe"))
{//sender no longer interested in recieving presence updates from recipient
sendersSubscriberItem.ask = null;
if (sendersSubscriberItem.subscription.Equals("both"))
{
sendersSubscriberItem.subscription = "to";
}
else if (sendersSubscriberItem.subscription.Equals("from"))
{
sendersSubscriberItem.subscription = "none";
}
}
//update the corresponding changes in <items> table in sender's record used for delivery
Packet item = (Packet)senderRoster.items[subscriber];
if (item != null)
{
item.setAttribute("subscription", sendersSubscriberItem.subscription);
item.setAttribute("ask", sendersSubscriberItem.ask);
Packet iq = new Packet("iq");
iq.Type = "set";
Packet query = new Packet("query");
query.setAttribute("xmlns", "jabber:iq:roster");
query.setParent(iq);
item.setParent(query);
iq.To = this.user + "@" + Server.JabberServer.server_name;
//Forward the subscription packet to recipient
MessageHandler.deliverPacketToAll(user, iq);//may need some correction!
}
//processing of recipients roster begins
if (sendersSubscriberItem.ask == null)
{
//the recipient user rosters update!
Subscriber recipientsSubscriberItem = recipientRoster.subscribers[user + "@" + JabberServer.server_name] as Subscriber;//extra
if (recipientsSubscriberItem == null)
{
//create subscription item
recipientsSubscriberItem = new Subscriber();
recipientsSubscriberItem.subscription = "none";
recipientRoster.subscribers.Add(user + "@" + JabberServer.server_name, recipientsSubscriberItem);
}
//if type is subscribe, changes in this roster are not needed!!!
if (type.Equals("subscribed"))//if subscription accepted
{
recipientsSubscriberItem.ask = null;
if (recipientsSubscriberItem.subscription.Equals("none"))
{
recipientsSubscriberItem.subscription = "from";
}
else if (recipientsSubscriberItem.subscription.Equals("to"))
{
recipientsSubscriberItem.subscription = "both";
}
}
else if (type.Equals("unsubscribed"))
{//sender no longer interested in giving presence updates to recipient
recipientsSubscriberItem.ask = null;
if (recipientsSubscriberItem.subscription.Equals("both"))
{
recipientsSubscriberItem.subscription = "to";
}
else if (recipientsSubscriberItem.subscription.Equals("from"))
{
recipientsSubscriberItem.subscription = "none";
}
}
else if (type.Equals("unsubscribe"))
{//sender no longer interested in recieving presence updates from recipient
recipientsSubscriberItem.ask = null;
if (recipientsSubscriberItem.subscription.Equals("to"))
{
recipientsSubscriberItem.subscription = "none";
}
else if (recipientsSubscriberItem.subscription.Equals("both"))
{
recipientsSubscriberItem.subscription = "from";
}
//notify sender of unavailble status:
Packet unavailiblePacket = new Packet("presence");
unavailiblePacket.setFrom(presence.To);
unavailiblePacket.To = presence.From;
unavailiblePacket.setAttribute("type", "unavailable");
MessageHandler.deliverPacket(unavailiblePacket);
}
//update the corresponding changes in <items> table in sender's record used for delivery
Packet item2 = (Packet)recipientRoster.items[user + "@" + JabberServer.server_name];
if (item2 != null)
{
item2.setAttribute("subscription", recipientsSubscriberItem.subscription);
item2.setAttribute("ask", recipientsSubscriberItem.ask);
Packet iq = new Packet("iq");
iq.Type = "set";
Packet query = new Packet("query");
query.setAttribute("xmlns", "jabber:iq:roster");
query.setParent(iq);
item2.setParent(query);
iq.To = recipientID.User + "@" + recipientID.Domain;
//Forward the subscription packet to recipient
MessageHandler.deliverPacketToAll(user, iq);//may need some correction!
}
}
MessageHandler.deliverPacket(presence);
}//UpdatePresense
}//getPacket
public void updateRoster(Packet packet)
{
//Extract the query packet
Packet rosterQuery = packet.getFirstChild("query");
rosterQuery.setAttribute("xmlns", "jabber:iq:roster");
foreach (object rosterItems in rosterQuery.getChildren())
{
//for each <item> packet in the query
Object child = rosterItems;
if (child is Packet)
{
Packet itemPacket = child as Packet;
String subJID = itemPacket.getAttribute("jid");
//create Subscriber object if needed
//Subscriber sendersSubscriberItem = this.subscribers[subJID] as Subscriber;
if ((Server.JabberServer.RosterHashtable[this.user] as UserRoster) == null)
{
Server.JabberServer.RosterHashtable[this.user] = new UserRoster();
}
Subscriber sub = (Server.JabberServer.RosterHashtable[this.user] as UserRoster).subscribers[subJID] as Subscriber;
if (sub == null)
{
sub = new Subscriber();
sub.subscription = "none";
sub.ask = null;
(Server.JabberServer.RosterHashtable[this.user] as UserRoster).subscribers.Add(subJID, sub);
}
if (itemPacket.getAttribute("subscription") != null)
{
//signal to remove
if (itemPacket.getAttribute("subscription").Equals("remove"))
{
//remove from subscribers item
JabberID otherID = new JabberID(subJID);
(Server.JabberServer.RosterHashtable[this.user] as UserRoster).subscribers.Remove(otherID.ToString());
if ((Server.JabberServer.RosterHashtable[this.user] as UserRoster).items.ContainsKey(subJID))
{
//remove <item> packet
(Server.JabberServer.RosterHashtable[this.user] as UserRoster).items.Remove(subJID);
}
if ((Server.JabberServer.RosterHashtable[otherID.User] as UserRoster) == null)
{
Server.JabberServer.RosterHashtable[otherID.User] = new UserRoster();
}
//set only subscription=none from other item
sub = ((Server.JabberServer.RosterHashtable[otherID.User] as UserRoster).subscribers[this.user + "@" + Server.JabberServer.server_name]) as Subscriber;
if (sub == null)
{
sub = new Subscriber();
}
sub.subscription = "none";
sub.ask = null;
(Server.JabberServer.RosterHashtable[otherID.User] as UserRoster).subscribers[this.user + "@" + Server.JabberServer.server_name] = sub;
//(Server.JabberServer.RosterHashtable[subJID] as UserRoster).subscribers.Add(subJID, sendersSubscriberItem);
//}
if ((Server.JabberServer.RosterHashtable[otherID.User] as UserRoster).items.ContainsKey(this.user + "@" + Server.JabberServer.server_name))
{
//update <item> packet
//itemPacket.removeAttribute("subscription");
Packet mypacket = (Server.JabberServer.RosterHashtable[otherID.User] as UserRoster).items[this.user + "@" + Server.JabberServer.server_name] as Packet;
mypacket.setAttribute("subscription", "none");
mypacket.setAttribute("ask", null);
(Server.JabberServer.RosterHashtable[otherID.User] as UserRoster).items[this.user + "@" + Server.JabberServer.server_name] = mypacket;
}
}
}
else
{
//update <item> with the subscriber info for roster push
itemPacket.setAttribute("subscription", sub.subscription);
itemPacket.setAttribute("ask", sub.ask);
//Packet test = items[subJID];
if (!(Server.JabberServer.RosterHashtable[this.user] as UserRoster).items.ContainsKey(subJID))
{
//store updated <item> packet
(Server.JabberServer.RosterHashtable[this.user] as UserRoster).items.Add(subJID, itemPacket);
}
else
{
(Server.JabberServer.RosterHashtable[this.user] as UserRoster).items[subJID] = itemPacket;
}
}
}
}
//Roster push
packet.Type = "set";
JabberID jidTo = packet.getSession().getJID();
packet.To = jidTo.User + "@" + jidTo.Domain;
packet.removeAttribute("from");
//needs to be verified
MessageHandler.deliverPacket(packet);
}//update roster
private void restoreFromFile() {
DataSet ds = new DataSet();
try {
ds.ReadXml(users_file_name);
} catch (FileNotFoundException ex) {
JabberServer.output.WriteLine("Could not load User database!" + ex.Message);
return;
}
foreach (DataRow user in ds.Tables["User"].Rows) {
User u = getUserIndex().addUser(user["username"] as String);
u.setPassword(user["password"] as String);
u.setHash(user["hash"] as String);
u.setSequence(user["sequence"] as String);
u.setToken(user["token"] as String);
foreach (DataRow item in user.GetChildRows("Roster")) {
UserRoster ur = (UserRoster)RosterHashtable[item["username"]];
Hashtable itemsHash = ur.items;
Hashtable subscribersHash = ur.subscribers;
Packet itemPacket = new Packet("item");
itemPacket["jid"] = item["jid"] as String;
itemPacket["name"] = item["name"] as String;
itemPacket["subscription"] = "both";
String group = item["group"] as String;
if (group != null) {
new Packet("group", group).Parent = itemPacket;
}
itemsHash[item["jid"]] = itemPacket;
Subscriber s = new Subscriber();
s.subscription = "both";
subscribersHash[item["jid"]] = s;
}
}
}