#!/usr/bin/perl

# The above Perl path may vary on your system; fix it!!! -*- perl -*-

# dnssd - Search for network printers with the avahi-browse command

#         (Zeroconf, DNS-SD)

# Printer discovery CUPS backend (like the SNMP backend)

# See also http://qa.mandriva.com/show_bug.cgi?id=21812

# Copyright 2007 Till Kamppeter <till.kamppeter@gmail.com>

#

#  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 2 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, write to the Free Software

#  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,

#  USA.

# Usage: 

#

# cp dnssd /usr/lib/cups/backend/

# chmod 755 /usr/lib/cups/backend/dnssd

# killall -HUP cupsd (or "/etc/init.d/cups restart", CUPS 1.1.x only)

# lpinfo -v (or use any printer setup tool)

#use strict;

$0 =~ m!^(.*)/([^/]+)\s*$!;

my $progname = ($2 || $0 || "dnssd");

my $progpath = ($1 || "/usr/lib/cups/backend");

if ($ARGV[0]){

    die "This backend is only for printer discovery, not for actual printing.\n";

}

my $avahicmd = "avahi-browse -k -t -v -r -a 2> /dev/null";

# IPs which are for computers, consider their printer entries as queues

# set up with the local printing system (CUPS, LPD, Windows/Samba SMB, ...)

my @computerips = ();

my $output;

my $hosts;

my ($interface, $nettype, $ip, $host, $make, $model, $description, $cmd, $makemodel, $deviceid, $protocol, $port, $uriext, $uri);

open (AVAHI, "$avahicmd |") or exit 0;

while (my $line = <AVAHI>) {

    chomp ($line);

    if ($line =~ /^\s*=\s+(\S+)\s+(\S+)\s+(.*?)\s+(\S+)\s+(\S+)\s*$/) {

	# New item

	$interface = $1;

	$nettype = $2;

	my $itemname = $3;

	my $protocolinfo = $4;

        if ($protocolinfo =~ /_workstation/) {

	    $protocol = "computer";

	} elsif ($protocolinfo =~ /_pdl-datastream/) {

	    $protocol = "socket";

	} elsif ($protocolinfo =~ /_printer/) {

	    $protocol = "lpd";

	} elsif ($protocolinfo =~ /_ipp/) {

	    $protocol = "ipp";

	}

    } elsif ($line =~ /^\s*hostname\s*=\s*\[([^\]]+)\]\s*$/) {

	$host = $1;

	$host =~ s/\.local\.?$//;

    } elsif ($line =~ /^\s*address\s*=\s*\[([^\]]+)\]\s*$/) {

	$ip = $1;

	if ($protocol eq "computer") {

	    push (@computerips, $ip);

	    $protocol = "";

	}

    } elsif ($line =~ /^\s*port\s*=\s*\[([^\]]+)\]\s*$/) {

	$port = $1;

    } elsif ($line =~ /^\s*txt\s*=\s*\[(.+)\]\s*$/) {

	my $info = $1;

	if ($protocol) {

	    my ($ty, $product, $pdls, $usb_MFG, $usb_MDL, $usb_DES, $usb_CMD) = 

		("", "", "", "", "", "", "");

	    while ($info =~ s/^\s*\"([^\"]+)\"\s*//) {

		my $infoitem = $1;

		if ($infoitem =~ /^([^=]*)=(.*)$/) {

		    my $field = $1;

		    my $content = $2;

		    if ($field eq "ty") {

			$ty = $content;

		    } elsif ($field eq "product") {

			$product = $content;

			$product =~ s/^\((.*)\)$/$1/;

		    } elsif ($field eq "usb_MFG") {

			$usb_MFG = $content;

		    } elsif ($field eq "usb_MDL") {

			$usb_MDL = $content;

		    } elsif ($field eq "usb_DES") {

			$usb_DES = $content;

		    } elsif ($field eq "usb_CMD") {

			$usb_CMD = $content;

		    } elsif ($field eq "rp") {

			$uriext = $content;

		    } elsif ($field eq "pdl") {

			while ($content =~ s/^\s*([^\,]+?)\s*\,\s*//) {

			    my $i = $1;

			    if ($i =~ m!\b(postscript|ps)\b!i) {

				$pdls .= "POSTSCRIPT,";

			    } elsif ($i =~ m!\b(pdf)\b!i) {

				$pdls .= "PDF,";

			    } elsif ($i =~ m!\b(pcl6|pclxl|pxl)\b!i) {

				$pdls .= "PCLXL,";

			    } elsif ($i =~ m!\b(pcl[345][ce]?|pcl)\b!i) {

				$pdls .= "PCL,";

			    }

			}

			$pdls =~ s/\,$//;

		    }

		}

	    }

	    $usb_MDL ||= $ty;

	    $usb_DES ||= $product;

	    if ($usb_MFG) {

		$make = $usb_MFG;

	    } elsif ($usb_DES =~ /^KONICA\s*MINOLTA\b/i) { 

		$make = "KONICA MINOLTA";

	    } elsif ($usb_DES) {

		$usb_DES =~ /^\s*(\S*)\b/;

		$make = $1;

	    }

	    $model = $usb_MDL;

	    if (!$model) {

		$usb_DES =~ /^\s*\S*\s*(.*)$/;

		$model = $1;

	    }

	    $usb_CMD ||= $pdls;

	    my $extra;

	    if ($protocol eq "socket") {

		if ($port eq "9100") {

		    $uri = "socket://$ip";

		} else {

		    $uri = "socket://$ip:$port";

		}

		$extra = "Port $port";

	    } elsif ($protocol eq "lpd") {

		$uri = "lpd://$ip" . ($uriext ? "/$uriext" : "");

		$extra = ($uriext ? "Queue: $uriext" : "Default queue");

	    } elsif ($protocol eq "ipp") {

		$uri = "ipp://$ip:$port" . ($uriext ? "/$uriext" : "");

		$extra = ($uriext ? "Queue: $uriext" : "Default queue");

	    }

	    if ($make && $model) {

		$make =~ s/Hewlett.?Packard/HP/i;

		$make =~ s/Lexmark.?International/Lexmark/i;

		$model =~ s/Hewlett.?Packard/HP/i;

		$model =~ s/Lexmark.?International/Lexmark/i;

		while ($model =~ s/^\s*$make\s*//i) {};

		$makemodel = "$make $model";

	    } elsif ($usb_DES) {

		$makemodel = $usb_DES;

	    } else {

		$makemodel = "Unknown";

	    }

	    $deviceid = ($usb_MFG ? "MFG:$usb_MFG;" : "") .

		($usb_MDL ? "MDL:$usb_MDL;" : "") .

		($usb_DES ? "DES:$usb_DES;" : "") .

		($usb_CMD ? "CMD:$usb_CMD;" : "");

	    $deviceid .= "CLS:PRINTER;" if $deviceid;

	    $hosts->{$ip} = $hostname if ($hostname);

	    $output->{$ip}{$protocol}{$uriext} =

		"network $uri \"$makemodel\" \"$makemodel $ip ($extra)\" \"$deviceid\"\n";

	    ($interface, $nettype, $ip, $host, $make, $model, $description, $cmd, $makemodel, $deviceid, $protocol, $port, $uriext, $uri) =

		("", "", "", "", "", "", "", "", "", "", "", "", "", "");

	}

    }

}

my $localqueues = {};

my $queue = undef;

if (open LPSTAT, "LC_ALL=C lpstat -l -p -v |") {

    while (my $line = <LPSTAT>) {

	chomp $line;

	if ($line =~ /^printer\s+(\S+)/i) {

	    $queue = $1;

	    $localqueues->{$queue} = {};

	} elsif ($queue) {

	    if ($line =~ /^\s+Connection:\s+remote/i) {

		$localqueues->{$queue}{remote} = 1;

	    } elsif ($line =~ /^\s+Interface:\s+(\S+)/i) {

		$localqueues->{$queue}{interface} = $1;

	    } elsif ($line =~ /^device\s+for\s+(\S+)\s*:\s*(\S+)/i) {

		$localqueues->{$1}{uri} = $2;

	    }

	}

    }

    close LPSTAT

}

my @localips = ();

if (open IFCONFIG, "LC_ALL=C /sbin/ifconfig |") {

    while (my $line = <IFCONFIG>) {

	chomp $line;

	if ($line =~ /^\s*inet\s+addr:\s*(\S+)/i) {

	    push (@localips, $1);

	}

    }

    close IFCONFIG;

}

foreach my $ip (keys(%{$output})) {

    # Do not list print queues of the local machine

    next if member($ip, @localips);

    if ($output->{$ip}{"socket"}) {

	foreach my $uriext (keys(%{$output->{$ip}{"socket"}})) {

	    if (keys(%{$output->{$ip}{"socket"}}) == 1) {

		$output->{$ip}{"socket"}{$uriext} =~

		    s/^(\s*\S*\s*\S*\s*\"[^\"]*\"\s*\"[^\"\(]*?)\s*\([^\)]*\)\s*(\"\s*.*)$/$1$2/;

	    }

	    print $output->{$ip}{"socket"}{$uriext};

	}

    } elsif ($output->{$ip}{"lpd"}) {

	foreach my $uriext (keys(%{$output->{$ip}{"lpd"}})) {

	    if (keys(%{$output->{$ip}{"lpd"}}) == 1) {

		$output->{$ip}{"lpd"}{$uriext} =~

		    s/^(\s*\S*\s*\S*\s*\"[^\"]*\"\s*\"[^\"\(]*?)\s*\([^\)]*\)\s*(\"\s*.*)$/$1$2/;

	    }

	    print $output->{$ip}{"lpd"}{$uriext};

	}

    } elsif ($output->{$ip}{"ipp"}) {

	foreach my $uriext (keys(%{$output->{$ip}{"ipp"}})) {

	    if ($uriext =~ /^(printers|classes)\/(\S+)$/) {

		# Queue from a CUPS server. We should suppress it if it

		# is a queue received via CUPS broadcast

		$queue=$2;

		if (defined($localqueues->{"$queue\@$ip"})) {

		    $queue = "$queue\@$ip";

		} elsif (defined($localqueues->{"$queue\@$hosts->{$ip}"})) {

		    $queue = "$queue\@$hosts->{$ip}";

		}

		if (defined($localqueues->{$queue})) {

		    if ($localqueues->{$queue}{remote} &&

			($localqueues->{$queue}{uri} =~

			 /^ipp:\/\/([^\/:]+)(:\d+|)\/(\S+)/)) {

			my $host = $1;

			my $ue = $3;

			if ($host !~ /\d+\.\d+\.\d+\.\d+/) {

			    $host =

				join(".", unpack("C4", gethostbyname($host)));

			}

			next if ($host eq $ip) && ($ue eq $uriext);

		    }

		}

	    }

	    if (keys(%{$output->{$ip}{"ipp"}}) == 1) {

		$output->{$ip}{"ipp"}{$uriext} =~

		    s/^(\s*\S*\s*\S*\s*\"[^\"]*\"\s*\"[^\"]*?)\s*\([^\)]*\)\s*(\"\s*.*)$/$1$2/;

	    }

	    print $output->{$ip}{"ipp"}{$uriext};

	}

    }

}

exit 0;

# member( $a, @b ) returns 1 if $a is in @b, 0 otherwise.

sub member { my $e = shift; foreach (@_) { $e eq $_ and return 1 } 0 };